| // Copyright 2018 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "chrome/browser/component_updater/metadata_table_chromeos.h" |
| #include "content/public/test/test_browser_thread_bundle.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace component_updater { |
| |
| namespace { |
| |
| constexpr char kMetadataContentItemHashedUserIdKey[] = "hashed_user_id"; |
| constexpr char kMetadataContentItemComponentKey[] = "component"; |
| constexpr char kHashedUserId[2][6] = {"user1", "user2"}; |
| constexpr char kComponent[2][11] = {"component1", "component2"}; |
| |
| } // namespace |
| |
| class CrOSComponentInstallerMetadataTest : public testing::Test { |
| public: |
| CrOSComponentInstallerMetadataTest() {} |
| ~CrOSComponentInstallerMetadataTest() override {} |
| |
| protected: |
| content::TestBrowserThreadBundle thread_bundle_; |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(CrOSComponentInstallerMetadataTest); |
| }; |
| |
| TEST_F(CrOSComponentInstallerMetadataTest, Add) { |
| // Initialize protobuf object. |
| std::unique_ptr<MetadataTable> metadata_table = |
| MetadataTable::CreateForTest(); |
| |
| // Add component items. |
| metadata_table->AddItem(kHashedUserId[0], kComponent[0]); |
| metadata_table->AddItem(kHashedUserId[1], kComponent[1]); |
| metadata_table->AddItem(kHashedUserId[0], kComponent[1]); |
| ASSERT_EQ(metadata_table->installed_items_.GetList().size(), 3U); |
| EXPECT_TRUE( |
| metadata_table->HasComponentForUser(kHashedUserId[0], kComponent[0])); |
| EXPECT_TRUE( |
| metadata_table->HasComponentForUser(kHashedUserId[1], kComponent[1])); |
| EXPECT_TRUE( |
| metadata_table->HasComponentForUser(kHashedUserId[0], kComponent[1])); |
| EXPECT_FALSE( |
| metadata_table->HasComponentForUser(kHashedUserId[1], kComponent[0])); |
| // Add an existing item. |
| metadata_table->AddItem(kHashedUserId[0], kComponent[0]); |
| EXPECT_EQ(metadata_table->installed_items_.GetList().size(), 3U); |
| } |
| |
| TEST_F(CrOSComponentInstallerMetadataTest, Delete) { |
| // Initialize protobuf object. |
| std::unique_ptr<MetadataTable> metadata_table = |
| MetadataTable::CreateForTest(); |
| |
| // Add component items. |
| metadata_table->AddItem(kHashedUserId[0], kComponent[0]); |
| metadata_table->AddItem(kHashedUserId[1], kComponent[1]); |
| metadata_table->AddItem(kHashedUserId[0], kComponent[1]); |
| |
| // Delete a component item that does not exist. |
| EXPECT_FALSE(metadata_table->DeleteItem(kHashedUserId[1], kComponent[0])); |
| |
| // Delete component items. |
| EXPECT_TRUE(metadata_table->DeleteItem(kHashedUserId[1], kComponent[1])); |
| EXPECT_TRUE(metadata_table->DeleteItem(kHashedUserId[0], kComponent[1])); |
| ASSERT_EQ(metadata_table->installed_items_.GetList().size(), 1U); |
| const base::Value& item = metadata_table->installed_items_.GetList()[0]; |
| EXPECT_EQ(item.FindKey(kMetadataContentItemHashedUserIdKey)->GetString(), |
| kHashedUserId[0]); |
| EXPECT_EQ(item.FindKey(kMetadataContentItemComponentKey)->GetString(), |
| kComponent[0]); |
| } |
| |
| } // namespace component_updater |