| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 // | 4 // |
| 5 // This test creates a safebrowsing service using test safebrowsing database | 5 // This test creates a safebrowsing service using test safebrowsing database |
| 6 // and a test protocol manager. It is used to test logics in safebrowsing | 6 // and a test protocol manager. It is used to test logics in safebrowsing |
| 7 // service. | 7 // service. |
| 8 | 8 |
| 9 #include <algorithm> | 9 #include <algorithm> |
| 10 | 10 |
| (...skipping 381 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 392 static void GenUrlFullhashResult(const GURL& url, | 392 static void GenUrlFullhashResult(const GURL& url, |
| 393 int list_id, | 393 int list_id, |
| 394 SBFullHashResult* full_hash) { | 394 SBFullHashResult* full_hash) { |
| 395 std::string host; | 395 std::string host; |
| 396 std::string path; | 396 std::string path; |
| 397 safe_browsing_util::CanonicalizeUrl(url, &host, &path, NULL); | 397 safe_browsing_util::CanonicalizeUrl(url, &host, &path, NULL); |
| 398 full_hash->hash = SBFullHashForString(host + path); | 398 full_hash->hash = SBFullHashForString(host + path); |
| 399 full_hash->list_id = list_id; | 399 full_hash->list_id = list_id; |
| 400 } | 400 } |
| 401 | 401 |
| 402 virtual void SetUp() { | 402 void SetUp() override { |
| 403 // InProcessBrowserTest::SetUp() instantiates SafebrowsingService and | 403 // InProcessBrowserTest::SetUp() instantiates SafebrowsingService and |
| 404 // RegisterFactory has to be called before SafeBrowsingService is created. | 404 // RegisterFactory has to be called before SafeBrowsingService is created. |
| 405 sb_factory_.reset(new TestSafeBrowsingServiceFactory( | 405 sb_factory_.reset(new TestSafeBrowsingServiceFactory( |
| 406 "https://definatelynotarealdomain/safebrowsing")); | 406 "https://definatelynotarealdomain/safebrowsing")); |
| 407 SafeBrowsingService::RegisterFactory(sb_factory_.get()); | 407 SafeBrowsingService::RegisterFactory(sb_factory_.get()); |
| 408 SafeBrowsingDatabase::RegisterFactory(&db_factory_); | 408 SafeBrowsingDatabase::RegisterFactory(&db_factory_); |
| 409 SafeBrowsingProtocolManager::RegisterFactory(&pm_factory_); | 409 SafeBrowsingProtocolManager::RegisterFactory(&pm_factory_); |
| 410 InProcessBrowserTest::SetUp(); | 410 InProcessBrowserTest::SetUp(); |
| 411 } | 411 } |
| 412 | 412 |
| 413 virtual void TearDown() { | 413 void TearDown() override { |
| 414 InProcessBrowserTest::TearDown(); | 414 InProcessBrowserTest::TearDown(); |
| 415 | 415 |
| 416 // Unregister test factories after InProcessBrowserTest::TearDown | 416 // Unregister test factories after InProcessBrowserTest::TearDown |
| 417 // (which destructs SafeBrowsingService). | 417 // (which destructs SafeBrowsingService). |
| 418 SafeBrowsingDatabase::RegisterFactory(NULL); | 418 SafeBrowsingDatabase::RegisterFactory(NULL); |
| 419 SafeBrowsingProtocolManager::RegisterFactory(NULL); | 419 SafeBrowsingProtocolManager::RegisterFactory(NULL); |
| 420 SafeBrowsingService::RegisterFactory(NULL); | 420 SafeBrowsingService::RegisterFactory(NULL); |
| 421 } | 421 } |
| 422 | 422 |
| 423 virtual void SetUpCommandLine(base::CommandLine* command_line) override { | 423 void SetUpCommandLine(base::CommandLine* command_line) override { |
| 424 // Makes sure the auto update is not triggered during the test. | 424 // Makes sure the auto update is not triggered during the test. |
| 425 // This test will fill up the database using testing prefixes | 425 // This test will fill up the database using testing prefixes |
| 426 // and urls. | 426 // and urls. |
| 427 command_line->AppendSwitch(switches::kSbDisableAutoUpdate); | 427 command_line->AppendSwitch(switches::kSbDisableAutoUpdate); |
| 428 #if defined(OS_CHROMEOS) | 428 #if defined(OS_CHROMEOS) |
| 429 command_line->AppendSwitch( | 429 command_line->AppendSwitch( |
| 430 chromeos::switches::kIgnoreUserProfileMappingForTests); | 430 chromeos::switches::kIgnoreUserProfileMappingForTests); |
| 431 #endif | 431 #endif |
| 432 } | 432 } |
| 433 | 433 |
| 434 void SetUpOnMainThread() override { | 434 void SetUpOnMainThread() override { |
| 435 InProcessBrowserTest::SetUpOnMainThread(); | 435 InProcessBrowserTest::SetUpOnMainThread(); |
| 436 g_browser_process->safe_browsing_service()->ui_manager()->AddObserver( | 436 g_browser_process->safe_browsing_service()->ui_manager()->AddObserver( |
| 437 &observer_); | 437 &observer_); |
| 438 } | 438 } |
| 439 | 439 |
| 440 void TearDownOnMainThread() override { | 440 void TearDownOnMainThread() override { |
| 441 g_browser_process->safe_browsing_service()->ui_manager()->RemoveObserver( | 441 g_browser_process->safe_browsing_service()->ui_manager()->RemoveObserver( |
| 442 &observer_); | 442 &observer_); |
| 443 InProcessBrowserTest::TearDownOnMainThread(); | 443 InProcessBrowserTest::TearDownOnMainThread(); |
| 444 } | 444 } |
| 445 | 445 |
| 446 virtual void SetUpInProcessBrowserTestFixture() { | 446 void SetUpInProcessBrowserTestFixture() override { |
| 447 ASSERT_TRUE(test_server()->Start()); | 447 ASSERT_TRUE(test_server()->Start()); |
| 448 } | 448 } |
| 449 | 449 |
| 450 // This will setup the "url" prefix in database and prepare protocol manager | 450 // This will setup the "url" prefix in database and prepare protocol manager |
| 451 // to respond with |full_hash|, as well as other |full_hash|es previously set | 451 // to respond with |full_hash|, as well as other |full_hash|es previously set |
| 452 // via this call, on GetFullHash requests. | 452 // via this call, on GetFullHash requests. |
| 453 void SetupResponseForUrl(const GURL& url, const SBFullHashResult& full_hash) { | 453 void SetupResponseForUrl(const GURL& url, const SBFullHashResult& full_hash) { |
| 454 std::vector<SBPrefix> prefix_hits; | 454 std::vector<SBPrefix> prefix_hits; |
| 455 prefix_hits.push_back(full_hash.hash.prefix); | 455 prefix_hits.push_back(full_hash.hash.prefix); |
| 456 | 456 |
| (...skipping 806 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1263 content::WindowedNotificationObserver observer( | 1263 content::WindowedNotificationObserver observer( |
| 1264 chrome::NOTIFICATION_SAFE_BROWSING_UPDATE_COMPLETE, | 1264 chrome::NOTIFICATION_SAFE_BROWSING_UPDATE_COMPLETE, |
| 1265 content::Source<SafeBrowsingDatabaseManager>( | 1265 content::Source<SafeBrowsingDatabaseManager>( |
| 1266 sb_service_->database_manager().get())); | 1266 sb_service_->database_manager().get())); |
| 1267 BrowserThread::PostTask( | 1267 BrowserThread::PostTask( |
| 1268 BrowserThread::IO, | 1268 BrowserThread::IO, |
| 1269 FROM_HERE, | 1269 FROM_HERE, |
| 1270 base::Bind(&SafeBrowsingDatabaseManagerCookieTest::ForceUpdate, this)); | 1270 base::Bind(&SafeBrowsingDatabaseManagerCookieTest::ForceUpdate, this)); |
| 1271 observer.Wait(); | 1271 observer.Wait(); |
| 1272 } | 1272 } |
| OLD | NEW |