devtools_server_unittest.cc 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899
  1. // Copyright 2018 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "components/ui_devtools/devtools_server.h"
  5. #include "base/command_line.h"
  6. #include "base/run_loop.h"
  7. #include "base/test/task_environment.h"
  8. #include "build/build_config.h"
  9. #include "components/ui_devtools/switches.h"
  10. #include "mojo/public/cpp/bindings/remote.h"
  11. #include "net/base/address_list.h"
  12. #include "net/base/completion_once_callback.h"
  13. #include "net/base/test_completion_callback.h"
  14. #include "net/socket/tcp_client_socket.h"
  15. #include "services/network/network_context.h"
  16. #include "services/network/network_service.h"
  17. #include "services/network/public/cpp/server/http_server_request_info.h"
  18. #include "services/network/test/fake_test_cert_verifier_params_factory.h"
  19. #include "testing/gtest/include/gtest/gtest.h"
  20. namespace ui_devtools {
  21. class UIDevToolsServerTest : public testing::Test {
  22. public:
  23. UIDevToolsServerTest() {
  24. base::CommandLine::ForCurrentProcess()->AppendSwitch(
  25. switches::kEnableUiDevTools);
  26. }
  27. ~UIDevToolsServerTest() override = default;
  28. void SetUp() override {
  29. network_service_ = network::NetworkService::Create(
  30. network_service_remote_.BindNewPipeAndPassReceiver());
  31. network::mojom::NetworkContextParamsPtr context_params =
  32. network::mojom::NetworkContextParams::New();
  33. // Use a dummy CertVerifier that always passes cert verification, since
  34. // these unittests don't need to test CertVerifier behavior.
  35. context_params->cert_verifier_params =
  36. network::FakeTestCertVerifierParamsFactory::GetCertVerifierParams();
  37. network_service_remote_->CreateNetworkContext(
  38. network_context_remote_.BindNewPipeAndPassReceiver(),
  39. std::move(context_params));
  40. base::RunLoop run_loop;
  41. server_ =
  42. UiDevToolsServer::CreateForViews(network_context_remote_.get(), 0);
  43. server_->SetOnSocketConnectedForTesting(run_loop.QuitClosure());
  44. run_loop.Run();
  45. }
  46. protected:
  47. UiDevToolsServer* server() { return server_.get(); }
  48. base::test::TaskEnvironment task_environment{
  49. base::test::TaskEnvironment::MainThreadType::IO};
  50. std::unique_ptr<network::NetworkService> network_service_;
  51. mojo::Remote<network::mojom::NetworkService> network_service_remote_;
  52. mojo::Remote<network::mojom::NetworkContext> network_context_remote_;
  53. private:
  54. std::unique_ptr<UiDevToolsServer> server_;
  55. };
  56. // Tests whether the server for Views is properly created so we can connect to
  57. // it.
  58. TEST_F(UIDevToolsServerTest, ConnectionToViewsServer) {
  59. // Connect to the server socket.
  60. net::AddressList addr(
  61. net::IPEndPoint(net::IPAddress(127, 0, 0, 1), server()->port()));
  62. auto client_socket = std::make_unique<net::TCPClientSocket>(
  63. addr, nullptr, nullptr, nullptr, net::NetLogSource());
  64. net::TestCompletionCallback callback;
  65. int connect_result =
  66. callback.GetResult(client_socket->Connect(callback.callback()));
  67. ASSERT_EQ(net::OK, connect_result);
  68. ASSERT_TRUE(client_socket->IsConnected());
  69. }
  70. // Ensure we don't crash from OOB vector access when passed an incorrect
  71. // client ID.
  72. TEST_F(UIDevToolsServerTest, OutOfBoundsClientTest) {
  73. auto client =
  74. std::make_unique<UiDevToolsClient>("UiDevToolsClient", server());
  75. server()->AttachClient(std::move(client));
  76. // Cast for public `OnWebSocketRequest`.
  77. network::server::HttpServer::Delegate* devtools_server =
  78. static_cast<network::server::HttpServer::Delegate*>(server());
  79. network::server::HttpServerRequestInfo request;
  80. request.path = "/1";
  81. devtools_server->OnWebSocketRequest(0, request);
  82. request.path = "/42";
  83. devtools_server->OnWebSocketRequest(0, request);
  84. request.path = "/-5000";
  85. devtools_server->OnWebSocketRequest(0, request);
  86. }
  87. } // namespace ui_devtools