content/shell/renderer/test_runner/notification_presenter.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 CONTENT_SHELL_RENDERER_TEST_RUNNER_NOTIFICATION_PRESENTER_H_
   6 #define CONTENT_SHELL_RENDERER_TEST_RUNNER_NOTIFICATION_PRESENTER_H_
   7
   8 #include <map>
   9 #include <string>
  10
  11 #include "base/basictypes.h"
  12 #include "third_party/WebKit/public/web/WebNotification.h"
  13 #include "third_party/WebKit/public/web/WebNotificationPresenter.h"
  14
  15 namespace WebTestRunner {
  16 class WebTestDelegate;
  17 }
  18
  19 namespace content {
  20
  21 // A class that implements WebNotificationPresenter for the TestRunner library.
  22 class NotificationPresenter : public blink::WebNotificationPresenter {
  23  public:
  24   NotificationPresenter();
  25   virtual ~NotificationPresenter();
  26
  27   // Called by the TestRunner to simulate a user granting permission.
  28   void GrantPermission(const std::string& origin, bool permission_granted);
  29
  30   // Called by the TestRunner to simulate a user clicking on a notification.
  31   bool SimulateClick(const std::string& title);
  32
  33   // Called by the TestRunner to reset the presenter to an default state.
  34   void Reset();
  35
  36   void set_delegate(WebTestRunner::WebTestDelegate* delegate) {
  37     delegate_ = delegate;
  38   }
  39
  40   // blink::WebNotificationPresenter interface
  41   virtual bool show(const blink::WebNotification& notification);
  42   virtual void cancel(const blink::WebNotification& notification);
  43   virtual void objectDestroyed(const blink::WebNotification& notification);
  44   virtual Permission checkPermission(
  45       const blink::WebSecurityOrigin& security_origin);
  46   virtual void requestPermission(
  47       const blink::WebSecurityOrigin& security_origin,
  48       blink::WebNotificationPermissionCallback* callback);
  49
  50  private:
  51   WebTestRunner::WebTestDelegate* delegate_;
  52
  53   // Map of known origins and whether they are allowed to show notifications.
  54   typedef std::map<std::string, bool> KnownOriginMap;
  55   KnownOriginMap known_origins_;
  56
  57   // Map of currently active notifications.
  58   typedef std::map<std::string, blink::WebNotification> ActiveNotificationMap;
  59   ActiveNotificationMap active_notifications_;
  60
  61   // Map of replacement identifiers to the titles of those notifications.
  62   std::map<std::string, std::string> replacements_;
  63
  64   DISALLOW_COPY_AND_ASSIGN(NotificationPresenter);
  65 };
  66
  67 }  // namespace content
  68
  69 #endif  // CONTENT_SHELL_RENDERER_TEST_RUNNER_NOTIFICATION_PRESENTER_H_