| 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 943 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 954 ASSERT_EQ("a", smt.ColumnString(0)); | 954 ASSERT_EQ("a", smt.ColumnString(0)); |
| 955 ASSERT_EQ("b", smt.ColumnString(1)); | 955 ASSERT_EQ("b", smt.ColumnString(1)); |
| 956 ASSERT_TRUE(smt.Step()); | 956 ASSERT_TRUE(smt.Step()); |
| 957 ASSERT_EQ("c", smt.ColumnString(0)); | 957 ASSERT_EQ("c", smt.ColumnString(0)); |
| 958 ASSERT_EQ("d", smt.ColumnString(1)); | 958 ASSERT_EQ("d", smt.ColumnString(1)); |
| 959 EXPECT_FALSE(smt.Step()); | 959 EXPECT_FALSE(smt.Step()); |
| 960 } | 960 } |
| 961 | 961 |
| 962 virtual void SetUpOnMainThread() OVERRIDE { | 962 virtual void SetUpOnMainThread() OVERRIDE { |
| 963 sb_service_ = g_browser_process->safe_browsing_service(); | 963 sb_service_ = g_browser_process->safe_browsing_service(); |
| 964 ASSERT_TRUE(sb_service_ != NULL); | 964 ASSERT_TRUE(sb_service_.get() != NULL); |
| 965 } | 965 } |
| 966 | 966 |
| 967 virtual void CleanUpOnMainThread() OVERRIDE { | 967 virtual void CleanUpOnMainThread() OVERRIDE { |
| 968 sb_service_ = NULL; | 968 sb_service_ = NULL; |
| 969 } | 969 } |
| 970 | 970 |
| 971 void ForceUpdate() { | 971 void ForceUpdate() { |
| 972 sb_service_->protocol_manager()->ForceScheduleNextUpdate( | 972 sb_service_->protocol_manager()->ForceScheduleNextUpdate( |
| 973 base::TimeDelta::FromSeconds(0)); | 973 base::TimeDelta::FromSeconds(0)); |
| 974 } | 974 } |
| 975 | 975 |
| 976 scoped_refptr<SafeBrowsingService> sb_service_; | 976 scoped_refptr<SafeBrowsingService> sb_service_; |
| 977 | 977 |
| 978 private: | 978 private: |
| 979 DISALLOW_COPY_AND_ASSIGN(SafeBrowsingDatabaseManagerCookieTest); | 979 DISALLOW_COPY_AND_ASSIGN(SafeBrowsingDatabaseManagerCookieTest); |
| 980 }; | 980 }; |
| 981 | 981 |
| 982 // Test that a Safe Browsing database update request both sends cookies and can | 982 // Test that a Safe Browsing database update request both sends cookies and can |
| 983 // save cookies. | 983 // save cookies. |
| 984 IN_PROC_BROWSER_TEST_F(SafeBrowsingDatabaseManagerCookieTest, | 984 IN_PROC_BROWSER_TEST_F(SafeBrowsingDatabaseManagerCookieTest, |
| 985 TestSBUpdateCookies) { | 985 TestSBUpdateCookies) { |
| 986 content::WindowedNotificationObserver observer( | 986 content::WindowedNotificationObserver observer( |
| 987 chrome::NOTIFICATION_SAFE_BROWSING_UPDATE_COMPLETE, | 987 chrome::NOTIFICATION_SAFE_BROWSING_UPDATE_COMPLETE, |
| 988 content::Source<SafeBrowsingDatabaseManager>( | 988 content::Source<SafeBrowsingDatabaseManager>( |
| 989 sb_service_->database_manager())); | 989 sb_service_->database_manager().get())); |
| 990 BrowserThread::PostTask( | 990 BrowserThread::PostTask( |
| 991 BrowserThread::IO, FROM_HERE, | 991 BrowserThread::IO, |
| 992 FROM_HERE, |
| 992 base::Bind(&SafeBrowsingDatabaseManagerCookieTest::ForceUpdate, this)); | 993 base::Bind(&SafeBrowsingDatabaseManagerCookieTest::ForceUpdate, this)); |
| 993 observer.Wait(); | 994 observer.Wait(); |
| 994 } | 995 } |
| OLD | NEW |