content/shell/browser/layout_test/layout_test_push_messaging_service.cc

   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/browser/layout_test/layout_test_push_messaging_service.h"
   6
   7 #include "base/callback.h"
   8 #include "base/logging.h"
   9
  10 namespace content {
  11
  12 LayoutTestPushMessagingService::LayoutTestPushMessagingService() {
  13 }
  14
  15 LayoutTestPushMessagingService::~LayoutTestPushMessagingService() {
  16 }
  17
  18 void LayoutTestPushMessagingService::SetPermission(const GURL& origin,
  19                                                    bool allowed) {
  20   permission_map_[origin] = allowed ? blink::WebPushPermissionStatusGranted
  21                                     : blink::WebPushPermissionStatusDenied;
  22 }
  23
  24 void LayoutTestPushMessagingService::ClearPermissions() {
  25   permission_map_.clear();
  26 }
  27
  28 GURL LayoutTestPushMessagingService::GetPushEndpoint() {
  29   return GURL("https://example.com/LayoutTestEndpoint");
  30 }
  31
  32 void LayoutTestPushMessagingService::RegisterFromDocument(
  33     const GURL& requesting_origin,
  34     int64 service_worker_registration_id,
  35     const std::string& sender_id,
  36     int renderer_id,
  37     int render_frame_id,
  38     bool user_gesture,
  39     const PushMessagingService::RegisterCallback& callback) {
  40   RegisterFromWorker(requesting_origin, service_worker_registration_id,
  41                      sender_id, callback);
  42 }
  43
  44 void LayoutTestPushMessagingService::RegisterFromWorker(
  45     const GURL& requesting_origin,
  46     int64 service_worker_registration_id,
  47     const std::string& sender_id,
  48     const PushMessagingService::RegisterCallback& callback) {
  49   if (GetPermissionStatus(requesting_origin, requesting_origin) ==
  50       blink::WebPushPermissionStatusGranted) {
  51     callback.Run("layoutTestRegistrationId",
  52                  PUSH_REGISTRATION_STATUS_SUCCESS_FROM_PUSH_SERVICE);
  53   } else {
  54     callback.Run("registration_id", PUSH_REGISTRATION_STATUS_PERMISSION_DENIED);
  55   }
  56 }
  57
  58 blink::WebPushPermissionStatus
  59 LayoutTestPushMessagingService::GetPermissionStatus(
  60     const GURL& requesting_origin,
  61     const GURL& embedding_origin) {
  62   const auto& it = permission_map_.find(requesting_origin);
  63   if (it == permission_map_.end())
  64     return blink::WebPushPermissionStatusDefault;
  65   return it->second;
  66 }
  67
  68 void LayoutTestPushMessagingService::Unregister(
  69     const GURL& requesting_origin,
  70     int64 service_worker_registration_id,
  71     const std::string& sender_id,
  72     const UnregisterCallback& callback) {
  73   callback.Run(PUSH_UNREGISTRATION_STATUS_SUCCESS_UNREGISTERED);
  74 }
  75
  76 }  // namespace content