42 lines
808 B
Go
42 lines
808 B
Go
package updater
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_uniqueSortedIPs(t *testing.T) {
|
|
t.Parallel()
|
|
testCases := map[string]struct {
|
|
inputIPs []net.IP
|
|
outputIPs []net.IP
|
|
}{
|
|
"nil": {
|
|
inputIPs: nil,
|
|
outputIPs: []net.IP{},
|
|
},
|
|
"empty": {
|
|
inputIPs: []net.IP{},
|
|
outputIPs: []net.IP{},
|
|
},
|
|
"single IPv4": {
|
|
inputIPs: []net.IP{{1, 1, 1, 1}},
|
|
outputIPs: []net.IP{{1, 1, 1, 1}},
|
|
},
|
|
"two IPv4s": {
|
|
inputIPs: []net.IP{{1, 1, 2, 1}, {1, 1, 1, 1}},
|
|
outputIPs: []net.IP{{1, 1, 1, 1}, {1, 1, 2, 1}},
|
|
},
|
|
}
|
|
for name, testCase := range testCases {
|
|
testCase := testCase
|
|
t.Run(name, func(t *testing.T) {
|
|
t.Parallel()
|
|
outputIPs := uniqueSortedIPs(testCase.inputIPs)
|
|
assert.Equal(t, testCase.outputIPs, outputIPs)
|
|
})
|
|
}
|
|
}
|