OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "remoting/host/heartbeat_sender.h" | 5 #include "remoting/host/heartbeat_sender.h" |
6 | 6 |
7 #include <set> | 7 #include <set> |
8 | 8 |
9 #include "base/memory/ref_counted.h" | 9 #include "base/memory/ref_counted.h" |
10 #include "base/message_loop/message_loop.h" | 10 #include "base/message_loop/message_loop.h" |
(...skipping 26 matching lines...) Expand all Loading... |
37 namespace { | 37 namespace { |
38 | 38 |
39 const char kTestBotJid[] = "remotingunittest@bot.talk.google.com"; | 39 const char kTestBotJid[] = "remotingunittest@bot.talk.google.com"; |
40 const char kHostId[] = "0"; | 40 const char kHostId[] = "0"; |
41 const char kTestJid[] = "user@gmail.com/chromoting123"; | 41 const char kTestJid[] = "user@gmail.com/chromoting123"; |
42 const char kStanzaId[] = "123"; | 42 const char kStanzaId[] = "123"; |
43 | 43 |
44 class MockListener : public HeartbeatSender::Listener { | 44 class MockListener : public HeartbeatSender::Listener { |
45 public: | 45 public: |
46 // Overridden from HeartbeatSender::Listener | 46 // Overridden from HeartbeatSender::Listener |
47 virtual void OnUnknownHostIdError() OVERRIDE { | 47 virtual void OnUnknownHostIdError() override { |
48 NOTREACHED(); | 48 NOTREACHED(); |
49 } | 49 } |
50 | 50 |
51 // Overridden from HeartbeatSender::Listener | 51 // Overridden from HeartbeatSender::Listener |
52 MOCK_METHOD0(OnHeartbeatSuccessful, void()); | 52 MOCK_METHOD0(OnHeartbeatSuccessful, void()); |
53 }; | 53 }; |
54 | 54 |
55 } // namespace | 55 } // namespace |
56 | 56 |
57 ACTION_P(AddListener, list) { | 57 ACTION_P(AddListener, list) { |
58 list->insert(arg0); | 58 list->insert(arg0); |
59 } | 59 } |
60 ACTION_P(RemoveListener, list) { | 60 ACTION_P(RemoveListener, list) { |
61 EXPECT_TRUE(list->find(arg0) != list->end()); | 61 EXPECT_TRUE(list->find(arg0) != list->end()); |
62 list->erase(arg0); | 62 list->erase(arg0); |
63 } | 63 } |
64 | 64 |
65 class HeartbeatSenderTest | 65 class HeartbeatSenderTest |
66 : public testing::Test { | 66 : public testing::Test { |
67 protected: | 67 protected: |
68 virtual void SetUp() OVERRIDE { | 68 virtual void SetUp() override { |
69 key_pair_ = RsaKeyPair::FromString(kTestRsaKeyPair); | 69 key_pair_ = RsaKeyPair::FromString(kTestRsaKeyPair); |
70 ASSERT_TRUE(key_pair_.get()); | 70 ASSERT_TRUE(key_pair_.get()); |
71 | 71 |
72 EXPECT_CALL(signal_strategy_, GetState()) | 72 EXPECT_CALL(signal_strategy_, GetState()) |
73 .WillOnce(Return(SignalStrategy::DISCONNECTED)); | 73 .WillOnce(Return(SignalStrategy::DISCONNECTED)); |
74 EXPECT_CALL(signal_strategy_, AddListener(NotNull())) | 74 EXPECT_CALL(signal_strategy_, AddListener(NotNull())) |
75 .WillRepeatedly(AddListener(&signal_strategy_listeners_)); | 75 .WillRepeatedly(AddListener(&signal_strategy_listeners_)); |
76 EXPECT_CALL(signal_strategy_, RemoveListener(NotNull())) | 76 EXPECT_CALL(signal_strategy_, RemoveListener(NotNull())) |
77 .WillRepeatedly(RemoveListener(&signal_strategy_listeners_)); | 77 .WillRepeatedly(RemoveListener(&signal_strategy_listeners_)); |
78 EXPECT_CALL(signal_strategy_, GetLocalJid()) | 78 EXPECT_CALL(signal_strategy_, GetLocalJid()) |
79 .WillRepeatedly(Return(kTestJid)); | 79 .WillRepeatedly(Return(kTestJid)); |
80 | 80 |
81 heartbeat_sender_.reset(new HeartbeatSender( | 81 heartbeat_sender_.reset(new HeartbeatSender( |
82 &mock_listener_, kHostId, &signal_strategy_, key_pair_, kTestBotJid)); | 82 &mock_listener_, kHostId, &signal_strategy_, key_pair_, kTestBotJid)); |
83 } | 83 } |
84 | 84 |
85 virtual void TearDown() OVERRIDE { | 85 virtual void TearDown() override { |
86 heartbeat_sender_.reset(); | 86 heartbeat_sender_.reset(); |
87 EXPECT_TRUE(signal_strategy_listeners_.empty()); | 87 EXPECT_TRUE(signal_strategy_listeners_.empty()); |
88 } | 88 } |
89 | 89 |
90 void ValidateHeartbeatStanza(XmlElement* stanza, | 90 void ValidateHeartbeatStanza(XmlElement* stanza, |
91 const char* expectedSequenceId); | 91 const char* expectedSequenceId); |
92 | 92 |
93 base::MessageLoop message_loop_; | 93 base::MessageLoop message_loop_; |
94 MockSignalStrategy signal_strategy_; | 94 MockSignalStrategy signal_strategy_; |
95 MockListener mock_listener_; | 95 MockListener mock_listener_; |
(...skipping 154 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
250 EXPECT_TRUE(heartbeat_stanza->NextNamed(signature_tag) == NULL); | 250 EXPECT_TRUE(heartbeat_stanza->NextNamed(signature_tag) == NULL); |
251 | 251 |
252 scoped_refptr<RsaKeyPair> key_pair = RsaKeyPair::FromString(kTestRsaKeyPair); | 252 scoped_refptr<RsaKeyPair> key_pair = RsaKeyPair::FromString(kTestRsaKeyPair); |
253 ASSERT_TRUE(key_pair.get()); | 253 ASSERT_TRUE(key_pair.get()); |
254 std::string expected_signature = | 254 std::string expected_signature = |
255 key_pair->SignMessage(std::string(kTestJid) + ' ' + expectedSequenceId); | 255 key_pair->SignMessage(std::string(kTestJid) + ' ' + expectedSequenceId); |
256 EXPECT_EQ(expected_signature, signature->BodyText()); | 256 EXPECT_EQ(expected_signature, signature->BodyText()); |
257 } | 257 } |
258 | 258 |
259 } // namespace remoting | 259 } // namespace remoting |
OLD | NEW |