Separate Authenticator and Session unittests.

remoting/protocol/fake_authenticator.cc

+// Copyright (c) 2011 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/protocol/fake_authenticator.h"
+
+#include "base/string_number_conversions.h"
+#include "net/socket/stream_socket.h"
+#include "remoting/base/constants.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/libjingle/source/talk/xmllite/xmlelement.h"
+
+namespace remoting {
+namespace protocol {
+
+FakeChannelAuthenticator::FakeChannelAuthenticator(bool accept)
+    : accept_(accept) {
+}
+
+FakeChannelAuthenticator::~FakeChannelAuthenticator() {
+}
+
+void FakeChannelAuthenticator::SecureAndAuthenticate(
+    net::StreamSocket* socket, const DoneCallback& done_callback){
+  if (accept_) {
+    done_callback.Run(net::OK, socket);

[Wez, 2011/12/09 23:42:33]

The ChannelAuthenticator allows asynchronous completion, and I think our current
implementation completes asynchronously, so this implementation should really
PostTask() completion, or better still test both the synchronous and
asynchronous completion cases.

[Sergey Ulanov, 2011/12/12 22:52:00]

Done.

+  } else {
+    delete socket;
+    done_callback.Run(net::ERR_FAILED, NULL);
+  }
+}
+
+FakeAuthenticator::FakeAuthenticator(
+    Type type, Action action, int round_trips)
+    : type_(type),
+      action_(action),
+      round_trips_(round_trips),
+      messages_(0) {
+}
+
+FakeAuthenticator::~FakeAuthenticator() {
+}
+
+Authenticator::State FakeAuthenticator::state() const{
+  EXPECT_LE(messages_, round_trips_ * 2);
+  if (messages_ >= round_trips_ * 2) {
+    if (action_ == REJECT) {
+      return REJECTED;
+    } else {
+      return ACCEPTED;
+    }
+  }
+
+  // Don't send the last message if this is a host hat wants to reject

[Wez, 2011/12/09 23:42:33]

typo: hat -> that

[Sergey Ulanov, 2011/12/12 22:52:00]

Done.

+  // a connection.
+  if (messages_ == round_trips_ * 2 - 1 &&
+      type_ == HOST && action_ == REJECT) {
+    return REJECTED;
+  }
+
+  // We are not done yet. process next message.
+  if ((messages_ % 2 == 0 && type_ == CLIENT) ||
+      (messages_ % 2 == 1 && type_ == HOST)) {
+    return MESSAGE_READY;
+  } else {
+    return WAITING_MESSAGE;
+  }
+}
+
+void FakeAuthenticator::ProcessMessage(const buzz::XmlElement* message) {
+  EXPECT_EQ(WAITING_MESSAGE, state());
+  std::string id =
+      message->TextNamed(buzz::QName(kChromotingXmlNamespace, "id"));
+  EXPECT_EQ(id, base::IntToString(messages_));
+  ++messages_;
+}
+
+buzz::XmlElement* FakeAuthenticator::GetNextMessage() {
+  EXPECT_EQ(MESSAGE_READY, state());
+
+  buzz::XmlElement* result = new buzz::XmlElement(
+      buzz::QName(kChromotingXmlNamespace, "authentication"));
+  buzz::XmlElement* id = new buzz::XmlElement(
+      buzz::QName(kChromotingXmlNamespace, "id"));
+  id->AddText(base::IntToString(messages_));
+  result->AddElement(id);
+
+  ++messages_;
+  return result;
+}
+
+ChannelAuthenticator*
+FakeAuthenticator::CreateChannelAuthenticator() const {
+  EXPECT_EQ(ACCEPTED, state());
+  return new FakeChannelAuthenticator(action_ != REJECT_CHANNEL);
+}
+
+FakeHostAuthenticatorFactory::FakeHostAuthenticatorFactory(
+    FakeAuthenticator::Action action, int round_trips)
+    : action_(action),
+      round_trips_(round_trips) {
+}
+
+FakeHostAuthenticatorFactory::~FakeHostAuthenticatorFactory() {
+}
+
+Authenticator* FakeHostAuthenticatorFactory::CreateAuthenticator(
+    const std::string& remote_jid,
+    const buzz::XmlElement* first_message) {
+  return new FakeAuthenticator(FakeAuthenticator::HOST, action_, round_trips_);
+}
+
+}  // namespace protocol
+}  // namespace remoting