Index: remoting/host/heartbeat_sender_unittest.cc |
diff --git a/remoting/host/heartbeat_sender_unittest.cc b/remoting/host/heartbeat_sender_unittest.cc |
index 094793748f1cc297f39fd41c6451dddd2500b15e..fac5d4778fe7d0df1d26f1f4789222b4d75477b7 100644 |
--- a/remoting/host/heartbeat_sender_unittest.cc |
+++ b/remoting/host/heartbeat_sender_unittest.cc |
@@ -11,8 +11,8 @@ |
#include "base/message_loop_proxy.h" |
#include "base/string_number_conversions.h" |
#include "remoting/base/constants.h" |
-#include "remoting/host/host_key_pair.h" |
-#include "remoting/host/test_key_pair.h" |
+#include "remoting/base/rsa_key_pair.h" |
+#include "remoting/base/test_rsa_key_pair.h" |
#include "remoting/jingle_glue/iq_sender.h" |
#include "remoting/jingle_glue/mock_objects.h" |
#include "testing/gmock/include/gmock/gmock.h" |
@@ -58,7 +58,8 @@ class HeartbeatSenderTest |
} |
virtual void SetUp() OVERRIDE { |
- ASSERT_TRUE(key_pair_.LoadFromString(kTestHostKeyPair)); |
+ key_pair_ = new RsaKeyPair(); |
+ ASSERT_TRUE(key_pair_->LoadFromString(kTestHostKeyPair)); |
EXPECT_CALL(signal_strategy_, GetState()) |
.WillOnce(Return(SignalStrategy::DISCONNECTED)); |
@@ -70,7 +71,7 @@ class HeartbeatSenderTest |
.WillRepeatedly(Return(kTestJid)); |
heartbeat_sender_.reset(new HeartbeatSender( |
- this, kHostId, &signal_strategy_, &key_pair_, kTestBotJid)); |
+ this, kHostId, &signal_strategy_, key_pair_, kTestBotJid)); |
} |
virtual void TearDown() OVERRIDE { |
@@ -84,7 +85,7 @@ class HeartbeatSenderTest |
MessageLoop message_loop_; |
MockSignalStrategy signal_strategy_; |
std::set<SignalStrategy::Listener*> signal_strategy_listeners_; |
- HostKeyPair key_pair_; |
+ scoped_refptr<RsaKeyPair> key_pair_; |
scoped_ptr<HeartbeatSender> heartbeat_sender_; |
}; |
@@ -236,10 +237,10 @@ void HeartbeatSenderTest::ValidateHeartbeatStanza( |
ASSERT_TRUE(signature != NULL); |
EXPECT_TRUE(heartbeat_stanza->NextNamed(signature_tag) == NULL); |
- HostKeyPair key_pair; |
- key_pair.LoadFromString(kTestHostKeyPair); |
+ scoped_refptr<RsaKeyPair> key_pair = new RsaKeyPair(); |
+ key_pair->LoadFromString(kTestHostKeyPair); |
std::string expected_signature = |
- key_pair.GetSignature(std::string(kTestJid) + ' ' + expectedSequenceId); |
+ key_pair->GetSignature(std::string(kTestJid) + ' ' + expectedSequenceId); |
EXPECT_EQ(expected_signature, signature->BodyText()); |
} |