| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 "base/command_line.h" | 9 #include "base/command_line.h" |
| 10 #include "base/memory/ref_counted.h" | 10 #include "base/memory/ref_counted.h" |
| (...skipping 66 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 77 return false; | 77 return false; |
| 78 DCHECK_LE(1U, prefix_hits->size()); | 78 DCHECK_LE(1U, prefix_hits->size()); |
| 79 return true; | 79 return true; |
| 80 } | 80 } |
| 81 virtual bool ContainsDownloadHashPrefix(const SBPrefix& prefix) { | 81 virtual bool ContainsDownloadHashPrefix(const SBPrefix& prefix) { |
| 82 return download_digest_prefix_.count(prefix) > 0; | 82 return download_digest_prefix_.count(prefix) > 0; |
| 83 } | 83 } |
| 84 virtual bool ContainsCsdWhitelistedUrl(const GURL& url) { | 84 virtual bool ContainsCsdWhitelistedUrl(const GURL& url) { |
| 85 return true; | 85 return true; |
| 86 } | 86 } |
| 87 virtual bool ContainsDownloadWhitelistedString(const std::string& str) { |
| 88 return true; |
| 89 } |
| 90 virtual bool ContainsDownloadWhitelistedUrl(const GURL& url) { |
| 91 return true; |
| 92 } |
| 87 virtual bool UpdateStarted(std::vector<SBListChunkRanges>* lists) { | 93 virtual bool UpdateStarted(std::vector<SBListChunkRanges>* lists) { |
| 88 ADD_FAILURE() << "Not implemented."; | 94 ADD_FAILURE() << "Not implemented."; |
| 89 return false; | 95 return false; |
| 90 } | 96 } |
| 91 virtual void InsertChunks(const std::string& list_name, | 97 virtual void InsertChunks(const std::string& list_name, |
| 92 const SBChunkList& chunks) { | 98 const SBChunkList& chunks) { |
| 93 ADD_FAILURE() << "Not implemented."; | 99 ADD_FAILURE() << "Not implemented."; |
| 94 } | 100 } |
| 95 virtual void DeleteChunks(const std::vector<SBChunkDelete>& chunk_deletes) { | 101 virtual void DeleteChunks(const std::vector<SBChunkDelete>& chunk_deletes) { |
| 96 ADD_FAILURE() << "Not implemented."; | 102 ADD_FAILURE() << "Not implemented."; |
| (...skipping 63 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 160 }; | 166 }; |
| 161 | 167 |
| 162 // Factory that creates TestSafeBrowsingDatabase instances. | 168 // Factory that creates TestSafeBrowsingDatabase instances. |
| 163 class TestSafeBrowsingDatabaseFactory : public SafeBrowsingDatabaseFactory { | 169 class TestSafeBrowsingDatabaseFactory : public SafeBrowsingDatabaseFactory { |
| 164 public: | 170 public: |
| 165 TestSafeBrowsingDatabaseFactory() : db_(NULL) {} | 171 TestSafeBrowsingDatabaseFactory() : db_(NULL) {} |
| 166 virtual ~TestSafeBrowsingDatabaseFactory() {} | 172 virtual ~TestSafeBrowsingDatabaseFactory() {} |
| 167 | 173 |
| 168 virtual SafeBrowsingDatabase* CreateSafeBrowsingDatabase( | 174 virtual SafeBrowsingDatabase* CreateSafeBrowsingDatabase( |
| 169 bool enable_download_protection, | 175 bool enable_download_protection, |
| 170 bool enable_client_side_whitelist) { | 176 bool enable_client_side_whitelist, |
| 177 bool enable_download_whitelist) { |
| 171 db_ = new TestSafeBrowsingDatabase(); | 178 db_ = new TestSafeBrowsingDatabase(); |
| 172 return db_; | 179 return db_; |
| 173 } | 180 } |
| 174 TestSafeBrowsingDatabase* GetDb() { | 181 TestSafeBrowsingDatabase* GetDb() { |
| 175 return db_; | 182 return db_; |
| 176 } | 183 } |
| 177 private: | 184 private: |
| 178 // Owned by the SafebrowsingService. | 185 // Owned by the SafebrowsingService. |
| 179 TestSafeBrowsingDatabase* db_; | 186 TestSafeBrowsingDatabase* db_; |
| 180 }; | 187 }; |
| (...skipping 576 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 757 | 764 |
| 758 // Delete the Profile. SBS stops again. | 765 // Delete the Profile. SBS stops again. |
| 759 pref_service2 = NULL; | 766 pref_service2 = NULL; |
| 760 profile2.reset(); | 767 profile2.reset(); |
| 761 WaitForIOThread(); | 768 WaitForIOThread(); |
| 762 EXPECT_FALSE(sb_service->enabled()); | 769 EXPECT_FALSE(sb_service->enabled()); |
| 763 EXPECT_FALSE(csd_service->enabled()); | 770 EXPECT_FALSE(csd_service->enabled()); |
| 764 } | 771 } |
| 765 | 772 |
| 766 } // namespace | 773 } // namespace |
| OLD | NEW |