| Index: components/offline_pages/offline_page_metadata_store_impl_unittest.cc
|
| diff --git a/components/offline_pages/offline_page_metadata_store_impl_unittest.cc b/components/offline_pages/offline_page_metadata_store_impl_unittest.cc
|
| index 376c87cd59445f6f2e84ee4d5ddbbc6b1a7401a5..2e536739422ed44cd29113921352980bf0c4098e 100644
|
| --- a/components/offline_pages/offline_page_metadata_store_impl_unittest.cc
|
| +++ b/components/offline_pages/offline_page_metadata_store_impl_unittest.cc
|
| @@ -210,7 +210,7 @@ class OfflinePageMetadataStoreFactory {
|
| }
|
| };
|
|
|
| -enum CalledCallback { NONE, LOAD, ADD, REMOVE, RESET };
|
| +enum CalledCallback { NONE, LOAD, ADD, UPDATE, REMOVE, RESET };
|
| enum Status { STATUS_NONE, STATUS_TRUE, STATUS_FALSE };
|
|
|
| class OfflinePageMetadataStoreTest : public testing::Test {
|
| @@ -233,6 +233,7 @@ class OfflinePageMetadataStoreTest : public testing::Test {
|
| void GetOfflinePagesCallback(
|
| OfflinePageMetadataStore::LoadStatus load_status,
|
| const std::vector<OfflinePageItem>& offline_pages);
|
| + void AddCallback(OfflinePageMetadataStore::ItemActionStatus status);
|
| void UpdateCallback(CalledCallback called_callback, bool success);
|
|
|
| void ClearResults();
|
| @@ -275,6 +276,15 @@ void OfflinePageMetadataStoreTest::GetOfflinePagesCallback(
|
| offline_pages_.swap(const_cast<std::vector<OfflinePageItem>&>(offline_pages));
|
| }
|
|
|
| +void OfflinePageMetadataStoreTest::AddCallback(
|
| + OfflinePageMetadataStore::ItemActionStatus status) {
|
| + last_called_callback_ = ADD;
|
| + // TODO(fgorski): Add specific add status.
|
| + // last_item_status_ = status;
|
| + last_status_ =
|
| + status == OfflinePageMetadataStore::SUCCESS ? STATUS_TRUE : STATUS_FALSE;
|
| +}
|
| +
|
| void OfflinePageMetadataStoreTest::UpdateCallback(
|
| CalledCallback called_callback,
|
| bool status) {
|
| @@ -305,9 +315,9 @@ void OfflinePageMetadataStoreTest::CheckThatOfflinePageCanBeSaved(
|
| base::Time expiration_time = base::Time::Now();
|
| offline_page.expiration_time = expiration_time;
|
|
|
| - store->AddOrUpdateOfflinePage(
|
| - offline_page, base::Bind(&OfflinePageMetadataStoreTest::UpdateCallback,
|
| - base::Unretained(this), ADD));
|
| + store->AddOfflinePage(offline_page,
|
| + base::Bind(&OfflinePageMetadataStoreTest::AddCallback,
|
| + base::Unretained(this)));
|
| PumpLoop();
|
| EXPECT_EQ(ADD, last_called_callback_);
|
| EXPECT_EQ(STATUS_TRUE, last_status_);
|
| @@ -445,9 +455,9 @@ TEST_F(OfflinePageMetadataStoreTest, RemoveOfflinePage) {
|
| // Add an offline page.
|
| OfflinePageItem offline_page(GURL(kTestURL), 1234LL, kTestClientId1,
|
| base::FilePath(kFilePath), kFileSize);
|
| - store->AddOrUpdateOfflinePage(
|
| - offline_page, base::Bind(&OfflinePageMetadataStoreTest::UpdateCallback,
|
| - base::Unretained(this), ADD));
|
| + store->AddOfflinePage(offline_page,
|
| + base::Bind(&OfflinePageMetadataStoreTest::AddCallback,
|
| + base::Unretained(this)));
|
| PumpLoop();
|
| EXPECT_EQ(ADD, last_called_callback_);
|
| EXPECT_EQ(STATUS_TRUE, last_status_);
|
| @@ -505,9 +515,9 @@ TEST_F(OfflinePageMetadataStoreTest, AddRemoveMultipleOfflinePages) {
|
| kTestClientId2, file_path_2, 12345,
|
| base::Time::Now());
|
| offline_page_2.expiration_time = base::Time::Now();
|
| - store->AddOrUpdateOfflinePage(
|
| - offline_page_1, base::Bind(&OfflinePageMetadataStoreTest::UpdateCallback,
|
| - base::Unretained(this), ADD));
|
| + store->AddOfflinePage(offline_page_1,
|
| + base::Bind(&OfflinePageMetadataStoreTest::AddCallback,
|
| + base::Unretained(this)));
|
| PumpLoop();
|
| EXPECT_EQ(ADD, last_called_callback_);
|
| EXPECT_EQ(STATUS_TRUE, last_status_);
|
| @@ -515,9 +525,9 @@ TEST_F(OfflinePageMetadataStoreTest, AddRemoveMultipleOfflinePages) {
|
| ClearResults();
|
|
|
| // Add anther offline page.
|
| - store->AddOrUpdateOfflinePage(
|
| - offline_page_2, base::Bind(&OfflinePageMetadataStoreTest::UpdateCallback,
|
| - base::Unretained(this), ADD));
|
| + store->AddOfflinePage(offline_page_2,
|
| + base::Bind(&OfflinePageMetadataStoreTest::AddCallback,
|
| + base::Unretained(this)));
|
| PumpLoop();
|
| EXPECT_EQ(ADD, last_called_callback_);
|
| EXPECT_EQ(STATUS_TRUE, last_status_);
|
| @@ -576,9 +586,9 @@ TEST_F(OfflinePageMetadataStoreTest, UpdateOfflinePage) {
|
| // First, adds a fresh page.
|
| OfflinePageItem offline_page(GURL(kTestURL), 1234LL, kTestClientId1,
|
| base::FilePath(kFilePath), kFileSize);
|
| - store->AddOrUpdateOfflinePage(
|
| - offline_page, base::Bind(&OfflinePageMetadataStoreTest::UpdateCallback,
|
| - base::Unretained(this), ADD));
|
| + store->AddOfflinePage(offline_page,
|
| + base::Bind(&OfflinePageMetadataStoreTest::AddCallback,
|
| + base::Unretained(this)));
|
| PumpLoop();
|
| EXPECT_EQ(ADD, last_called_callback_);
|
| EXPECT_EQ(STATUS_TRUE, last_status_);
|
| @@ -606,11 +616,13 @@ TEST_F(OfflinePageMetadataStoreTest, UpdateOfflinePage) {
|
| offline_page.file_size = kFileSize + 1;
|
| offline_page.access_count++;
|
| offline_page.expiration_time = base::Time::Now();
|
| - store->AddOrUpdateOfflinePage(
|
| - offline_page, base::Bind(&OfflinePageMetadataStoreTest::UpdateCallback,
|
| - base::Unretained(this), ADD));
|
| + std::vector<OfflinePageItem> items_to_update;
|
| + items_to_update.push_back(offline_page);
|
| + store->UpdateOfflinePages(
|
| + items_to_update, base::Bind(&OfflinePageMetadataStoreTest::UpdateCallback,
|
| + base::Unretained(this), UPDATE));
|
| PumpLoop();
|
| - EXPECT_EQ(ADD, last_called_callback_);
|
| + EXPECT_EQ(UPDATE, last_called_callback_);
|
| EXPECT_EQ(STATUS_TRUE, last_status_);
|
|
|
| ClearResults();
|
| @@ -639,9 +651,9 @@ TEST_F(OfflinePageMetadataStoreTest, ClearAllOfflinePages) {
|
| // Add 2 offline pages.
|
| OfflinePageItem offline_page(GURL(kTestURL), 1234LL, kTestClientId1,
|
| base::FilePath(kFilePath), kFileSize);
|
| - store->AddOrUpdateOfflinePage(
|
| - offline_page, base::Bind(&OfflinePageMetadataStoreTest::UpdateCallback,
|
| - base::Unretained(this), ADD));
|
| + store->AddOfflinePage(offline_page,
|
| + base::Bind(&OfflinePageMetadataStoreTest::AddCallback,
|
| + base::Unretained(this)));
|
| PumpLoop();
|
| EXPECT_EQ(ADD, last_called_callback_);
|
| EXPECT_EQ(STATUS_TRUE, last_status_);
|
| @@ -650,9 +662,9 @@ TEST_F(OfflinePageMetadataStoreTest, ClearAllOfflinePages) {
|
|
|
| OfflinePageItem offline_page2(GURL("http://test.com"), 5678LL, kTestClientId2,
|
| base::FilePath(kFilePath), kFileSize);
|
| - store->AddOrUpdateOfflinePage(
|
| - offline_page2, base::Bind(&OfflinePageMetadataStoreTest::UpdateCallback,
|
| - base::Unretained(this), ADD));
|
| + store->AddOfflinePage(offline_page2,
|
| + base::Bind(&OfflinePageMetadataStoreTest::AddCallback,
|
| + base::Unretained(this)));
|
| PumpLoop();
|
| EXPECT_EQ(ADD, last_called_callback_);
|
| EXPECT_EQ(STATUS_TRUE, last_status_);
|
|
|