content/browser/direct_sockets/direct_sockets_unittest.cc

// Copyright 2020 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 <memory>

#include "base/test/scoped_feature_list.h"
#include "content/browser/direct_sockets/direct_sockets_service_impl.h"
#include "content/public/common/content_features.h"
#include "content/public/test/test_renderer_host.h"
#include "net/base/ip_address.h"
#include "net/base/net_errors.h"
#include "third_party/blink/public/mojom/direct_sockets/direct_sockets.mojom.h"

namespace content {

class DirectSocketsUnitTest : public RenderViewHostTestHarness {
 public:
  DirectSocketsUnitTest() {
    feature_list_.InitAndEnableFeature(features::kDirectSockets);
  }
  ~DirectSocketsUnitTest() override = default;

  void SetUp() override {
    RenderViewHostTestHarness::SetUp();
    direct_sockets_service_ =
        std::make_unique<DirectSocketsServiceImpl>(*main_rfh());
  }

  DirectSocketsServiceImpl& direct_sockets_service() {
    return *direct_sockets_service_;
  }

  net::Error ValidateOptions(const blink::mojom::DirectSocketOptions& options) {
    net::IPAddress remote_address;
    return direct_sockets_service().ValidateOptions(options, remote_address);
  }

 private:
  base::test::ScopedFeatureList feature_list_;
  std::unique_ptr<DirectSocketsServiceImpl> direct_sockets_service_;
};

TEST_F(DirectSocketsUnitTest, RenderFrameDeleted) {
  direct_sockets_service().RenderFrameDeleted(main_rfh());

  blink::mojom::DirectSocketOptions options;
  EXPECT_EQ(ValidateOptions(options), net::ERR_CONTEXT_SHUT_DOWN);
}

TEST_F(DirectSocketsUnitTest, WebContentsDestroyed) {
  direct_sockets_service().WebContentsDestroyed();

  blink::mojom::DirectSocketOptions options;
  EXPECT_EQ(ValidateOptions(options), net::ERR_CONTEXT_SHUT_DOWN);
}

// TODO(crbug.com/1119597): Allow the user to enter the address.
TEST_F(DirectSocketsUnitTest, RemoteAddressCurrentlyRequired) {
  blink::mojom::DirectSocketOptions options;
  EXPECT_EQ(ValidateOptions(options), net::ERR_NAME_NOT_RESOLVED);
}

}  // namespace content