OLD | NEW |
1 // Copyright 2012 The Chromium Authors. All rights reserved. | 1 // Copyright 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 #ifndef CHROME_BROWSER_EXTENSIONS_TEST_BLACKLIST_H_ | 5 #ifndef CHROME_BROWSER_EXTENSIONS_TEST_BLACKLIST_H_ |
6 #define CHROME_BROWSER_EXTENSIONS_TEST_BLACKLIST_H_ | 6 #define CHROME_BROWSER_EXTENSIONS_TEST_BLACKLIST_H_ |
7 | 7 |
8 #include <map> | 8 #include <map> |
9 #include <string> | 9 #include <string> |
10 | 10 |
11 #include "base/basictypes.h" | 11 #include "base/basictypes.h" |
12 #include "chrome/browser/extensions/blacklist.h" | 12 #include "chrome/browser/extensions/blacklist.h" |
13 #include "chrome/browser/extensions/blacklist_state_fetcher.h" | 13 #include "chrome/browser/extensions/blacklist_state_fetcher.h" |
14 | 14 |
15 namespace extensions { | 15 namespace extensions { |
16 | 16 |
17 class FakeSafeBrowsingDatabaseManager; | 17 class FakeSafeBrowsingDatabaseManager; |
18 | 18 |
19 // Replace BlacklistStateFetcher for testing of the Blacklist class. | 19 // Replace BlacklistStateFetcher for testing of the Blacklist class. |
20 class BlacklistStateFetcherMock : public BlacklistStateFetcher { | 20 class BlacklistStateFetcherMock : public BlacklistStateFetcher { |
21 public: | 21 public: |
22 BlacklistStateFetcherMock(); | 22 BlacklistStateFetcherMock(); |
23 | 23 |
24 virtual ~BlacklistStateFetcherMock(); | 24 ~BlacklistStateFetcherMock() override; |
25 | 25 |
26 virtual void Request(const std::string& id, | 26 void Request(const std::string& id, const RequestCallback& callback) override; |
27 const RequestCallback& callback) override; | |
28 | 27 |
29 void SetState(const std::string& id, BlacklistState state); | 28 void SetState(const std::string& id, BlacklistState state); |
30 | 29 |
31 void Clear(); | 30 void Clear(); |
32 | 31 |
33 int request_count() const { return request_count_; } | 32 int request_count() const { return request_count_; } |
34 | 33 |
35 private: | 34 private: |
36 std::map<std::string, BlacklistState> states_; | 35 std::map<std::string, BlacklistState> states_; |
37 int request_count_; | 36 int request_count_; |
(...skipping 51 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
89 scoped_refptr<FakeSafeBrowsingDatabaseManager> blacklist_db_; | 88 scoped_refptr<FakeSafeBrowsingDatabaseManager> blacklist_db_; |
90 | 89 |
91 Blacklist::ScopedDatabaseManagerForTest scoped_blacklist_db_; | 90 Blacklist::ScopedDatabaseManagerForTest scoped_blacklist_db_; |
92 | 91 |
93 DISALLOW_COPY_AND_ASSIGN(TestBlacklist); | 92 DISALLOW_COPY_AND_ASSIGN(TestBlacklist); |
94 }; | 93 }; |
95 | 94 |
96 } // namespace extensions | 95 } // namespace extensions |
97 | 96 |
98 #endif // CHROME_BROWSER_EXTENSIONS_TEST_BLACKLIST_H_ | 97 #endif // CHROME_BROWSER_EXTENSIONS_TEST_BLACKLIST_H_ |
OLD | NEW |