123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899 |
- // Copyright 2018 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 "components/ui_devtools/devtools_server.h"
- #include "base/command_line.h"
- #include "base/run_loop.h"
- #include "base/test/task_environment.h"
- #include "build/build_config.h"
- #include "components/ui_devtools/switches.h"
- #include "mojo/public/cpp/bindings/remote.h"
- #include "net/base/address_list.h"
- #include "net/base/completion_once_callback.h"
- #include "net/base/test_completion_callback.h"
- #include "net/socket/tcp_client_socket.h"
- #include "services/network/network_context.h"
- #include "services/network/network_service.h"
- #include "services/network/public/cpp/server/http_server_request_info.h"
- #include "services/network/test/fake_test_cert_verifier_params_factory.h"
- #include "testing/gtest/include/gtest/gtest.h"
- namespace ui_devtools {
- class UIDevToolsServerTest : public testing::Test {
- public:
- UIDevToolsServerTest() {
- base::CommandLine::ForCurrentProcess()->AppendSwitch(
- switches::kEnableUiDevTools);
- }
- ~UIDevToolsServerTest() override = default;
- void SetUp() override {
- network_service_ = network::NetworkService::Create(
- network_service_remote_.BindNewPipeAndPassReceiver());
- network::mojom::NetworkContextParamsPtr context_params =
- network::mojom::NetworkContextParams::New();
- // Use a dummy CertVerifier that always passes cert verification, since
- // these unittests don't need to test CertVerifier behavior.
- context_params->cert_verifier_params =
- network::FakeTestCertVerifierParamsFactory::GetCertVerifierParams();
- network_service_remote_->CreateNetworkContext(
- network_context_remote_.BindNewPipeAndPassReceiver(),
- std::move(context_params));
- base::RunLoop run_loop;
- server_ =
- UiDevToolsServer::CreateForViews(network_context_remote_.get(), 0);
- server_->SetOnSocketConnectedForTesting(run_loop.QuitClosure());
- run_loop.Run();
- }
- protected:
- UiDevToolsServer* server() { return server_.get(); }
- base::test::TaskEnvironment task_environment{
- base::test::TaskEnvironment::MainThreadType::IO};
- std::unique_ptr<network::NetworkService> network_service_;
- mojo::Remote<network::mojom::NetworkService> network_service_remote_;
- mojo::Remote<network::mojom::NetworkContext> network_context_remote_;
- private:
- std::unique_ptr<UiDevToolsServer> server_;
- };
- // Tests whether the server for Views is properly created so we can connect to
- // it.
- TEST_F(UIDevToolsServerTest, ConnectionToViewsServer) {
- // Connect to the server socket.
- net::AddressList addr(
- net::IPEndPoint(net::IPAddress(127, 0, 0, 1), server()->port()));
- auto client_socket = std::make_unique<net::TCPClientSocket>(
- addr, nullptr, nullptr, nullptr, net::NetLogSource());
- net::TestCompletionCallback callback;
- int connect_result =
- callback.GetResult(client_socket->Connect(callback.callback()));
- ASSERT_EQ(net::OK, connect_result);
- ASSERT_TRUE(client_socket->IsConnected());
- }
- // Ensure we don't crash from OOB vector access when passed an incorrect
- // client ID.
- TEST_F(UIDevToolsServerTest, OutOfBoundsClientTest) {
- auto client =
- std::make_unique<UiDevToolsClient>("UiDevToolsClient", server());
- server()->AttachClient(std::move(client));
- // Cast for public `OnWebSocketRequest`.
- network::server::HttpServer::Delegate* devtools_server =
- static_cast<network::server::HttpServer::Delegate*>(server());
- network::server::HttpServerRequestInfo request;
- request.path = "/1";
- devtools_server->OnWebSocketRequest(0, request);
- request.path = "/42";
- devtools_server->OnWebSocketRequest(0, request);
- request.path = "/-5000";
- devtools_server->OnWebSocketRequest(0, request);
- }
- } // namespace ui_devtools
|