Supervised user SafeSites: don't cache results for more than one hour

chrome/browser/supervised_user/experimental/supervised_user_async_url_checker_unittest.cc

 // Copyright 2014 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 <string>
 
 #include "base/callback.h"
 #include "base/json/json_writer.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop/message_loop.h"
@@ skipping 95 matching lines @@
   size_t next_url_;
   base::MessageLoop message_loop_;
   scoped_refptr<net::TestURLRequestContextGetter> request_context_;
   net::TestURLFetcherFactory url_fetcher_factory_;
   SupervisedUserAsyncURLChecker checker_;
 };
 
 TEST_F(SupervisedUserAsyncURLCheckerTest, Simple) {
   {
     GURL url(GetNewURL());
-    EXPECT_FALSE(CheckURL(url));
+    ASSERT_FALSE(CheckURL(url));
     EXPECT_CALL(*this, OnCheckDone(url, SupervisedUserURLFilter::ALLOW, false));
     SendValidResponse(false);
   }
   {
     GURL url(GetNewURL());
-    EXPECT_FALSE(CheckURL(url));
+    ASSERT_FALSE(CheckURL(url));
     EXPECT_CALL(*this, OnCheckDone(url, SupervisedUserURLFilter::BLOCK, false));
     SendValidResponse(true);
   }
   {
     GURL url(GetNewURL());
-    EXPECT_FALSE(CheckURL(url));
+    ASSERT_FALSE(CheckURL(url));
     EXPECT_CALL(*this, OnCheckDone(url, SupervisedUserURLFilter::ALLOW, true));
     SendFailedResponse();
   }
 }
 
 TEST_F(SupervisedUserAsyncURLCheckerTest, Equivalence) {
   // Leading "www." in the response should be ignored.
   {
     GURL url("http://example.com");
     GURL url_response("http://www.example.com");
-    EXPECT_FALSE(CheckURL(url));
+    ASSERT_FALSE(CheckURL(url));
     EXPECT_CALL(*this, OnCheckDone(url, SupervisedUserURLFilter::ALLOW, false));
     SendValidResponse(false);
   }
   // Scheme should be ignored.
   {
     GURL url("http://www.example2.com");
     GURL url_response("https://www.example2.com");
-    EXPECT_FALSE(CheckURL(url));
+    ASSERT_FALSE(CheckURL(url));
     EXPECT_CALL(*this, OnCheckDone(url, SupervisedUserURLFilter::ALLOW, false));
     SendValidResponse(false);
   }
   // Both at the same time should work as well.
   {
     GURL url("http://example3.com");
     GURL url_response("https://www.example3.com");
-    EXPECT_FALSE(CheckURL(url));
+    ASSERT_FALSE(CheckURL(url));
     EXPECT_CALL(*this, OnCheckDone(url, SupervisedUserURLFilter::ALLOW, false));
     SendValidResponse(false);
   }
 }
 
 TEST_F(SupervisedUserAsyncURLCheckerTest, Cache) {
   // One more URL than fit in the cache.
   ASSERT_EQ(2u, kCacheSize);
   GURL url1(GetNewURL());
   GURL url2(GetNewURL());
   GURL url3(GetNewURL());
 
   // Populate the cache.
-  EXPECT_FALSE(CheckURL(url1));
+  ASSERT_FALSE(CheckURL(url1));
   EXPECT_CALL(*this, OnCheckDone(url1, SupervisedUserURLFilter::ALLOW, false));
   SendValidResponse(false);
-  EXPECT_FALSE(CheckURL(url2));
+  ASSERT_FALSE(CheckURL(url2));
   EXPECT_CALL(*this, OnCheckDone(url2, SupervisedUserURLFilter::ALLOW, false));
   SendValidResponse(false);
 
   // Now we should get results synchronously.
   EXPECT_CALL(*this, OnCheckDone(url2, SupervisedUserURLFilter::ALLOW, false));
-  EXPECT_TRUE(CheckURL(url2));
+  ASSERT_TRUE(CheckURL(url2));
   EXPECT_CALL(*this, OnCheckDone(url1, SupervisedUserURLFilter::ALLOW, false));
-  EXPECT_TRUE(CheckURL(url1));
+  ASSERT_TRUE(CheckURL(url1));
 
   // Now |url2| is the LRU and should be evicted on the next check.
-  EXPECT_FALSE(CheckURL(url3));
+  ASSERT_FALSE(CheckURL(url3));
   EXPECT_CALL(*this, OnCheckDone(url3, SupervisedUserURLFilter::ALLOW, false));
   SendValidResponse(false);
 
-  EXPECT_FALSE(CheckURL(url2));
+  ASSERT_FALSE(CheckURL(url2));
   EXPECT_CALL(*this, OnCheckDone(url2, SupervisedUserURLFilter::ALLOW, false));
   SendValidResponse(false);
 }
 
 TEST_F(SupervisedUserAsyncURLCheckerTest, CoalesceRequestsToSameURL) {
   GURL url(GetNewURL());
   // Start two checks for the same URL.
-  EXPECT_FALSE(CheckURL(url));
-  EXPECT_FALSE(CheckURL(url));
+  ASSERT_FALSE(CheckURL(url));
+  ASSERT_FALSE(CheckURL(url));
   // A single response should answer both checks.
   EXPECT_CALL(*this, OnCheckDone(url, SupervisedUserURLFilter::ALLOW, false))
       .Times(2);
   SendValidResponse(false);
 }
+
+TEST_F(SupervisedUserAsyncURLCheckerTest, CacheTimeout) {
+  GURL url(GetNewURL());
+
+  checker_.SetCacheTimeoutForTesting(base::TimeDelta::FromSeconds(0));
+
+  ASSERT_FALSE(CheckURL(url));
+  EXPECT_CALL(*this, OnCheckDone(url, SupervisedUserURLFilter::ALLOW, false));
+  SendValidResponse(false);
+
+  // Since the cache timeout is zero, the cache entry should be invalidated
+  // immediately.
+  ASSERT_FALSE(CheckURL(url));
+  EXPECT_CALL(*this, OnCheckDone(url, SupervisedUserURLFilter::BLOCK, false));
+  SendValidResponse(true);
+}