Chromoting: stopping the service if the host ID is permanently not recognized by the could.

remoting/host/heartbeat_sender_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "remoting/host/heartbeat_sender.h"
 
 #include <set>
 
 #include "base/memory/ref_counted.h"
 #include "base/message_loop.h"
@@ skipping 29 matching lines @@
 }  // namespace
 
 ACTION_P(AddListener, list) {
   list->insert(arg0);
 }
 ACTION_P(RemoveListener, list) {
   EXPECT_TRUE(list->find(arg0) != list->end());
   list->erase(arg0);
 }
 
-class HeartbeatSenderTest : public testing::Test {
+class HeartbeatSenderTest
+    : public testing::Test,
+      public HeartbeatSender::Delegate {
  protected:
+  // Overridden from HeartbeatSender::Delegate
+  virtual void OnUnknownHostIdError() OVERRIDE {
+    NOTREACHED();
+  }
+
   virtual void SetUp() OVERRIDE {
     ASSERT_TRUE(key_pair_.LoadFromString(kTestHostKeyPair));
 
     EXPECT_CALL(signal_strategy_, GetState())
         .WillOnce(Return(SignalStrategy::DISCONNECTED));
     EXPECT_CALL(signal_strategy_, AddListener(NotNull()))
         .WillRepeatedly(AddListener(&signal_strategy_listeners_));
     EXPECT_CALL(signal_strategy_, RemoveListener(NotNull()))
         .WillRepeatedly(RemoveListener(&signal_strategy_listeners_));
     EXPECT_CALL(signal_strategy_, GetLocalJid())
         .WillRepeatedly(Return(kTestJid));
 
     heartbeat_sender_.reset(
-        new HeartbeatSender(kHostId, &signal_strategy_, &key_pair_));
+        new HeartbeatSender(this, kHostId, &signal_strategy_, &key_pair_));
   }
 
   virtual void TearDown() OVERRIDE {
     heartbeat_sender_.reset();
     EXPECT_TRUE(signal_strategy_listeners_.empty());
   }
 
   void ValidateHeartbeatStanza(XmlElement* stanza,
                                const char* expectedSequenceId);
 
@@ skipping 153 matching lines @@
   EXPECT_TRUE(heartbeat_stanza->NextNamed(signature_tag) == NULL);
 
   HostKeyPair key_pair;
   key_pair.LoadFromString(kTestHostKeyPair);
   std::string expected_signature =
       key_pair.GetSignature(std::string(kTestJid) + ' ' + expectedSequenceId);
   EXPECT_EQ(expected_signature, signature->BodyText());
 }
 
 }  // namespace remoting