chrome/browser/ui/login/login_prompt_test_utils.h

   1 // Copyright 2014 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_UI_LOGIN_LOGIN_PROMPT_TEST_UTILS_H_
   6 #define CHROME_BROWSER_UI_LOGIN_LOGIN_PROMPT_TEST_UTILS_H_
   7
   8 #include <list>
   9
  10 #include "chrome/browser/chrome_notification_types.h"
  11 #include "content/public/browser/navigation_controller.h"
  12 #include "content/public/browser/notification_observer.h"
  13 #include "content/public/test/test_utils.h"
  14
  15 class LoginHandler;
  16
  17 // Maintains a set of LoginHandlers that are currently active and
  18 // keeps a count of the notifications that were observed.
  19 class LoginPromptBrowserTestObserver : public content::NotificationObserver {
  20  public:
  21   LoginPromptBrowserTestObserver();
  22   ~LoginPromptBrowserTestObserver() override;
  23
  24   void Observe(int type,
  25                const content::NotificationSource& source,
  26                const content::NotificationDetails& details) override;
  27
  28   void AddHandler(LoginHandler* handler);
  29
  30   void RemoveHandler(LoginHandler* handler);
  31
  32   void Register(const content::NotificationSource& source);
  33
  34   const std::list<LoginHandler*>& handlers() const { return handlers_; }
  35
  36   int auth_needed_count() const { return auth_needed_count_; }
  37   int auth_supplied_count() const { return auth_supplied_count_; }
  38   int auth_cancelled_count() const { return auth_cancelled_count_; }
  39
  40  private:
  41   std::list<LoginHandler*> handlers_;
  42
  43   // The exact number of notifications we receive is depedent on the
  44   // number of requests that were dispatched and is subject to a
  45   // number of factors that we don't directly control here.  The
  46   // values below should only be used qualitatively.
  47   int auth_needed_count_;
  48   int auth_supplied_count_;
  49   int auth_cancelled_count_;
  50
  51  private:
  52   content::NotificationRegistrar registrar_;
  53
  54   DISALLOW_COPY_AND_ASSIGN(LoginPromptBrowserTestObserver);
  55 };
  56
  57 template <int T>
  58 class WindowedNavigationObserver
  59     : public content::WindowedNotificationObserver {
  60  public:
  61   explicit WindowedNavigationObserver(
  62       content::NavigationController* controller);
  63
  64  private:
  65   DISALLOW_COPY_AND_ASSIGN(WindowedNavigationObserver);
  66 };
  67
  68 template <int T>
  69 WindowedNavigationObserver<T>::WindowedNavigationObserver(
  70     content::NavigationController* controller)
  71     : content::WindowedNotificationObserver(
  72         T, content::Source<content::NavigationController>(controller)) {
  73 }
  74
  75 typedef WindowedNavigationObserver<chrome::NOTIFICATION_AUTH_NEEDED>
  76     WindowedAuthNeededObserver;
  77
  78 typedef WindowedNavigationObserver<chrome::NOTIFICATION_AUTH_CANCELLED>
  79     WindowedAuthCancelledObserver;
  80
  81 typedef WindowedNavigationObserver<chrome::NOTIFICATION_AUTH_SUPPLIED>
  82     WindowedAuthSuppliedObserver;
  83
  84 // LOAD_STOP observer is special since we want to be able to wait for
  85 // multiple LOAD_STOP events.
  86 class WindowedLoadStopObserver
  87     : public WindowedNavigationObserver<content::NOTIFICATION_LOAD_STOP> {
  88  public:
  89   WindowedLoadStopObserver(content::NavigationController* controller,
  90                            int notification_count);
  91
  92  protected:
  93   void Observe(int type,
  94                const content::NotificationSource& source,
  95                const content::NotificationDetails& details) override;
  96
  97  private:
  98   int remaining_notification_count_;  // Number of notifications remaining.
  99
 100   DISALLOW_COPY_AND_ASSIGN(WindowedLoadStopObserver);
 101 };
 102
 103 #endif  // CHROME_BROWSER_UI_LOGIN_LOGIN_PROMPT_TEST_UTILS_H_