Manage all the testing classes for Blacklist in TestBlacklist.

chrome/browser/extensions/test_blacklist.h

Index: chrome/browser/extensions/test_blacklist.h
diff --git a/chrome/browser/extensions/test_blacklist.h b/chrome/browser/extensions/test_blacklist.h
index 45ea6cbe78f1e78c33a59ef5b141fb0ced612fd3..e4e02e68bbe2f5a7db89da2926ff03b1eda41e0b 100644
--- a/chrome/browser/extensions/test_blacklist.h
+++ b/chrome/browser/extensions/test_blacklist.h
@@ -9,22 +9,79 @@
 
 #include "base/basictypes.h"
 #include "chrome/browser/extensions/blacklist.h"
+#include "chrome/browser/extensions/blacklist_state_fetcher.h"
 
 namespace extensions {
 
+class FakeSafeBrowsingDatabaseManager;
+
+class BlacklistStateFetcherMock : public BlacklistStateFetcher {

[not at google - send to devlin, 2013/12/21 00:34:24]

class needs comment(s)

[Oleg Eterevsky, 2014/01/09 13:54:01]

Done.

+ public:
+  BlacklistStateFetcherMock();
+
+  virtual void Request(const std::string& id,
+                       const RequestCallback& callback) OVERRIDE;
+
+  void SetState(const std::string& id, BlacklistState state);
+
+  void Clear();
+
+  int request_count() const { return request_count_; }
+
+ private:
+  std::map<std::string, BlacklistState> states_;
+  int request_count_;
+};
+
+
 // A wrapper for an extensions::Blacklist that provides functionality for
-// testing.
+// testing. It setts up mocks for SB database and BlacklistFetcher, that are

[not at google - send to devlin, 2013/12/21 00:34:24]

sets

[Oleg Eterevsky, 2014/01/09 13:54:01]

Done.

+// used by blacklist to

[not at google - send to devlin, 2013/12/21 00:34:24]

to..?

[Oleg Eterevsky, 2014/01/09 13:54:01]

Done.

 class TestBlacklist {
  public:
+  // Use this if the SafeBrowsing and/or StateFetcher mocks should be created
+  // before initializing the Blacklist.
+  explicit TestBlacklist();
+
   explicit TestBlacklist(Blacklist* blacklist);
 
+  ~TestBlacklist();
+
+  void Attach(Blacklist* blacklist);
+
+  // Only call this if Blacklist is destroyed before TestBlacklist, otherwise
+  // it will be performed from the destructor.
+  void Detach();
+
   Blacklist* blacklist() { return blacklist_; }
 
+  // Set the extension state in SafeBrowsingDatabaseManager and
+  // BlacklistFetcher.
+  void SetBlacklistState(const std::string& extension_id,
+                         BlacklistState state,
+                         bool notify);
+
   BlacklistState GetBlacklistState(const std::string& extension_id);
 
+  void Clear(bool notify);
+
+  void DisableSafeBrowsing();
+
+  void EnableSafeBrowsing();
+
+  void NotifyUpdate();
+
+  const BlacklistStateFetcherMock* fetcher() { return &state_fetcher_mock_; }
+
  private:
   Blacklist* blacklist_;
 
+  BlacklistStateFetcherMock state_fetcher_mock_;
+
+  scoped_refptr<FakeSafeBrowsingDatabaseManager> blacklist_db_;
+
+  Blacklist::ScopedDatabaseManagerForTest scoped_blacklist_db_;
+
   DISALLOW_COPY_AND_ASSIGN(TestBlacklist);
 };