| Index: blimp/net/engine_authentication_handler_unittest.cc
|
| diff --git a/blimp/net/engine_authentication_handler_unittest.cc b/blimp/net/engine_authentication_handler_unittest.cc
|
| deleted file mode 100644
|
| index 9b186e1a14e553f264c04fb9ee4ac7336c752aba..0000000000000000000000000000000000000000
|
| --- a/blimp/net/engine_authentication_handler_unittest.cc
|
| +++ /dev/null
|
| @@ -1,167 +0,0 @@
|
| -// Copyright 2015 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 <stddef.h>
|
| -#include <memory>
|
| -#include <string>
|
| -#include <utility>
|
| -
|
| -#include "base/memory/ref_counted.h"
|
| -#include "base/test/test_mock_time_task_runner.h"
|
| -#include "base/threading/thread_task_runner_handle.h"
|
| -#include "blimp/common/create_blimp_message.h"
|
| -#include "blimp/common/proto/blimp_message.pb.h"
|
| -#include "blimp/common/protocol_version.h"
|
| -#include "blimp/net/blimp_connection.h"
|
| -#include "blimp/net/blimp_transport.h"
|
| -#include "blimp/net/common.h"
|
| -#include "blimp/net/connection_error_observer.h"
|
| -#include "blimp/net/engine_authentication_handler.h"
|
| -#include "blimp/net/test_common.h"
|
| -#include "net/base/completion_callback.h"
|
| -#include "net/base/net_errors.h"
|
| -#include "net/base/test_completion_callback.h"
|
| -#include "testing/gmock/include/gmock/gmock.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -
|
| -using testing::_;
|
| -using testing::Eq;
|
| -using testing::Return;
|
| -using testing::SaveArg;
|
| -
|
| -namespace blimp {
|
| -namespace {
|
| -const char client_auth_token[] = "valid token";
|
| -} // namespace
|
| -
|
| -class EngineAuthenticationHandlerTest : public testing::Test {
|
| - public:
|
| - EngineAuthenticationHandlerTest()
|
| - : runner_(new base::TestMockTimeTaskRunner),
|
| - runner_handle_(runner_),
|
| - auth_handler_(new EngineAuthenticationHandler(&connection_handler_,
|
| - client_auth_token)),
|
| - connection_(new testing::StrictMock<MockBlimpConnection>()) {}
|
| -
|
| - ~EngineAuthenticationHandlerTest() override {}
|
| -
|
| - protected:
|
| - void ExpectOnConnection() {
|
| - EXPECT_CALL(*connection_, AddConnectionErrorObserver(_))
|
| - .WillOnce(SaveArg<0>(&error_observer_));
|
| - EXPECT_CALL(*connection_, SetIncomingMessageProcessor(_))
|
| - .WillOnce(SaveArg<0>(&incoming_message_processor_));
|
| - }
|
| -
|
| - scoped_refptr<base::TestMockTimeTaskRunner> runner_;
|
| - base::ThreadTaskRunnerHandle runner_handle_;
|
| - testing::StrictMock<MockConnectionHandler> connection_handler_;
|
| - std::unique_ptr<EngineAuthenticationHandler> auth_handler_;
|
| - std::unique_ptr<testing::StrictMock<MockBlimpConnection>> connection_;
|
| - ConnectionErrorObserver* error_observer_ = nullptr;
|
| - BlimpMessageProcessor* incoming_message_processor_ = nullptr;
|
| -};
|
| -
|
| -TEST_F(EngineAuthenticationHandlerTest, AuthenticationSucceeds) {
|
| - ExpectOnConnection();
|
| - EXPECT_CALL(*connection_, RemoveConnectionErrorObserver(_));
|
| - EXPECT_CALL(connection_handler_, HandleConnectionPtr(Eq(connection_.get())));
|
| - auth_handler_->HandleConnection(std::move(connection_));
|
| - EXPECT_NE(nullptr, error_observer_);
|
| - EXPECT_NE(nullptr, incoming_message_processor_);
|
| -
|
| - std::unique_ptr<BlimpMessage> blimp_message =
|
| - CreateStartConnectionMessage(client_auth_token, kProtocolVersion);
|
| - net::TestCompletionCallback process_message_cb;
|
| - incoming_message_processor_->ProcessMessage(std::move(blimp_message),
|
| - process_message_cb.callback());
|
| - EXPECT_EQ(net::OK, process_message_cb.WaitForResult());
|
| -}
|
| -
|
| -TEST_F(EngineAuthenticationHandlerTest, ProtocolMismatch) {
|
| - const int kInvalidProtocolVersion = -1;
|
| -
|
| - BlimpMessage end_connection_message;
|
| - MockBlimpMessageProcessor message_processor;
|
| - EXPECT_CALL(message_processor, MockableProcessMessage(_, _))
|
| - .WillOnce(SaveArg<0>(&end_connection_message));
|
| - EXPECT_CALL(*connection_, GetOutgoingMessageProcessor())
|
| - .WillOnce(Return(&message_processor));
|
| -
|
| - ExpectOnConnection();
|
| - auth_handler_->HandleConnection(std::move(connection_));
|
| -
|
| - std::unique_ptr<BlimpMessage> blimp_message =
|
| - CreateStartConnectionMessage(client_auth_token, kInvalidProtocolVersion);
|
| - net::TestCompletionCallback process_message_cb;
|
| - incoming_message_processor_->ProcessMessage(std::move(blimp_message),
|
| - process_message_cb.callback());
|
| - EXPECT_EQ(net::OK, process_message_cb.WaitForResult());
|
| -
|
| - EXPECT_EQ(BlimpMessage::kProtocolControl,
|
| - end_connection_message.feature_case());
|
| - EXPECT_EQ(
|
| - ProtocolControlMessage::kEndConnection,
|
| - end_connection_message.protocol_control().connection_message_case());
|
| - EXPECT_EQ(
|
| - EndConnectionMessage::PROTOCOL_MISMATCH,
|
| - end_connection_message.protocol_control().end_connection().reason());
|
| -}
|
| -
|
| -TEST_F(EngineAuthenticationHandlerTest, AuthenticationFailed) {
|
| - ExpectOnConnection();
|
| - auth_handler_->HandleConnection(std::move(connection_));
|
| -
|
| - std::unique_ptr<BlimpMessage> blimp_message =
|
| - CreateStartConnectionMessage("invalid token", kProtocolVersion);
|
| - net::TestCompletionCallback process_message_cb;
|
| - incoming_message_processor_->ProcessMessage(std::move(blimp_message),
|
| - process_message_cb.callback());
|
| - EXPECT_EQ(net::OK, process_message_cb.WaitForResult());
|
| -}
|
| -
|
| -TEST_F(EngineAuthenticationHandlerTest, WrongMessageReceived) {
|
| - ExpectOnConnection();
|
| - auth_handler_->HandleConnection(std::move(connection_));
|
| -
|
| - InputMessage* input_message;
|
| - std::unique_ptr<BlimpMessage> blimp_message =
|
| - CreateBlimpMessage(&input_message);
|
| - net::TestCompletionCallback process_message_cb;
|
| - incoming_message_processor_->ProcessMessage(std::move(blimp_message),
|
| - process_message_cb.callback());
|
| - EXPECT_EQ(net::OK, process_message_cb.WaitForResult());
|
| -}
|
| -
|
| -TEST_F(EngineAuthenticationHandlerTest, ConnectionError) {
|
| - ExpectOnConnection();
|
| - auth_handler_->HandleConnection(std::move(connection_));
|
| - EXPECT_NE(nullptr, error_observer_);
|
| - EXPECT_NE(nullptr, incoming_message_processor_);
|
| - error_observer_->OnConnectionError(net::ERR_FAILED);
|
| -}
|
| -
|
| -TEST_F(EngineAuthenticationHandlerTest, Timeout) {
|
| - ExpectOnConnection();
|
| - auth_handler_->HandleConnection(std::move(connection_));
|
| - EXPECT_NE(nullptr, error_observer_);
|
| - EXPECT_NE(nullptr, incoming_message_processor_);
|
| -
|
| - runner_->FastForwardBy(base::TimeDelta::FromSeconds(11));
|
| -}
|
| -
|
| -TEST_F(EngineAuthenticationHandlerTest, AuthHandlerDeletedFirst) {
|
| - ExpectOnConnection();
|
| - auth_handler_->HandleConnection(std::move(connection_));
|
| - auth_handler_.reset();
|
| -
|
| - std::unique_ptr<BlimpMessage> blimp_message =
|
| - CreateStartConnectionMessage(client_auth_token, kProtocolVersion);
|
| - net::TestCompletionCallback process_message_cb;
|
| - incoming_message_processor_->ProcessMessage(std::move(blimp_message),
|
| - process_message_cb.callback());
|
| - EXPECT_EQ(net::OK, process_message_cb.WaitForResult());
|
| -}
|
| -
|
| -} // namespace blimp
|
|
|