| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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 "base/message_loop.h" | 5 #include "base/message_loop.h" |
| 6 #include "base/message_loop_proxy.h" | 6 #include "base/message_loop_proxy.h" |
| 7 #include "base/ref_counted.h" | 7 #include "base/ref_counted.h" |
| 8 #include "base/string_number_conversions.h" | 8 #include "base/string_number_conversions.h" |
| 9 #include "remoting/base/constants.h" | 9 #include "remoting/base/constants.h" |
| 10 #include "remoting/host/heartbeat_sender.h" | 10 #include "remoting/host/heartbeat_sender.h" |
| (...skipping 68 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 79 scoped_refptr<InMemoryHostConfig> config_; | 79 scoped_refptr<InMemoryHostConfig> config_; |
| 80 }; | 80 }; |
| 81 | 81 |
| 82 TEST_F(HeartbeatSenderTest, DoSendStanza) { | 82 TEST_F(HeartbeatSenderTest, DoSendStanza) { |
| 83 // This test calls Start() followed by Stop(), and makes sure an Iq | 83 // This test calls Start() followed by Stop(), and makes sure an Iq |
| 84 // stanza is being send. | 84 // stanza is being send. |
| 85 | 85 |
| 86 // |iq_request| is freed by HeartbeatSender. | 86 // |iq_request| is freed by HeartbeatSender. |
| 87 MockIqRequest* iq_request = new MockIqRequest(jingle_client_); | 87 MockIqRequest* iq_request = new MockIqRequest(jingle_client_); |
| 88 | 88 |
| 89 scoped_refptr<HeartbeatSender> heartbeat_sender = new HeartbeatSender(); | 89 scoped_refptr<HeartbeatSender> heartbeat_sender(new HeartbeatSender()); |
| 90 ASSERT_TRUE(heartbeat_sender->Init(config_, jingle_client_)); | 90 ASSERT_TRUE(heartbeat_sender->Init(config_, jingle_client_)); |
| 91 | 91 |
| 92 EXPECT_CALL(*jingle_client_, CreateIqRequest()) | 92 EXPECT_CALL(*jingle_client_, CreateIqRequest()) |
| 93 .WillOnce(Return(iq_request)); | 93 .WillOnce(Return(iq_request)); |
| 94 | 94 |
| 95 EXPECT_CALL(*iq_request, SendIq(buzz::STR_SET, kChromotingBotJid, NotNull())) | 95 EXPECT_CALL(*iq_request, SendIq(buzz::STR_SET, kChromotingBotJid, NotNull())) |
| 96 .WillOnce(DoAll(DeleteArg<2>(), Return())); | 96 .WillOnce(DoAll(DeleteArg<2>(), Return())); |
| 97 | 97 |
| 98 heartbeat_sender->Start(); | 98 heartbeat_sender->Start(); |
| 99 message_loop_.RunAllPending(); | 99 message_loop_.RunAllPending(); |
| 100 | 100 |
| 101 heartbeat_sender->Stop(); | 101 heartbeat_sender->Stop(); |
| 102 message_loop_.RunAllPending(); | 102 message_loop_.RunAllPending(); |
| 103 } | 103 } |
| 104 | 104 |
| 105 TEST_F(HeartbeatSenderTest, CreateHeartbeatMessage) { | 105 TEST_F(HeartbeatSenderTest, CreateHeartbeatMessage) { |
| 106 // This test validates format of the heartbeat stanza. | 106 // This test validates format of the heartbeat stanza. |
| 107 | 107 |
| 108 scoped_refptr<HeartbeatSender> heartbeat_sender = new HeartbeatSender(); | 108 scoped_refptr<HeartbeatSender> heartbeat_sender(new HeartbeatSender()); |
| 109 ASSERT_TRUE(heartbeat_sender->Init(config_, jingle_client_)); | 109 ASSERT_TRUE(heartbeat_sender->Init(config_, jingle_client_)); |
| 110 | 110 |
| 111 int64 start_time = static_cast<int64>(base::Time::Now().ToDoubleT()); | 111 int64 start_time = static_cast<int64>(base::Time::Now().ToDoubleT()); |
| 112 | 112 |
| 113 scoped_ptr<buzz::XmlElement> stanza( | 113 scoped_ptr<buzz::XmlElement> stanza( |
| 114 heartbeat_sender->CreateHeartbeatMessage()); | 114 heartbeat_sender->CreateHeartbeatMessage()); |
| 115 ASSERT_TRUE(stanza.get() != NULL); | 115 ASSERT_TRUE(stanza.get() != NULL); |
| 116 | 116 |
| 117 EXPECT_TRUE(buzz::QName(kChromotingXmlNamespace, "heartbeat") == | 117 EXPECT_TRUE(buzz::QName(kChromotingXmlNamespace, "heartbeat") == |
| 118 stanza->Name()); | 118 stanza->Name()); |
| (...skipping 14 matching lines...) Expand all Loading... |
| 133 EXPECT_GE(now, time); | 133 EXPECT_GE(now, time); |
| 134 | 134 |
| 135 HostKeyPair key_pair; | 135 HostKeyPair key_pair; |
| 136 key_pair.LoadFromString(kTestHostKeyPair); | 136 key_pair.LoadFromString(kTestHostKeyPair); |
| 137 std::string expected_signature = | 137 std::string expected_signature = |
| 138 key_pair.GetSignature(std::string(kTestJid) + ' ' + time_str); | 138 key_pair.GetSignature(std::string(kTestJid) + ' ' + time_str); |
| 139 EXPECT_EQ(expected_signature, signature->BodyText()); | 139 EXPECT_EQ(expected_signature, signature->BodyText()); |
| 140 } | 140 } |
| 141 | 141 |
| 142 } // namespace remoting | 142 } // namespace remoting |
| OLD | NEW |