ice/agent_config_test.go at main · pion/ice · GitHub

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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
// SPDX-FileCopyrightText: 2026 The Pion community <https://pion.ly>
// SPDX-License-Identifier: MIT

package ice

import (
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func TestAgentConfig_initWithDefaults(t *testing.T) {
	relayAcceptanceMinWait := 5 * time.Second
	tests := []struct {
		name   string
		config *AgentConfig
		fn     func(*testing.T, *Agent)
	}{
		{
			"default config",
			&AgentConfig{},
			func(t *testing.T, result *Agent) {
				t.Helper()
				assert.Equal(t, result.relayAcceptanceMinWait, defaultRelayAcceptanceMinWait)
			},
		},
		{
			"multiple relay candidate types",
			&AgentConfig{CandidateTypes: []CandidateType{CandidateTypeHost, CandidateTypeServerReflexive}},
			func(t *testing.T, result *Agent) {
				t.Helper()
				assert.Equal(t, result.relayAcceptanceMinWait, defaultRelayAcceptanceMinWait)
			},
		},
		{
			"host only candidate type",
			&AgentConfig{CandidateTypes: []CandidateType{CandidateTypeHost}},
			func(t *testing.T, result *Agent) {
				t.Helper()
				assert.Equal(t, result.relayAcceptanceMinWait, defaultRelayAcceptanceMinWait)
			},
		},
		{
			"relay only candidate type",
			&AgentConfig{CandidateTypes: []CandidateType{CandidateTypeRelay}},
			func(t *testing.T, result *Agent) {
				t.Helper()
				assert.Equal(t, result.relayAcceptanceMinWait, defaultRelayOnlyAcceptanceMinWait)
			},
		},
		{
			"relay only with relayAcceptanceMinWait set",
			&AgentConfig{CandidateTypes: []CandidateType{CandidateTypeRelay}, RelayAcceptanceMinWait: &relayAcceptanceMinWait},
			func(t *testing.T, result *Agent) {
				t.Helper()
				assert.Equal(t, result.relayAcceptanceMinWait, relayAcceptanceMinWait)
			},
		},
	}

	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			agent, err := NewAgent(test.config)
			if !assert.NoError(t, err) {
				return
			}
			defer func() { _ = agent.Close() }()
			test.fn(t, agent)
		})
	}
}

func TestDefaultRelayAcceptanceMinWaitForCandidates(t *testing.T) {
	tests := []struct {
		name          string
		candidateType []CandidateType
		expectedWait  time.Duration
	}{
		{
			name:          "relay only",
			candidateType: []CandidateType{CandidateTypeRelay},
			expectedWait:  defaultRelayOnlyAcceptanceMinWait,
		},
		{
			name:          "mixed types",
			candidateType: []CandidateType{CandidateTypeHost, CandidateTypeRelay},
			expectedWait:  defaultRelayAcceptanceMinWait,
		},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			assert.Equal(t, tc.expectedWait, defaultRelayAcceptanceMinWaitFor(tc.candidateType))
		})
	}
}