| 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 |