xbmc/addons/test/TestAddonDatabase.cpp

   1 /*
   2  *  Copyright (C) 2016-2018 Team Kodi
   3  *  This file is part of Kodi - https://kodi.tv
   4  *
   5  *  SPDX-License-Identifier: GPL-2.0-or-later
   6  *  See LICENSES/README.md for more information.
   7  */
   8
   9 #include "addons/AddonDatabase.h"
  10 #include "addons/addoninfo/AddonInfoBuilder.h"
  11 #include "filesystem/SpecialProtocol.h"
  12 #include "settings/AdvancedSettings.h"
  13
  14 #include <set>
  15 #include <utility>
  16
  17 #include <gtest/gtest.h>
  18
  19 using namespace ADDON;
  20
  21
  22 class AddonDatabaseTest : public ::testing::Test
  23 {
  24 protected:
  25   DatabaseSettings settings;
  26   CAddonDatabase database;
  27
  28   void SetUp() override
  29   {
  30     settings.type = "sqlite3";
  31     settings.name = "test";
  32     settings.host = CSpecialProtocol::TranslatePath("special://temp/");
  33
  34     database.Connect("test", settings, true);
  35
  36     std::set<std::string> installed{"repository.a", "repository.b"};
  37     database.SyncInstalled(installed, installed, std::set<std::string>());
  38
  39     std::vector<AddonInfoPtr> addons;
  40     CreateAddon(addons, "foo.bar", "1.0.0");
  41     database.SetRepoUpdateData("repository.a", {});
  42     database.UpdateRepositoryContent("repository.a", CAddonVersion("1.0.0"), "test", addons);
  43
  44     addons.clear();
  45     CreateAddon(addons, "foo.baz", "1.1.0");
  46     database.SetRepoUpdateData("repository.b", {});
  47     database.UpdateRepositoryContent("repository.b", CAddonVersion("1.0.0"), "test", addons);
  48   }
  49
  50   void CreateAddon(std::vector<AddonInfoPtr>& addons, std::string id, const std::string& version)
  51   {
  52     CAddonInfoBuilderFromDB builder;
  53     builder.SetId(std::move(id));
  54     builder.SetVersion(CAddonVersion(version));
  55     addons.push_back(builder.get());
  56   }
  57
  58   void TearDown() override
  59   {
  60     database.Close();
  61   }
  62 };
  63
  64
  65 TEST_F(AddonDatabaseTest, TestFindById)
  66 {
  67   VECADDONS addons;
  68   EXPECT_TRUE(database.FindByAddonId("foo.baz", addons));
  69   EXPECT_EQ(1U, addons.size());
  70   EXPECT_EQ(addons.at(0)->ID(), "foo.baz");
  71   EXPECT_EQ(addons.at(0)->Version().asString(), "1.1.0");
  72   EXPECT_EQ(addons.at(0)->Origin(), "repository.b");
  73 }
  74
  75 TEST_F(AddonDatabaseTest, TestFindByNonExistingId)
  76 {
  77   VECADDONS addons;
  78   EXPECT_TRUE(database.FindByAddonId("does.not.exist", addons));
  79   EXPECT_EQ(0U, addons.size());
  80 }