| 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 47 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 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 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 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 |