// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "remoting/protocol/network_settings.h"

#include "remoting/base/port_range.h"
#include "remoting/base/session_policies.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace remoting::protocol {

TEST(NetworkSettings, CreateFromSessionPolicies_ImplicitNoRestrictions) {
  NetworkSettings settings(SessionPolicies{});

  EXPECT_EQ(settings.flags, NetworkSettings::NAT_TRAVERSAL_FULL);
  EXPECT_TRUE(settings.port_range.is_null());
}

TEST(NetworkSettings, CreateFromSessionPolicies_ExplicitNoRestrictions) {
  SessionPolicies session_policies;
  session_policies.allow_stun_connections = true;
  session_policies.allow_relayed_connections = true;

  NetworkSettings settings(session_policies);

  EXPECT_EQ(settings.flags, NetworkSettings::NAT_TRAVERSAL_FULL);
  EXPECT_TRUE(settings.port_range.is_null());
}

TEST(NetworkSettings, CreateFromSessionPolicies_RestrictedPortRange) {
  auto port_range = *PortRange::Create(100, 200);
  SessionPolicies session_policies;
  session_policies.host_udp_port_range = port_range;
  NetworkSettings settings(session_policies);

  EXPECT_EQ(settings.flags, NetworkSettings::NAT_TRAVERSAL_FULL);
  EXPECT_EQ(settings.port_range, port_range);
}

TEST(NetworkSettings, CreateFromSessionPolicies_RelayDisabled) {
  SessionPolicies session_policies;
  session_policies.allow_relayed_connections = false;

  NetworkSettings settings(session_policies);

  EXPECT_EQ(settings.flags, NetworkSettings::NAT_TRAVERSAL_OUTGOING |
                                NetworkSettings::NAT_TRAVERSAL_STUN);
  EXPECT_TRUE(settings.port_range.is_null());
}

TEST(NetworkSettings, CreateFromSessionPolicies_StunDisabled) {
  SessionPolicies session_policies;
  session_policies.allow_stun_connections = false;

  NetworkSettings settings(session_policies);

  EXPECT_EQ(settings.flags, NetworkSettings::NAT_TRAVERSAL_OUTGOING |
                                NetworkSettings::NAT_TRAVERSAL_RELAY);
  EXPECT_TRUE(settings.port_range.is_null());
}

TEST(NetworkSettings,
     CreateFromSessionPolicies_NatTraversalDisabled_DefaultPortRange) {
  SessionPolicies session_policies;
  session_policies.allow_stun_connections = false;
  session_policies.allow_relayed_connections = false;

  NetworkSettings settings(session_policies);

  EXPECT_EQ(settings.flags, NetworkSettings::NAT_TRAVERSAL_DISABLED);
  EXPECT_EQ(settings.port_range.min_port(), NetworkSettings::kDefaultMinPort);
  EXPECT_EQ(settings.port_range.max_port(), NetworkSettings::kDefaultMaxPort);
}

}  // namespace remoting::protocol