Index: components/previews/core/previews_black_list_unittest.cc |
diff --git a/components/previews/core/previews_black_list_unittest.cc b/components/previews/core/previews_black_list_unittest.cc |
index d9fd376183c263b64f159a2df011c60cb69283f5..507491da841627e7cd3cfc950e1d0e6b0ab411f4 100644 |
--- a/components/previews/core/previews_black_list_unittest.cc |
+++ b/components/previews/core/previews_black_list_unittest.cc |
@@ -33,37 +33,45 @@ namespace previews { |
namespace { |
void RunLoadCallback(LoadBlackListCallback callback, |
std::unique_ptr<BlackListItemMap> black_list_item_map) { |
callback.Run(std::move(black_list_item_map)); |
} |
class TestPreviewsOptOutStore : public PreviewsOptOutStore { |
public: |
- TestPreviewsOptOutStore() {} |
+ TestPreviewsOptOutStore() : clear_blacklist_count_(0) {} |
~TestPreviewsOptOutStore() override {} |
+ int clear_blacklist_count() { return clear_blacklist_count_; } |
+ |
private: |
// PreviewsOptOutStore implementation: |
void AddPreviewNavigation(bool opt_out, |
const std::string& host_name, |
PreviewsType type, |
base::Time now) override {} |
void LoadBlackList(LoadBlackListCallback callback) override { |
std::unique_ptr<BlackListItemMap> black_list_item_map( |
new BlackListItemMap()); |
base::MessageLoop::current()->task_runner()->PostTask( |
FROM_HERE, base::Bind(&RunLoadCallback, callback, |
base::Passed(&black_list_item_map))); |
} |
+ |
+ void ClearBlackList(base::Time begin_time, base::Time end_time) override { |
+ ++clear_blacklist_count_; |
+ } |
+ |
+ int clear_blacklist_count_; |
}; |
} // namespace |
TEST_F(PreviewsBlackListTest, BlackListNoStore) { |
// Tests the black list behavior when a null OptOutSture is passed in. |
const GURL url_a("http://www.url_a.com"); |
const GURL url_b("http://www.url_b.com"); |
const size_t history = 4; |
const int threshold = 2; |
@@ -71,21 +79,23 @@ TEST_F(PreviewsBlackListTest, BlackListNoStore) { |
base::FieldTrialList field_trial_list(nullptr); |
std::map<std::string, std::string> params; |
params["stored_history_length"] = base::SizeTToString(history); |
params["opt_out_threshold"] = base::IntToString(threshold); |
params["black_list_duration_in_days"] = base::IntToString(duration_in_days); |
ASSERT_TRUE( |
variations::AssociateVariationParams("ClientSidePreviews", "Enabled", |
params) && |
base::FieldTrialList::CreateFieldTrial("ClientSidePreviews", "Enabled")); |
- base::Clock* test_clock = new base::SimpleTestClock(); |
+ base::SimpleTestClock* test_clock = new base::SimpleTestClock(); |
+ base::Time start = test_clock->Now(); |
+ test_clock->Advance(base::TimeDelta::FromSeconds(1)); |
std::unique_ptr<PreviewsBlackList> black_list( |
new PreviewsBlackList(nullptr, base::WrapUnique(test_clock))); |
EXPECT_TRUE(black_list->IsLoadedAndAllowed(url_a, PreviewsType::OFFLINE)); |
EXPECT_TRUE(black_list->IsLoadedAndAllowed(url_b, PreviewsType::OFFLINE)); |
black_list->AddPreviewNavigation(url_a, true, PreviewsType::OFFLINE); |
black_list->AddPreviewNavigation(url_a, true, PreviewsType::OFFLINE); |
@@ -98,20 +108,26 @@ TEST_F(PreviewsBlackListTest, BlackListNoStore) { |
EXPECT_FALSE(black_list->IsLoadedAndAllowed(url_a, PreviewsType::OFFLINE)); |
EXPECT_FALSE(black_list->IsLoadedAndAllowed(url_b, PreviewsType::OFFLINE)); |
black_list->AddPreviewNavigation(url_b, false, PreviewsType::OFFLINE); |
black_list->AddPreviewNavigation(url_b, false, PreviewsType::OFFLINE); |
black_list->AddPreviewNavigation(url_b, false, PreviewsType::OFFLINE); |
EXPECT_FALSE(black_list->IsLoadedAndAllowed(url_a, PreviewsType::OFFLINE)); |
EXPECT_TRUE(black_list->IsLoadedAndAllowed(url_b, PreviewsType::OFFLINE)); |
+ test_clock->Advance(base::TimeDelta::FromSeconds(1)); |
+ black_list->ClearBlackList(start, test_clock->Now()); |
+ |
+ EXPECT_TRUE(black_list->IsLoadedAndAllowed(url_a, PreviewsType::OFFLINE)); |
+ EXPECT_TRUE(black_list->IsLoadedAndAllowed(url_b, PreviewsType::OFFLINE)); |
+ |
variations::testing::ClearAllVariationParams(); |
} |
TEST_F(PreviewsBlackListTest, BlackListWithStore) { |
// Tests the black list behavior when a non-null OptOutSture is passed in. |
const GURL url_a1("http://www.url_a.com/a1"); |
const GURL url_a2("http://www.url_a.com/a2"); |
const GURL url_b("http://www.url_b.com"); |
const size_t history = 4; |
const int threshold = 2; |
@@ -121,25 +137,28 @@ TEST_F(PreviewsBlackListTest, BlackListWithStore) { |
params["stored_history_length"] = base::SizeTToString(history); |
params["opt_out_threshold"] = base::IntToString(threshold); |
params["black_list_duration_in_days"] = base::IntToString(duration_in_days); |
ASSERT_TRUE( |
variations::AssociateVariationParams("ClientSidePreviews", "Enabled", |
params) && |
base::FieldTrialList::CreateFieldTrial("ClientSidePreviews", "Enabled")); |
base::MessageLoop loop; |
- base::Clock* test_clock = new base::SimpleTestClock(); |
+ base::SimpleTestClock* test_clock = new base::SimpleTestClock(); |
+ base::Time start = test_clock->Now(); |
+ test_clock->Advance(base::TimeDelta::FromSeconds(1)); |
+ |
+ TestPreviewsOptOutStore* opt_out_store = new TestPreviewsOptOutStore(); |
- std::unique_ptr<PreviewsBlackList> black_list( |
- new PreviewsBlackList(base::MakeUnique<TestPreviewsOptOutStore>(), |
- base::WrapUnique(test_clock))); |
+ std::unique_ptr<PreviewsBlackList> black_list(new PreviewsBlackList( |
+ base::WrapUnique(opt_out_store), base::WrapUnique(test_clock))); |
EXPECT_FALSE(black_list->IsLoadedAndAllowed(url_a1, PreviewsType::OFFLINE)); |
EXPECT_FALSE(black_list->IsLoadedAndAllowed(url_a2, PreviewsType::OFFLINE)); |
EXPECT_FALSE(black_list->IsLoadedAndAllowed(url_b, PreviewsType::OFFLINE)); |
base::RunLoop().RunUntilIdle(); |
EXPECT_TRUE(black_list->IsLoadedAndAllowed(url_a1, PreviewsType::OFFLINE)); |
EXPECT_TRUE(black_list->IsLoadedAndAllowed(url_a2, PreviewsType::OFFLINE)); |
EXPECT_TRUE(black_list->IsLoadedAndAllowed(url_b, PreviewsType::OFFLINE)); |
@@ -159,54 +178,90 @@ TEST_F(PreviewsBlackListTest, BlackListWithStore) { |
EXPECT_FALSE(black_list->IsLoadedAndAllowed(url_b, PreviewsType::OFFLINE)); |
black_list->AddPreviewNavigation(url_b, false, PreviewsType::OFFLINE); |
black_list->AddPreviewNavigation(url_b, false, PreviewsType::OFFLINE); |
black_list->AddPreviewNavigation(url_b, false, PreviewsType::OFFLINE); |
EXPECT_FALSE(black_list->IsLoadedAndAllowed(url_a1, PreviewsType::OFFLINE)); |
EXPECT_FALSE(black_list->IsLoadedAndAllowed(url_a2, PreviewsType::OFFLINE)); |
EXPECT_TRUE(black_list->IsLoadedAndAllowed(url_b, PreviewsType::OFFLINE)); |
+ test_clock->Advance(base::TimeDelta::FromSeconds(1)); |
+ EXPECT_EQ(0, opt_out_store->clear_blacklist_count()); |
+ black_list->ClearBlackList(start, base::Time::Now()); |
+ EXPECT_EQ(1, opt_out_store->clear_blacklist_count()); |
+ |
+ EXPECT_FALSE(black_list->IsLoadedAndAllowed(url_a1, PreviewsType::OFFLINE)); |
+ EXPECT_FALSE(black_list->IsLoadedAndAllowed(url_a2, PreviewsType::OFFLINE)); |
+ EXPECT_FALSE(black_list->IsLoadedAndAllowed(url_b, PreviewsType::OFFLINE)); |
+ |
+ base::RunLoop().RunUntilIdle(); |
+ EXPECT_EQ(1, opt_out_store->clear_blacklist_count()); |
+ |
+ EXPECT_TRUE(black_list->IsLoadedAndAllowed(url_a1, PreviewsType::OFFLINE)); |
+ EXPECT_TRUE(black_list->IsLoadedAndAllowed(url_a1, PreviewsType::OFFLINE)); |
+ EXPECT_TRUE(black_list->IsLoadedAndAllowed(url_b, PreviewsType::OFFLINE)); |
+ |
variations::testing::ClearAllVariationParams(); |
} |
TEST_F(PreviewsBlackListTest, QueueBehavior) { |
// Tests the black list asynchronous queue behavior. Methods called while |
// loading are queued and should run in the order they were queued. |
const GURL url("http://www.url.com"); |
+ const GURL url2("http://www.url2.com"); |
const int duration_in_days = 365; |
base::FieldTrialList field_trial_list(nullptr); |
std::map<std::string, std::string> params; |
params["black_list_duration_in_days"] = base::IntToString(duration_in_days); |
ASSERT_TRUE( |
variations::AssociateVariationParams("ClientSidePreviews", "Enabled", |
params) && |
base::FieldTrialList::CreateFieldTrial("ClientSidePreviews", "Enabled")); |
base::MessageLoop loop; |
std::vector<bool> test_opt_out{true, false}; |
for (auto opt_out : test_opt_out) { |
- base::Clock* test_clock = new base::SimpleTestClock(); |
+ base::SimpleTestClock* test_clock = new base::SimpleTestClock(); |
+ |
+ TestPreviewsOptOutStore* opt_out_store = new TestPreviewsOptOutStore(); |
- std::unique_ptr<PreviewsBlackList> black_list( |
- new PreviewsBlackList(base::MakeUnique<TestPreviewsOptOutStore>(), |
- base::WrapUnique(test_clock))); |
+ std::unique_ptr<PreviewsBlackList> black_list(new PreviewsBlackList( |
+ base::WrapUnique(opt_out_store), base::WrapUnique(test_clock))); |
EXPECT_FALSE(black_list->IsLoadedAndAllowed(url, PreviewsType::OFFLINE)); |
black_list->AddPreviewNavigation(url, opt_out, PreviewsType::OFFLINE); |
black_list->AddPreviewNavigation(url, opt_out, PreviewsType::OFFLINE); |
EXPECT_FALSE(black_list->IsLoadedAndAllowed(url, PreviewsType::OFFLINE)); |
base::RunLoop().RunUntilIdle(); |
EXPECT_EQ(!opt_out, |
black_list->IsLoadedAndAllowed(url, PreviewsType::OFFLINE)); |
+ |
+ base::Time start = test_clock->Now(); |
+ test_clock->Advance(base::TimeDelta::FromSeconds(1)); |
+ black_list->AddPreviewNavigation(url, opt_out, PreviewsType::OFFLINE); |
+ black_list->AddPreviewNavigation(url, opt_out, PreviewsType::OFFLINE); |
+ test_clock->Advance(base::TimeDelta::FromSeconds(1)); |
+ EXPECT_EQ(0, opt_out_store->clear_blacklist_count()); |
+ black_list->ClearBlackList( |
+ start, test_clock->Now() + base::TimeDelta::FromSeconds(1)); |
+ EXPECT_EQ(1, opt_out_store->clear_blacklist_count()); |
+ black_list->AddPreviewNavigation(url2, opt_out, PreviewsType::OFFLINE); |
+ black_list->AddPreviewNavigation(url2, opt_out, PreviewsType::OFFLINE); |
+ base::RunLoop().RunUntilIdle(); |
+ EXPECT_EQ(1, opt_out_store->clear_blacklist_count()); |
+ |
+ EXPECT_TRUE(black_list->IsLoadedAndAllowed(url, PreviewsType::OFFLINE)); |
+ EXPECT_EQ(!opt_out, |
+ black_list->IsLoadedAndAllowed(url2, PreviewsType::OFFLINE)); |
} |
variations::testing::ClearAllVariationParams(); |
} |
TEST_F(PreviewsBlackListTest, MaxHosts) { |
// Test that the black list only stores n hosts, and it stores the correct n |
// hosts. |
const GURL url_a("http://www.url_a.com"); |
const GURL url_b("http://www.url_b.com"); |