blob: 765ff70064db752f9d820f44805227d6a17e87f9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
// 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 "base/base64url.h"
#include "base/memory/scoped_ptr.h"
#include "components/copresence/handlers/gcm_handler_impl.h"
#include "components/copresence/proto/push_message.pb.h"
#include "components/copresence/test/fake_directive_handler.h"
#include "components/gcm_driver/fake_gcm_driver.h"
#include "components/gcm_driver/gcm_client.h"
#include "testing/gmock/include/gmock/gmock.h"
namespace copresence {
namespace {
using google::protobuf::RepeatedPtrField;
void IgnoreMessages(
const RepeatedPtrField<SubscribedMessage>& /* messages */) {}
} // namespace
class GCMHandlerTest : public testing::Test {
public:
GCMHandlerTest()
: driver_(new gcm::FakeGCMDriver),
directive_handler_(new FakeDirectiveHandler),
gcm_handler_(driver_.get(),
directive_handler_.get(),
base::Bind(&IgnoreMessages)) {
}
protected:
void ProcessMessage(const gcm::IncomingMessage& message) {
gcm_handler_.OnMessage(GCMHandlerImpl::kCopresenceAppId, message);
}
scoped_ptr<gcm::GCMDriver> driver_;
scoped_ptr<FakeDirectiveHandler> directive_handler_;
GCMHandlerImpl gcm_handler_;
};
TEST_F(GCMHandlerTest, OnMessage) {
// Create a PushMessage.
PushMessage push_message;
push_message.set_type(PushMessage::REPORT);
Report* report = push_message.mutable_report();
report->add_directive()->set_subscription_id("subscription 1");
report->add_directive()->set_subscription_id("subscription 2");
// Encode it.
std::string serialized_proto;
std::string encoded_proto;
push_message.SerializeToString(&serialized_proto);
base::Base64UrlEncode(serialized_proto,
base::Base64UrlEncodePolicy::INCLUDE_PADDING,
&encoded_proto);
// Send it in a GCM message.
gcm::IncomingMessage gcm_message;
gcm_message.data[GCMHandlerImpl::kGcmMessageKey] = encoded_proto;
ProcessMessage(gcm_message);
// Check that the correct directives were passed along.
EXPECT_THAT(directive_handler_->added_directives(),
testing::ElementsAre("subscription 1", "subscription 2"));
}
} // namespace copresence
|
