| Index: remoting/host/heartbeat_sender_unittest.cc
|
| diff --git a/remoting/host/heartbeat_sender_unittest.cc b/remoting/host/heartbeat_sender_unittest.cc
|
| index e02cc35ea3886461d52b20fce16f830e12f48843..9d2fe4de523e636509d8eec9bc73c9f14fa84c99 100644
|
| --- a/remoting/host/heartbeat_sender_unittest.cc
|
| +++ b/remoting/host/heartbeat_sender_unittest.cc
|
| @@ -75,18 +75,9 @@ class HeartbeatSenderTest : public testing::Test {
|
| config_ = new InMemoryHostConfig();
|
| config_->SetString(kHostIdConfigPath, kHostId);
|
| config_->SetString(kPrivateKeyConfigPath, kTestHostKeyPair);
|
| -
|
| - jingle_thread_.message_loop_ = &message_loop_;
|
| -
|
| - signal_strategy_.reset(new MockSignalStrategy());
|
| - jingle_client_ =
|
| - new JingleClient(&jingle_thread_, signal_strategy_.get(), NULL, NULL);
|
| - jingle_client_->full_jid_ = kTestJid;
|
| }
|
|
|
| - JingleThread jingle_thread_;
|
| - scoped_ptr<MockSignalStrategy> signal_strategy_;
|
| - scoped_refptr<JingleClient> jingle_client_;
|
| + MockSignalStrategy signal_strategy_;
|
| MessageLoop message_loop_;
|
| scoped_refptr<InMemoryHostConfig> config_;
|
| };
|
| @@ -101,32 +92,32 @@ TEST_F(HeartbeatSenderTest, DoSendStanza) {
|
| EXPECT_CALL(*iq_request, set_callback(_)).Times(1);
|
|
|
| scoped_refptr<HeartbeatSender> heartbeat_sender(
|
| - new HeartbeatSender(&message_loop_, jingle_client_.get(), config_));
|
| + new HeartbeatSender(&message_loop_, config_));
|
| ASSERT_TRUE(heartbeat_sender->Init());
|
|
|
| - EXPECT_CALL(*signal_strategy_, CreateIqRequest())
|
| + EXPECT_CALL(signal_strategy_, CreateIqRequest())
|
| .WillOnce(Return(iq_request));
|
|
|
| EXPECT_CALL(*iq_request, SendIq(buzz::STR_SET, kChromotingBotJid, NotNull()))
|
| .WillOnce(DoAll(DeleteArg<2>(), Return()));
|
|
|
| - heartbeat_sender->Start();
|
| + heartbeat_sender->OnSignallingConnected(&signal_strategy_, kTestJid);
|
| message_loop_.RunAllPending();
|
|
|
| - heartbeat_sender->Stop();
|
| + heartbeat_sender->OnSignallingDisconnected();
|
| message_loop_.RunAllPending();
|
| }
|
|
|
| // Validate format of the heartbeat stanza.
|
| TEST_F(HeartbeatSenderTest, CreateHeartbeatMessage) {
|
| scoped_refptr<HeartbeatSender> heartbeat_sender(
|
| - new HeartbeatSender(&message_loop_, jingle_client_.get(), config_));
|
| + new HeartbeatSender(&message_loop_, config_));
|
| ASSERT_TRUE(heartbeat_sender->Init());
|
|
|
| int64 start_time = static_cast<int64>(base::Time::Now().ToDoubleT());
|
|
|
| - scoped_ptr<XmlElement> stanza(
|
| - heartbeat_sender->CreateHeartbeatMessage());
|
| + heartbeat_sender->full_jid_ = kTestJid;
|
| + scoped_ptr<XmlElement> stanza(heartbeat_sender->CreateHeartbeatMessage());
|
| ASSERT_TRUE(stanza.get() != NULL);
|
|
|
| EXPECT_TRUE(QName(kChromotingXmlNamespace, "heartbeat") ==
|
| @@ -171,7 +162,7 @@ TEST_F(HeartbeatSenderTest, ProcessResponse) {
|
| set_interval->AddText(base::IntToString(kTestInterval));
|
|
|
| scoped_refptr<HeartbeatSender> heartbeat_sender(
|
| - new HeartbeatSender(&message_loop_, jingle_client_.get(), config_));
|
| + new HeartbeatSender(&message_loop_, config_));
|
| heartbeat_sender->ProcessResponse(response.get());
|
|
|
| EXPECT_EQ(kTestInterval * 1000, heartbeat_sender->interval_ms_);
|
|
|