Add gnubby authentication to remoting host

remoting/host/gnubby_advertiser_unittest.cc

+// Copyright 2014 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 "base/test/test_simple_task_runner.h"
+#include "net/base/net_errors.h"
+#include "net/socket/socket_test_util.h"
+#include "remoting/host/gnubby_advertiser.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace remoting {
+
+class GnubbyAdvertiserTest : public testing::Test {
+ public:
+  GnubbyAdvertiserTest() {}
+
+  virtual void SetUp() OVERRIDE;
+  virtual void TearDown() OVERRIDE;
+
+ protected:
+  // Returns mock sockets.
+  scoped_ptr<net::MockClientSocketFactory> client_socket_factory_;
+
+  // GnubbyAdvertiser instance under test.
+  scoped_refptr<GnubbyAdvertiser> gnubby_advertiser_;
+};
+
+void GnubbyAdvertiserTest::SetUp() {
+  scoped_refptr<base::SingleThreadTaskRunner> task_runner =
+      new base::TestSimpleTaskRunner();
+
+  client_socket_factory_.reset(new net::MockClientSocketFactory());
+
+  gnubby_advertiser_ =
+      new GnubbyAdvertiser(task_runner, client_socket_factory_.get());
+}
+
+void GnubbyAdvertiserTest::TearDown() {
+  EXPECT_TRUE(gnubby_advertiser_->HasOneRef());
+}
+
+TEST_F(GnubbyAdvertiserTest, ConnectFailed) {
+  net::StaticSocketDataProvider socket_data(NULL, 0, NULL, 0);
+  socket_data.set_connect_data(
+      net::MockConnect(net::SYNCHRONOUS, net::ERR_FAILED));
+
+  client_socket_factory_->AddSocketDataProvider(&socket_data);
+
+  gnubby_advertiser_->Advertise(80);
+}
+
+TEST_F(GnubbyAdvertiserTest, WriteFailed) {
+  net::MockWrite write_fail(net::SYNCHRONOUS, net::ERR_FAILED);
+
+  net::StaticSocketDataProvider socket_data(NULL, 0, &write_fail, 1);
+  socket_data.set_connect_data(net::MockConnect(net::SYNCHRONOUS, net::OK));
+
+  client_socket_factory_->AddSocketDataProvider(&socket_data);
+
+  gnubby_advertiser_->Advertise(80);
+}
+
+TEST_F(GnubbyAdvertiserTest, ReadFailed) {
+  net::MockRead read_fail(net::SYNCHRONOUS, net::ERR_FAILED);
+  net::MockWrite write(net::SYNCHRONOUS, net::OK);
+
+  net::StaticSocketDataProvider socket_data(&read_fail, 1, &write, 1);
+  socket_data.set_connect_data(net::MockConnect(net::SYNCHRONOUS, net::OK));
+
+  client_socket_factory_->AddSocketDataProvider(&socket_data);
+
+  gnubby_advertiser_->Advertise(80);
+}
+
+}  // namespace remoting