Home | History | Annotate | Download | only in test_runner 
      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 #include "content/shell/renderer/test_runner/notification_presenter.h"
      6 
      7 #include "content/shell/renderer/test_runner/WebTestDelegate.h"
      8 #include "third_party/WebKit/public/platform/Platform.h"
      9 #include "third_party/WebKit/public/platform/WebString.h"
     10 #include "third_party/WebKit/public/platform/WebURL.h"
     11 #include "third_party/WebKit/public/web/WebKit.h"
     12 #include "third_party/WebKit/public/web/WebNotificationPermissionCallback.h"
     13 #include "third_party/WebKit/public/web/WebSecurityOrigin.h"
     14 #include "url/gurl.h"
     15 
     16 using blink::Platform;
     17 using blink::WebNotification;
     18 using blink::WebNotificationPermissionCallback;
     19 using blink::WebNotificationPresenter;
     20 using blink::WebSecurityOrigin;
     21 using blink::WebTextDirectionRightToLeft;
     22 
     23 namespace content {
     24 
     25 namespace {
     26 
     27 void DeferredDisplayDispatch(void* context) {
     28   WebNotification* notification = static_cast<WebNotification*>(context);
     29   notification->dispatchDisplayEvent();
     30 
     31   delete notification;
     32 }
     33 
     34 }  // namespace
     35 
     36 NotificationPresenter::NotificationPresenter() : delegate_(0) {}
     37 
     38 NotificationPresenter::~NotificationPresenter() {}
     39 
     40 void NotificationPresenter::GrantPermission(const std::string& origin,
     41                                             bool permission_granted) {
     42   known_origins_[origin] = permission_granted;
     43 }
     44 
     45 bool NotificationPresenter::SimulateClick(const std::string& title) {
     46   ActiveNotificationMap::iterator iter = active_notifications_.find(title);
     47   if (iter == active_notifications_.end())
     48     return false;
     49 
     50   const WebNotification& notification = iter->second;
     51 
     52   WebNotification event_target(notification);
     53   event_target.dispatchClickEvent();
     54 
     55   return true;
     56 }
     57 
     58 void NotificationPresenter::Reset() {
     59   while (!active_notifications_.empty()) {
     60     const WebNotification& notification = active_notifications_.begin()->second;
     61     cancel(notification);
     62   }
     63 
     64   known_origins_.clear();
     65   replacements_.clear();
     66 }
     67 
     68 bool NotificationPresenter::show(const WebNotification& notification) {
     69   if (!notification.replaceId().isEmpty()) {
     70     std::string replaceId(notification.replaceId().utf8());
     71     if (replacements_.find(replaceId) != replacements_.end()) {
     72       delegate_->printMessage(std::string("REPLACING NOTIFICATION ") +
     73                               replacements_.find(replaceId)->second + "\n");
     74     }
     75     replacements_[replaceId] = notification.title().utf8();
     76   }
     77 
     78   delegate_->printMessage("DESKTOP NOTIFICATION SHOWN: ");
     79   if (!notification.title().isEmpty())
     80     delegate_->printMessage(notification.title().utf8().data());
     81 
     82   if (notification.direction() == WebTextDirectionRightToLeft)
     83     delegate_->printMessage(", RTL");
     84 
     85   // TODO(beverloo): WebNotification should expose the "lang" attribute's value.
     86 
     87   if (!notification.body().isEmpty()) {
     88     delegate_->printMessage(std::string(", body: ") +
     89                             notification.body().utf8().data());
     90   }
     91 
     92   if (!notification.replaceId().isEmpty()) {
     93     delegate_->printMessage(std::string(", tag: ") +
     94                             notification.replaceId().utf8().data());
     95   }
     96 
     97   if (!notification.iconURL().isEmpty()) {
     98     delegate_->printMessage(std::string(", icon: ") +
     99                             notification.iconURL().spec().data());
    100   }
    101 
    102   delegate_->printMessage("\n");
    103 
    104   std::string title = notification.title().utf8();
    105   active_notifications_[title] = notification;
    106 
    107   Platform::current()->callOnMainThread(DeferredDisplayDispatch,
    108                                         new WebNotification(notification));
    109 
    110   return true;
    111 }
    112 
    113 void NotificationPresenter::cancel(const WebNotification& notification) {
    114   std::string title = notification.title().utf8();
    115 
    116   delegate_->printMessage(std::string("DESKTOP NOTIFICATION CLOSED: ") + title +
    117                           "\n");
    118 
    119   WebNotification event_target(notification);
    120   event_target.dispatchCloseEvent(false);
    121 
    122   active_notifications_.erase(title);
    123 }
    124 
    125 void NotificationPresenter::objectDestroyed(
    126     const WebNotification& notification) {
    127   std::string title = notification.title().utf8();
    128   active_notifications_.erase(title);
    129 }
    130 
    131 WebNotificationPresenter::Permission NotificationPresenter::checkPermission(
    132     const WebSecurityOrigin& security_origin) {
    133   const std::string origin = security_origin.toString().utf8();
    134   const KnownOriginMap::iterator it = known_origins_.find(origin);
    135   if (it == known_origins_.end())
    136     return WebNotificationPresenter::PermissionNotAllowed;
    137 
    138   // Values in |known_origins_| indicate whether permission has been granted.
    139   if (it->second)
    140     return WebNotificationPresenter::PermissionAllowed;
    141 
    142   return WebNotificationPresenter::PermissionDenied;
    143 }
    144 
    145 void NotificationPresenter::requestPermission(
    146     const WebSecurityOrigin& security_origin,
    147     WebNotificationPermissionCallback* callback) {
    148   std::string origin = security_origin.toString().utf8();
    149   delegate_->printMessage("DESKTOP NOTIFICATION PERMISSION REQUESTED: " +
    150                           origin + "\n");
    151   callback->permissionRequestComplete();
    152 }
    153 
    154 }  // namespace content
    155