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 |