gluetun/internal/models/server_test.go at 4ed91527d720027413b42222bff8f8ac4c6fc78d · passteque/gluetun · 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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
package models

import (
	"net/netip"
	"testing"

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

func Test_Server_Equal(t *testing.T) {
	t.Parallel()

	testCases := map[string]struct {
		a     *Server
		b     Server
		equal bool
	}{
		"same IPs": {
			a: &Server{
				IPs: []netip.Addr{netip.AddrFrom4([4]byte{1, 2, 3, 4})},
			},
			b: Server{
				IPs: []netip.Addr{netip.AddrFrom4([4]byte{1, 2, 3, 4})},
			},
			equal: true,
		},
		"same IP strings": {
			a: &Server{
				IPs: []netip.Addr{netip.AddrFrom4([4]byte{1, 2, 3, 4})},
			},
			b: Server{
				IPs: []netip.Addr{netip.AddrFrom4([4]byte{1, 2, 3, 4})},
			},
			equal: true,
		},
		"different IPs": {
			a: &Server{
				IPs: []netip.Addr{netip.AddrFrom4([4]byte{1, 2, 3, 4}), netip.AddrFrom4([4]byte{2, 3, 4, 5})},
			},
			b: Server{
				IPs: []netip.Addr{netip.AddrFrom4([4]byte{1, 2, 3, 4}), netip.AddrFrom4([4]byte{1, 2, 3, 4})},
			},
		},
		"all fields equal": {
			a: &Server{
				VPN:              "vpn",
				Country:          "country",
				Region:           "region",
				City:             "city",
				ISP:              "isp",
				Owned:            true,
				Number:           1,
				ServerName:       "server_name",
				Hostname:         "hostname",
				TCP:              true,
				UDP:              true,
				OvpnX509:         "x509",
				RetroLoc:         "retroloc",
				MultiHop:         true,
				WgPubKey:         "wgpubkey",
				Free:             true,
				Stream:           true,
				SecureCore:       true,
				Tor:              true,
				PortForward:      true,
				QuantumResistant: true,
				Obfuscated:       true,
				IPs:              []netip.Addr{netip.AddrFrom4([4]byte{1, 2, 3, 4})},
				Keep:             true,
			},
			b: Server{
				VPN:              "vpn",
				Country:          "country",
				Region:           "region",
				City:             "city",
				ISP:              "isp",
				Owned:            true,
				Number:           1,
				ServerName:       "server_name",
				Hostname:         "hostname",
				TCP:              true,
				UDP:              true,
				OvpnX509:         "x509",
				RetroLoc:         "retroloc",
				MultiHop:         true,
				WgPubKey:         "wgpubkey",
				Free:             true,
				Stream:           true,
				SecureCore:       true,
				Tor:              true,
				PortForward:      true,
				QuantumResistant: true,
				Obfuscated:       true,
				IPs:              []netip.Addr{netip.AddrFrom4([4]byte{1, 2, 3, 4})},
				Keep:             true,
			},
			equal: true,
		},
		"different field": {
			a: &Server{
				VPN: "vpn",
			},
			b: Server{
				VPN: "other vpn",
			},
		},
	}

	for name, testCase := range testCases {
		t.Run(name, func(t *testing.T) {
			t.Parallel()

			ipsOfANotNil := testCase.a.IPs != nil
			ipsOfBNotNil := testCase.b.IPs != nil

			equal := testCase.a.Equal(testCase.b)

			assert.Equal(t, testCase.equal, equal)

			// Ensure IPs field is not modified
			if ipsOfANotNil {
				assert.NotNil(t, testCase.a)
			}
			if ipsOfBNotNil {
				assert.NotNil(t, testCase.b)
			}
		})
	}
}