content/shell/renderer/test_runner/mock_web_push_client.cc

// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "content/shell/renderer/test_runner/mock_web_push_client.h"

#include "base/logging.h"
#include "base/memory/scoped_ptr.h"
#include "third_party/WebKit/public/platform/WebPushError.h"
#include "third_party/WebKit/public/platform/WebPushRegistration.h"
#include "third_party/WebKit/public/platform/WebString.h"

using blink::WebString;

namespace content {

MockWebPushClient::MockWebPushClient()
    : error_message_(
          "Registration failed (default mock client error message)") {
}

MockWebPushClient::~MockWebPushClient() {}

void MockWebPushClient::SetMockSuccessValues(
    const std::string& end_point, const std::string& registration_id) {
  end_point_ = end_point;
  registration_id_ = registration_id;
  error_message_ = "";
}

void MockWebPushClient::SetMockErrorValues(const std::string& message) {
  end_point_ = "";
  registration_id_ = "";
  error_message_ = message;
}

void MockWebPushClient::registerPushMessaging(
    const WebString& sender_id,
    blink::WebPushRegistrationCallbacks* callbacks,
    blink::WebServiceWorkerProvider* service_worker_provider) {
  registerPushMessaging(callbacks, service_worker_provider);
}

void MockWebPushClient::registerPushMessaging(
    blink::WebPushRegistrationCallbacks* callbacks,
    blink::WebServiceWorkerProvider* service_worker_provider) {
  DCHECK(callbacks);

  if (!error_message_.empty()) {
    scoped_ptr<blink::WebPushError> error(
        new blink::WebPushError(blink::WebPushError::ErrorTypeAbort,
                                WebString::fromUTF8(error_message_)));
    callbacks->onError(error.release());
  } else {
    DCHECK(!end_point_.empty() && !registration_id_.empty());

    scoped_ptr<blink::WebPushRegistration> registration(
        new blink::WebPushRegistration(WebString::fromUTF8(end_point_),
                                       WebString::fromUTF8(registration_id_)));
    callbacks->onSuccess(registration.release());
  }

  delete callbacks;
}

}  // namespace content