| Index: chrome/browser/push_messaging/push_messaging_service_unittest.cc
|
| diff --git a/chrome/browser/push_messaging/push_messaging_service_unittest.cc b/chrome/browser/push_messaging/push_messaging_service_unittest.cc
|
| index 31446b929af4659a93fd3d6da179981c2e680f1f..19ddd3b7cd48fe5b6dea97070278f15dfb1ff636 100644
|
| --- a/chrome/browser/push_messaging/push_messaging_service_unittest.cc
|
| +++ b/chrome/browser/push_messaging/push_messaging_service_unittest.cc
|
| @@ -27,6 +27,7 @@
|
| #include "components/gcm_driver/fake_gcm_client_factory.h"
|
| #include "components/gcm_driver/gcm_profile_service.h"
|
| #include "content/public/common/content_switches.h"
|
| +#include "content/public/common/push_event_payload.h"
|
| #include "content/public/test/test_browser_thread_bundle.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
|
|
| @@ -123,15 +124,15 @@ class PushMessagingServiceTest : public ::testing::Test {
|
| void DidDispatchMessage(std::string* app_id_out,
|
| GURL* origin_out,
|
| int64_t* service_worker_registration_id_out,
|
| - std::string* message_data_out,
|
| + content::PushEventPayload* payload_out,
|
| const std::string& app_id,
|
| const GURL& origin,
|
| int64_t service_worker_registration_id,
|
| - const std::string& message_data) {
|
| + const content::PushEventPayload& payload) {
|
| *app_id_out = app_id;
|
| *origin_out = origin;
|
| *service_worker_registration_id_out = service_worker_registration_id;
|
| - *message_data_out = message_data;
|
| + *payload_out = payload;
|
| }
|
|
|
| protected:
|
| @@ -193,17 +194,17 @@ TEST_F(PushMessagingServiceTest, PayloadEncryptionTest) {
|
|
|
| ASSERT_FALSE(app_identifier.is_null());
|
|
|
| - std::string app_id, message_data;
|
| + std::string app_id;
|
| GURL dispatched_origin;
|
| int64_t service_worker_registration_id;
|
| + content::PushEventPayload payload;
|
|
|
| // (5) Observe message dispatchings from the Push Messaging service, and then
|
| // dispatch the |message| on the GCM driver as if it had actually been
|
| // received by Google Cloud Messaging.
|
| - push_service->SetMessageDispatchedCallbackForTesting(
|
| - base::Bind(&PushMessagingServiceTest::DidDispatchMessage,
|
| - base::Unretained(this), &app_id, &dispatched_origin,
|
| - &service_worker_registration_id, &message_data));
|
| + push_service->SetMessageDispatchedCallbackForTesting(base::Bind(
|
| + &PushMessagingServiceTest::DidDispatchMessage, base::Unretained(this),
|
| + &app_id, &dispatched_origin, &service_worker_registration_id, &payload));
|
|
|
| gcm::FakeGCMProfileService* fake_profile_service =
|
| static_cast<gcm::FakeGCMProfileService*>(
|
| @@ -220,5 +221,6 @@ TEST_F(PushMessagingServiceTest, PayloadEncryptionTest) {
|
| EXPECT_EQ(origin, dispatched_origin);
|
| EXPECT_EQ(service_worker_registration_id, kTestServiceWorkerId);
|
|
|
| - EXPECT_EQ(kTestPayload, message_data);
|
| + EXPECT_FALSE(payload.isNull());
|
| + EXPECT_EQ(kTestPayload, payload.getData());
|
| }
|
|
|