1234567891011121314151617181920212223242526272829303132333435363738394041424344 |
- // Copyright 2019 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 COMPONENTS_SAFE_SEARCH_API_FAKE_URL_CHECKER_CLIENT_H_
- #define COMPONENTS_SAFE_SEARCH_API_FAKE_URL_CHECKER_CLIENT_H_
- #include "base/callback.h"
- #include "components/safe_search_api/url_checker_client.h"
- namespace safe_search_api {
- // Helper class with fake URLCheckerClient for use with URLChecker. This
- // lets tests control the response the URLChecker will receive from the
- // URLCheckerClient. Used to test URLChecker.
- class FakeURLCheckerClient : public URLCheckerClient {
- public:
- FakeURLCheckerClient();
- FakeURLCheckerClient(const FakeURLCheckerClient&) = delete;
- FakeURLCheckerClient& operator=(const FakeURLCheckerClient&) = delete;
- ~FakeURLCheckerClient() override;
- // Fake override that simply holds references of |url| and |callback|.
- //
- // See RunCallback() method documentation below on how to run the callback.
- void CheckURL(const GURL& url, ClientCheckCallback callback) override;
- // Runs the callback function input by the last call of CheckURL() with the
- // result input with the last call of SetResult().
- void RunCallback(ClientClassification classification);
- // Asynchronous version of RunCallback().
- void RunCallbackAsync(ClientClassification classification);
- private:
- ClientCheckCallback callback_;
- GURL url_;
- };
- } // namespace safe_search_api
- #endif // COMPONENTS_SAFE_SEARCH_API_FAKE_URL_CHECKER_CLIENT_H_
|