chrome/browser/web_resource/resource_request_allowed_notifier_test_util.h

   1 // Copyright 2013 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_WEB_RESOURCE_RESOURCE_REQUEST_ALLOWED_NOTIFIER_TEST_UTIL_H_
   6 #define CHROME_BROWSER_WEB_RESOURCE_RESOURCE_REQUEST_ALLOWED_NOTIFIER_TEST_UTIL_H_
   7
   8 #include "chrome/browser/web_resource/resource_request_allowed_notifier.h"
   9
  10 // A subclass of ResourceRequestAllowedNotifier used to expose some
  11 // functionality for testing.
  12 //
  13 // By default, the constructor sets this class to override
  14 // ResourceRequestsAllowed, so its state can be set with SetRequestsAllowed.
  15 // This is meant for higher level tests of services to ensure they adhere to the
  16 // notifications of the ResourceRequestAllowedNotifier. Lower level tests can
  17 // disable this by calling SetRequestsAllowedOverride with the value they want
  18 // it to return.
  19 class TestRequestAllowedNotifier : public ResourceRequestAllowedNotifier {
  20  public:
  21   TestRequestAllowedNotifier();
  22   virtual ~TestRequestAllowedNotifier();
  23
  24   // A version of |Init()| that accepts a custom EulaAcceptedNotifier.
  25   void InitWithEulaAcceptNotifier(
  26       Observer* observer,
  27       scoped_ptr<EulaAcceptedNotifier> eula_notifier);
  28
  29   // Makes ResourceRequestsAllowed return |allowed| when it is called.
  30   void SetRequestsAllowedOverride(bool allowed);
  31
  32   // Notify observers that requests are allowed. This will only work if
  33   // the observer is expecting a notification.
  34   void NotifyObserver();
  35
  36   // ResourceRequestAllowedNotifier overrides:
  37   virtual State GetResourceRequestsAllowedState() OVERRIDE;
  38   virtual EulaAcceptedNotifier* CreateEulaNotifier() OVERRIDE;
  39
  40  private:
  41   scoped_ptr<EulaAcceptedNotifier> test_eula_notifier_;
  42   bool override_requests_allowed_;
  43   bool requests_allowed_;
  44
  45   DISALLOW_COPY_AND_ASSIGN(TestRequestAllowedNotifier);
  46 };
  47
  48 #endif  // CHROME_BROWSER_WEB_RESOURCE_RESOURCE_REQUEST_ALLOWED_NOTIFIER_TEST_UTIL_H_