| 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 |