Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(338)

Unified Diff: components/subresource_filter/content/common/document_subresource_filter_unittest.cc

Issue 2677223002: Distinguish between subresource filtering and dryrun matching. (Closed)
Patch Set: rebase Created 3 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: components/subresource_filter/content/common/document_subresource_filter_unittest.cc
diff --git a/components/subresource_filter/content/common/document_subresource_filter_unittest.cc b/components/subresource_filter/content/common/document_subresource_filter_unittest.cc
index 3fd55630f8b9ca2094fcc9395940ff305b098320..b7550af61c190544d2e72aad3f857398e573192a 100644
--- a/components/subresource_filter/content/common/document_subresource_filter_unittest.cc
+++ b/components/subresource_filter/content/common/document_subresource_filter_unittest.cc
@@ -92,9 +92,12 @@ TEST_F(DocumentSubresourceFilterTest, DryRun) {
url::Origin(), activation_state, ruleset(),
first_disallowed_load_callback_receiver.closure());
- EXPECT_TRUE(filter.allowLoad(GURL(kTestAlphaURL), request_context));
- EXPECT_TRUE(filter.allowLoad(GURL(kTestAlphaDataURI), request_context));
- EXPECT_TRUE(filter.allowLoad(GURL(kTestBetaURL), request_context));
+ EXPECT_EQ(blink::WebDocumentSubresourceFilter::WouldDisallow,
+ filter.getLoadPolicy(GURL(kTestAlphaURL), request_context));
+ EXPECT_EQ(blink::WebDocumentSubresourceFilter::Allow,
+ filter.getLoadPolicy(GURL(kTestAlphaDataURI), request_context));
+ EXPECT_EQ(blink::WebDocumentSubresourceFilter::Allow,
+ filter.getLoadPolicy(GURL(kTestBetaURL), request_context));
const auto& statistics = filter.statistics();
EXPECT_EQ(3, statistics.num_loads_total);
@@ -109,15 +112,16 @@ TEST_F(DocumentSubresourceFilterTest, Enabled) {
auto test_impl = [this](bool measure_performance) {
blink::WebURLRequest::RequestContext request_context =
blink::WebURLRequest::RequestContextImage;
-
ActivationState activation_state(kEnabled);
activation_state.measure_performance = measure_performance;
DocumentSubresourceFilter filter(url::Origin(), activation_state, ruleset(),
base::OnceClosure());
-
- EXPECT_FALSE(filter.allowLoad(GURL(kTestAlphaURL), request_context));
- EXPECT_TRUE(filter.allowLoad(GURL(kTestAlphaDataURI), request_context));
- EXPECT_TRUE(filter.allowLoad(GURL(kTestBetaURL), request_context));
+ EXPECT_EQ(blink::WebDocumentSubresourceFilter::Disallow,
+ filter.getLoadPolicy(GURL(kTestAlphaURL), request_context));
+ EXPECT_EQ(blink::WebDocumentSubresourceFilter::Allow,
+ filter.getLoadPolicy(GURL(kTestAlphaDataURI), request_context));
+ EXPECT_EQ(blink::WebDocumentSubresourceFilter::Allow,
+ filter.getLoadPolicy(GURL(kTestBetaURL), request_context));
const auto& statistics = filter.statistics();
EXPECT_EQ(3, statistics.num_loads_total);
@@ -139,8 +143,6 @@ TEST_F(DocumentSubresourceFilterTest, Enabled) {
TEST_F(DocumentSubresourceFilterTest,
CallbackFiredExactlyOnceAfterFirstDisallowedLoad) {
- blink::WebURLRequest::RequestContext request_context =
- blink::WebURLRequest::RequestContextImage;
TestCallbackReceiver first_disallowed_load_callback_receiver;
ActivationState activation_state(kEnabled);
@@ -149,11 +151,10 @@ TEST_F(DocumentSubresourceFilterTest,
url::Origin(), activation_state, ruleset(),
first_disallowed_load_callback_receiver.closure());
- EXPECT_TRUE(filter.allowLoad(GURL(kTestAlphaDataURI), request_context));
EXPECT_EQ(0u, first_disallowed_load_callback_receiver.callback_count());
- EXPECT_FALSE(filter.allowLoad(GURL(kTestAlphaURL), request_context));
+ filter.reportDisallowedLoad();
EXPECT_EQ(1u, first_disallowed_load_callback_receiver.callback_count());
- EXPECT_FALSE(filter.allowLoad(GURL(kTestAlphaURL), request_context));
+ filter.reportDisallowedLoad();
EXPECT_EQ(1u, first_disallowed_load_callback_receiver.callback_count());
}

Powered by Google App Engine
This is Rietveld 408576698