Home | History | Annotate | Download | only in extensions 
      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
      3 // found in the LICENSE file.
      4 
      5 #ifndef CHROME_BROWSER_EXTENSIONS_TEST_BLACKLIST_H_
      6 #define CHROME_BROWSER_EXTENSIONS_TEST_BLACKLIST_H_
      7 
      8 #include <map>
      9 #include <string>
     10 
     11 #include "base/basictypes.h"
     12 #include "chrome/browser/extensions/blacklist.h"
     13 #include "chrome/browser/extensions/blacklist_state_fetcher.h"
     14 
     15 namespace extensions {
     16 
     17 class FakeSafeBrowsingDatabaseManager;
     18 
     19 // Replace BlacklistStateFetcher for testing of the Blacklist class.
     20 class BlacklistStateFetcherMock : public BlacklistStateFetcher {
     21  public:
     22   BlacklistStateFetcherMock();
     23 
     24   virtual ~BlacklistStateFetcherMock();
     25 
     26   virtual void Request(const std::string& id,
     27                        const RequestCallback& callback) OVERRIDE;
     28 
     29   void SetState(const std::string& id, BlacklistState state);
     30 
     31   void Clear();
     32 
     33   int request_count() const { return request_count_; }
     34 
     35  private:
     36   std::map<std::string, BlacklistState> states_;
     37   int request_count_;
     38 };
     39 
     40 
     41 // A wrapper for an extensions::Blacklist that provides functionality for
     42 // testing. It sets up mocks for SafeBrowsing database and BlacklistFetcher,
     43 // that are used by blacklist to retrieve respectively the set of blacklisted
     44 // extensions and their blacklist states.
     45 class TestBlacklist {
     46  public:
     47   // Use this if the SafeBrowsing and/or StateFetcher mocks should be created
     48   // before initializing the Blacklist.
     49   explicit TestBlacklist();
     50 
     51   explicit TestBlacklist(Blacklist* blacklist);
     52 
     53   ~TestBlacklist();
     54 
     55   void Attach(Blacklist* blacklist);
     56 
     57   // Only call this if Blacklist is destroyed before TestBlacklist, otherwise
     58   // it will be performed from the destructor.
     59   void Detach();
     60 
     61   Blacklist* blacklist() { return blacklist_; }
     62 
     63   // Set the extension state in SafeBrowsingDatabaseManager and
     64   // BlacklistFetcher.
     65   void SetBlacklistState(const std::string& extension_id,
     66                          BlacklistState state,
     67                          bool notify);
     68 
     69   BlacklistState GetBlacklistState(const std::string& extension_id);
     70 
     71   void Clear(bool notify);
     72 
     73   void DisableSafeBrowsing();
     74 
     75   void EnableSafeBrowsing();
     76 
     77   void NotifyUpdate();
     78 
     79   const BlacklistStateFetcherMock* fetcher() { return &state_fetcher_mock_; }
     80 
     81  private:
     82   Blacklist* blacklist_;
     83 
     84   // The BlacklistStateFetcher object is normally managed by Blacklist. Because
     85   // of this, we need to prevent this object from being deleted with Blacklist.
     86   // For this, Detach() should be called before blacklist_ is deleted.
     87   BlacklistStateFetcherMock state_fetcher_mock_;
     88 
     89   scoped_refptr<FakeSafeBrowsingDatabaseManager> blacklist_db_;
     90 
     91   Blacklist::ScopedDatabaseManagerForTest scoped_blacklist_db_;
     92 
     93   DISALLOW_COPY_AND_ASSIGN(TestBlacklist);
     94 };
     95 
     96 }  // namespace extensions
     97 
     98 #endif  // CHROME_BROWSER_EXTENSIONS_TEST_BLACKLIST_H_
     99