Fetch extension blacklist states from SafeBrowsing server

chrome/browser/extensions/test_blacklist_fetcher.h

+// Copyright 2013 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#ifndef CHROME_BROWSER_EXTENSIONS_TEST_BLACKLIST_FETCHER_H_
+#define CHROME_BROWSER_EXTENSIONS_TEST_BLACKLIST_FETCHER_H_
+
+#include <string>
+
+#include "chrome/browser/extensions/blacklist_fetcher.h"
+#include "chrome/common/safe_browsing/crx_info.pb.h"
+#include "net/url_request/test_url_fetcher_factory.h"
+
+namespace extensions {
+
+// A wrapper for extensions::BlacklistFetcher, emulating server responses.
+class TestBlacklistFetcher {
+ public:
+  explicit TestBlacklistFetcher(BlacklistFetcher* fetcher);
+
+  void SetBlacklistVerdict(const std::string& id,
+                           ClientCRXListInfoResponse_Verdict state);
+
+  // Get URLFetcher by its id from factory, send the appropriate response.
+  // Return false, if fetcher with fiven id doesn't exist or in case of
+  // incorrect request. Otherwise return true.
+  bool HandleFetcher(int id);
+
+ private:
+  BlacklistFetcher* fetcher_;
+
+  std::map<std::string, ClientCRXListInfoResponse_Verdict> verdicts_;
+
+  net::TestURLFetcherFactory url_fetcher_factory_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestBlacklistFetcher);
+};
+
+}  // namespace extensions
+
+#endif  // CHROME_BROWSER_EXTENSIONS_TEST_BLACKLIST_FETCHER_H_