| Index: extensions/browser/api/cast_channel/cast_test_util.h
|
| diff --git a/extensions/browser/api/cast_channel/cast_test_util.h b/extensions/browser/api/cast_channel/cast_test_util.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..f70af01ff3e21eb66273a160d6304871e9bfe9b9
|
| --- /dev/null
|
| +++ b/extensions/browser/api/cast_channel/cast_test_util.h
|
| @@ -0,0 +1,127 @@
|
| +// 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.
|
| +
|
| +#ifndef EXTENSIONS_BROWSER_API_CAST_CHANNEL_CAST_TEST_UTIL_H_
|
| +#define EXTENSIONS_BROWSER_API_CAST_CHANNEL_CAST_TEST_UTIL_H_
|
| +
|
| +#include <string>
|
| +#include <utility>
|
| +
|
| +#include "base/macros.h"
|
| +#include "base/threading/thread_task_runner_handle.h"
|
| +#include "extensions/browser/api/cast_channel/cast_socket.h"
|
| +#include "extensions/browser/api/cast_channel/cast_transport.h"
|
| +#include "extensions/common/api/cast_channel/cast_channel.pb.h"
|
| +#include "net/base/ip_endpoint.h"
|
| +#include "testing/gmock/include/gmock/gmock.h"
|
| +
|
| +namespace extensions {
|
| +namespace api {
|
| +namespace cast_channel {
|
| +
|
| +extern const char kTestExtensionId[];
|
| +
|
| +class MockCastTransport : public extensions::api::cast_channel::CastTransport {
|
| + public:
|
| + MockCastTransport();
|
| + ~MockCastTransport() override;
|
| +
|
| + void SetReadDelegate(
|
| + std::unique_ptr<CastTransport::Delegate> delegate) override;
|
| +
|
| + MOCK_METHOD2(SendMessage,
|
| + void(const extensions::api::cast_channel::CastMessage& message,
|
| + const net::CompletionCallback& callback));
|
| +
|
| + MOCK_METHOD0(Start, void(void));
|
| +
|
| + // Gets the read delegate that is currently active for this transport.
|
| + CastTransport::Delegate* current_delegate() const;
|
| +
|
| + private:
|
| + std::unique_ptr<CastTransport::Delegate> delegate_;
|
| +
|
| + DISALLOW_COPY_AND_ASSIGN(MockCastTransport);
|
| +};
|
| +
|
| +class MockCastTransportDelegate : public CastTransport::Delegate {
|
| + public:
|
| + MockCastTransportDelegate();
|
| + ~MockCastTransportDelegate() override;
|
| +
|
| + MOCK_METHOD1(OnError, void(ChannelError error));
|
| + MOCK_METHOD1(OnMessage, void(const CastMessage& message));
|
| + MOCK_METHOD0(Start, void(void));
|
| +
|
| + private:
|
| + DISALLOW_COPY_AND_ASSIGN(MockCastTransportDelegate);
|
| +};
|
| +
|
| +class MockCastSocket : public CastSocket {
|
| + public:
|
| + MockCastSocket();
|
| + ~MockCastSocket() override;
|
| +
|
| + // Mockable version of Connect. Accepts a bare pointer to a mock object.
|
| + // (GMock won't compile with scoped_ptr method parameters.)
|
| + MOCK_METHOD2(ConnectRawPtr,
|
| + void(CastTransport::Delegate* delegate,
|
| + base::Callback<void(ChannelError)> callback));
|
| +
|
| + // Proxy for ConnectRawPtr. Unpacks scoped_ptr into a GMock-friendly bare
|
| + // ptr.
|
| + void Connect(std::unique_ptr<CastTransport::Delegate> delegate,
|
| + base::Callback<void(ChannelError)> callback) override {
|
| + delegate_ = std::move(delegate);
|
| + ConnectRawPtr(delegate_.get(), callback);
|
| + }
|
| +
|
| + MOCK_METHOD1(Close, void(const net::CompletionCallback& callback));
|
| + MOCK_CONST_METHOD0(ip_endpoint, const net::IPEndPoint&());
|
| + MOCK_CONST_METHOD0(id, int());
|
| + MOCK_METHOD1(set_id, void(int id));
|
| + MOCK_CONST_METHOD0(channel_auth, ChannelAuthType());
|
| + MOCK_CONST_METHOD0(ready_state, ReadyState());
|
| + MOCK_CONST_METHOD0(error_state, ChannelError());
|
| + MOCK_CONST_METHOD0(keep_alive, bool(void));
|
| + MOCK_CONST_METHOD0(audio_only, bool(void));
|
| + MOCK_METHOD1(SetErrorState, void(ChannelError error_state));
|
| +
|
| + CastTransport* transport() const override { return mock_transport_.get(); }
|
| +
|
| + MockCastTransport* mock_transport() const { return mock_transport_.get(); }
|
| +
|
| + private:
|
| + std::unique_ptr<MockCastTransport> mock_transport_;
|
| + std::unique_ptr<CastTransport::Delegate> delegate_;
|
| +
|
| + DISALLOW_COPY_AND_ASSIGN(MockCastSocket);
|
| +};
|
| +
|
| +// Creates the IPEndpoint 192.168.1.1.
|
| +net::IPEndPoint CreateIPEndPointForTest();
|
| +
|
| +// Checks if two proto messages are the same.
|
| +// From
|
| +// third_party/cacheinvalidation/overrides/google/cacheinvalidation/deps/gmock.h
|
| +// TODO(kmarshall): promote to a shared testing library.
|
| +MATCHER_P(EqualsProto, message, "") {
|
| + std::string expected_serialized, actual_serialized;
|
| + message.SerializeToString(&expected_serialized);
|
| + arg.SerializeToString(&actual_serialized);
|
| + return expected_serialized == actual_serialized;
|
| +}
|
| +
|
| +ACTION_TEMPLATE(PostCompletionCallbackTask,
|
| + HAS_1_TEMPLATE_PARAMS(int, cb_idx),
|
| + AND_1_VALUE_PARAMS(rv)) {
|
| + base::ThreadTaskRunnerHandle::Get()->PostTask(
|
| + FROM_HERE, base::Bind(testing::get<cb_idx>(args), rv));
|
| +}
|
| +
|
| +} // namespace cast_channel
|
| +} // namespace api
|
| +} // namespace extensions
|
| +
|
| +#endif // EXTENSIONS_BROWSER_API_CAST_CHANNEL_CAST_TEST_UTIL_H_
|
|
|