1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77
|
// 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) {
PortRange port_range{.min_port = 100, .max_port = 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
|