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 virtual ~BlacklistStateFetcherMock(); |
25 | 25 |
26 virtual void Request(const std::string& id, | 26 virtual void Request(const std::string& id, |
27 const RequestCallback& callback) OVERRIDE; | 27 const RequestCallback& callback) override; |
28 | 28 |
29 void SetState(const std::string& id, BlacklistState state); | 29 void SetState(const std::string& id, BlacklistState state); |
30 | 30 |
31 void Clear(); | 31 void Clear(); |
32 | 32 |
33 int request_count() const { return request_count_; } | 33 int request_count() const { return request_count_; } |
34 | 34 |
35 private: | 35 private: |
36 std::map<std::string, BlacklistState> states_; | 36 std::map<std::string, BlacklistState> states_; |
37 int request_count_; | 37 int request_count_; |
(...skipping 51 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
89 scoped_refptr<FakeSafeBrowsingDatabaseManager> blacklist_db_; | 89 scoped_refptr<FakeSafeBrowsingDatabaseManager> blacklist_db_; |
90 | 90 |
91 Blacklist::ScopedDatabaseManagerForTest scoped_blacklist_db_; | 91 Blacklist::ScopedDatabaseManagerForTest scoped_blacklist_db_; |
92 | 92 |
93 DISALLOW_COPY_AND_ASSIGN(TestBlacklist); | 93 DISALLOW_COPY_AND_ASSIGN(TestBlacklist); |
94 }; | 94 }; |
95 | 95 |
96 } // namespace extensions | 96 } // namespace extensions |
97 | 97 |
98 #endif // CHROME_BROWSER_EXTENSIONS_TEST_BLACKLIST_H_ | 98 #endif // CHROME_BROWSER_EXTENSIONS_TEST_BLACKLIST_H_ |
OLD | NEW |