OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 #ifndef CHROME_BROWSER_EXTENSIONS_FAKE_SAFE_BROWSING_DATABASE_MANAGER_H_ | 5 #ifndef CHROME_BROWSER_EXTENSIONS_FAKE_SAFE_BROWSING_DATABASE_MANAGER_H_ |
6 #define CHROME_BROWSER_EXTENSIONS_FAKE_SAFE_BROWSING_DATABASE_MANAGER_H_ | 6 #define CHROME_BROWSER_EXTENSIONS_FAKE_SAFE_BROWSING_DATABASE_MANAGER_H_ |
7 | 7 |
8 #include <set> | 8 #include <set> |
9 #include <string> | 9 #include <string> |
10 | 10 |
11 #include "chrome/browser/safe_browsing/database_manager.h" | 11 #include "chrome/browser/safe_browsing/local_database_manager.h" |
12 | 12 |
13 namespace extensions { | 13 namespace extensions { |
14 | 14 |
15 // A fake safe browsing database manager for use with extensions tests. | 15 // A fake safe browsing database manager for use with extensions tests. |
16 // | 16 // |
17 // By default it is disabled (returning true and ignoring |unsafe_ids_|); | 17 // By default it is disabled (returning true and ignoring |unsafe_ids_|); |
18 // call set_enabled to enable it. | 18 // call set_enabled to enable it. |
19 class FakeSafeBrowsingDatabaseManager : public SafeBrowsingDatabaseManager { | 19 class FakeSafeBrowsingDatabaseManager |
| 20 : public LocalSafeBrowsingDatabaseManager { |
20 public: | 21 public: |
21 explicit FakeSafeBrowsingDatabaseManager(bool enabled); | 22 explicit FakeSafeBrowsingDatabaseManager(bool enabled); |
22 | 23 |
23 // Returns true if synchronously safe, false if not in which case the unsafe | 24 // Returns true if synchronously safe, false if not in which case the unsafe |
24 // IDs taken from |unsafe_ids_| are passed to to |client| on the current | 25 // IDs taken from |unsafe_ids_| are passed to to |client| on the current |
25 // message loop. | 26 // message loop. |
26 bool CheckExtensionIDs(const std::set<std::string>& extension_ids, | 27 bool CheckExtensionIDs(const std::set<std::string>& extension_ids, |
27 Client* client) override; | 28 Client* client) override; |
28 | 29 |
29 // Return |this| to chain together SetUnsafe(...).NotifyUpdate() conveniently. | 30 // Return |this| to chain together SetUnsafe(...).NotifyUpdate() conveniently. |
(...skipping 12 matching lines...) Expand all Loading... |
42 const std::string& d); | 43 const std::string& d); |
43 FakeSafeBrowsingDatabaseManager& AddUnsafe(const std::string& a); | 44 FakeSafeBrowsingDatabaseManager& AddUnsafe(const std::string& a); |
44 FakeSafeBrowsingDatabaseManager& RemoveUnsafe(const std::string& a); | 45 FakeSafeBrowsingDatabaseManager& RemoveUnsafe(const std::string& a); |
45 | 46 |
46 // Send the update notification. | 47 // Send the update notification. |
47 void NotifyUpdate(); | 48 void NotifyUpdate(); |
48 | 49 |
49 private: | 50 private: |
50 ~FakeSafeBrowsingDatabaseManager() override; | 51 ~FakeSafeBrowsingDatabaseManager() override; |
51 | 52 |
52 // Runs client->SafeBrowsingResult(result). | 53 // Runs result->SafeBrowsingResult(). |
53 void OnSafeBrowsingResult(scoped_ptr<SafeBrowsingCheck> result, | 54 void OnSafeBrowsingResult(scoped_ptr<SafeBrowsingCheck> result); |
54 Client* client); | |
55 | 55 |
56 // Whether to respond to CheckExtensionIDs immediately with true (indicating | 56 // Whether to respond to CheckExtensionIDs immediately with true (indicating |
57 // that there is definitely no extension ID match). | 57 // that there is definitely no extension ID match). |
58 bool enabled_; | 58 bool enabled_; |
59 | 59 |
60 // The extension IDs considered unsafe. | 60 // The extension IDs considered unsafe. |
61 std::set<std::string> unsafe_ids_; | 61 std::set<std::string> unsafe_ids_; |
62 }; | 62 }; |
63 | 63 |
64 } // namespace extensions | 64 } // namespace extensions |
65 | 65 |
66 #endif // CHROME_BROWSER_EXTENSIONS_FAKE_SAFE_BROWSING_DATABASE_MANAGER_H_ | 66 #endif // CHROME_BROWSER_EXTENSIONS_FAKE_SAFE_BROWSING_DATABASE_MANAGER_H_ |
OLD | NEW |