Tommi | f7b22c6 | 2024-02-15 08:18:34 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2024 The WebRTC Project Authors. All rights reserved. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license |
| 5 | * that can be found in the LICENSE file in the root of the source |
| 6 | * tree. An additional intellectual property rights grant can be found |
| 7 | * in the file PATENTS. All contributing project authors may |
| 8 | * be found in the AUTHORS file in the root of the source tree. |
| 9 | */ |
| 10 | |
| 11 | #include "api/candidate.h" |
| 12 | |
| 13 | #include <string> |
| 14 | |
| 15 | #include "p2p/base/p2p_constants.h" |
| 16 | #include "rtc_base/gunit.h" |
| 17 | |
Tommi | c7a4b2a | 2024-02-25 22:34:39 | [diff] [blame] | 18 | using webrtc::IceCandidateType; |
| 19 | |
Tommi | f7b22c6 | 2024-02-15 08:18:34 | [diff] [blame] | 20 | namespace cricket { |
| 21 | |
| 22 | TEST(CandidateTest, Id) { |
| 23 | Candidate c; |
| 24 | EXPECT_EQ(c.id().size(), 8u); |
Tommi | bde80e3 | 2024-02-17 14:52:57 | [diff] [blame] | 25 | std::string current_id = c.id(); |
| 26 | // Generate a new ID. |
| 27 | c.generate_id(); |
| 28 | EXPECT_EQ(c.id().size(), 8u); |
| 29 | EXPECT_NE(current_id, c.id()); |
Tommi | f7b22c6 | 2024-02-15 08:18:34 | [diff] [blame] | 30 | } |
| 31 | |
| 32 | TEST(CandidateTest, Component) { |
| 33 | Candidate c; |
Tommi | f7b22c6 | 2024-02-15 08:18:34 | [diff] [blame] | 34 | EXPECT_EQ(c.component(), ICE_CANDIDATE_COMPONENT_DEFAULT); |
Tommi | f2431a9 | 2024-02-20 14:27:44 | [diff] [blame] | 35 | c.set_component(ICE_CANDIDATE_COMPONENT_RTCP); |
| 36 | EXPECT_EQ(c.component(), ICE_CANDIDATE_COMPONENT_RTCP); |
Tommi | f7b22c6 | 2024-02-15 08:18:34 | [diff] [blame] | 37 | } |
| 38 | |
| 39 | TEST(CandidateTest, TypeName) { |
| 40 | Candidate c; |
| 41 | // The `type_name()` property defaults to "host". |
| 42 | EXPECT_EQ(c.type_name(), "host"); |
Tommi | c7a4b2a | 2024-02-25 22:34:39 | [diff] [blame] | 43 | EXPECT_EQ(c.type(), IceCandidateType::kHost); |
Tommi | f7b22c6 | 2024-02-15 08:18:34 | [diff] [blame] | 44 | |
Tommi | c7a4b2a | 2024-02-25 22:34:39 | [diff] [blame] | 45 | c.set_type(IceCandidateType::kSrflx); |
Tommi | f7b22c6 | 2024-02-15 08:18:34 | [diff] [blame] | 46 | EXPECT_EQ(c.type_name(), "srflx"); |
Tommi | c7a4b2a | 2024-02-25 22:34:39 | [diff] [blame] | 47 | EXPECT_EQ(c.type(), IceCandidateType::kSrflx); |
Tommi | f7b22c6 | 2024-02-15 08:18:34 | [diff] [blame] | 48 | |
Tommi | c7a4b2a | 2024-02-25 22:34:39 | [diff] [blame] | 49 | c.set_type(IceCandidateType::kPrflx); |
Tommi | f7b22c6 | 2024-02-15 08:18:34 | [diff] [blame] | 50 | EXPECT_EQ(c.type_name(), "prflx"); |
Tommi | c7a4b2a | 2024-02-25 22:34:39 | [diff] [blame] | 51 | EXPECT_EQ(c.type(), IceCandidateType::kPrflx); |
Tommi | f7b22c6 | 2024-02-15 08:18:34 | [diff] [blame] | 52 | |
Tommi | c7a4b2a | 2024-02-25 22:34:39 | [diff] [blame] | 53 | c.set_type(IceCandidateType::kRelay); |
Tommi | f7b22c6 | 2024-02-15 08:18:34 | [diff] [blame] | 54 | EXPECT_EQ(c.type_name(), "relay"); |
Tommi | c7a4b2a | 2024-02-25 22:34:39 | [diff] [blame] | 55 | EXPECT_EQ(c.type(), IceCandidateType::kRelay); |
Tommi | f7b22c6 | 2024-02-15 08:18:34 | [diff] [blame] | 56 | } |
| 57 | |
Tommi | f2431a9 | 2024-02-20 14:27:44 | [diff] [blame] | 58 | TEST(CandidateTest, Foundation) { |
| 59 | Candidate c; |
| 60 | EXPECT_TRUE(c.foundation().empty()); |
| 61 | c.set_protocol("udp"); |
| 62 | c.set_relay_protocol("udp"); |
| 63 | |
| 64 | rtc::SocketAddress address("99.99.98.1", 1024); |
| 65 | c.set_address(address); |
| 66 | c.ComputeFoundation(c.address(), 1); |
| 67 | std::string foundation1 = c.foundation(); |
| 68 | EXPECT_FALSE(foundation1.empty()); |
| 69 | |
| 70 | // Change the tiebreaker. |
| 71 | c.ComputeFoundation(c.address(), 2); |
| 72 | std::string foundation2 = c.foundation(); |
| 73 | EXPECT_NE(foundation1, foundation2); |
| 74 | |
| 75 | // Provide a different base address. |
| 76 | address.SetIP("100.100.100.1"); |
| 77 | c.ComputeFoundation(address, 1); // Same tiebreaker as for foundation1. |
| 78 | foundation2 = c.foundation(); |
| 79 | EXPECT_NE(foundation1, foundation2); |
| 80 | |
| 81 | // Consistency check (just in case the algorithm ever changes to random!). |
| 82 | c.ComputeFoundation(c.address(), 1); |
| 83 | foundation2 = c.foundation(); |
| 84 | EXPECT_EQ(foundation1, foundation2); |
| 85 | |
| 86 | // Changing the protocol should affect the foundation. |
| 87 | auto prev_protocol = c.protocol(); |
| 88 | c.set_protocol("tcp"); |
| 89 | ASSERT_NE(prev_protocol, c.protocol()); |
| 90 | c.ComputeFoundation(c.address(), 1); |
| 91 | EXPECT_NE(foundation1, c.foundation()); |
| 92 | c.set_protocol(prev_protocol); |
| 93 | |
| 94 | // Changing the relay protocol should affect the foundation. |
| 95 | prev_protocol = c.relay_protocol(); |
| 96 | c.set_relay_protocol("tcp"); |
| 97 | ASSERT_NE(prev_protocol, c.relay_protocol()); |
| 98 | c.ComputeFoundation(c.address(), 1); |
| 99 | EXPECT_NE(foundation1, c.foundation()); |
| 100 | } |
| 101 | |
Tommi | f7b22c6 | 2024-02-15 08:18:34 | [diff] [blame] | 102 | } // namespace cricket |