42 lines
866 B
Go
42 lines
866 B
Go
package resolver
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_uniqueIPsToSlice(t *testing.T) {
|
|
t.Parallel()
|
|
testCases := map[string]struct {
|
|
inputIPs map[string]struct{}
|
|
outputIPs []net.IP
|
|
}{
|
|
"nil": {
|
|
inputIPs: nil,
|
|
outputIPs: []net.IP{},
|
|
},
|
|
"empty": {
|
|
inputIPs: map[string]struct{}{},
|
|
outputIPs: []net.IP{},
|
|
},
|
|
"single IPv4": {
|
|
inputIPs: map[string]struct{}{"1.1.1.1": {}},
|
|
outputIPs: []net.IP{{1, 1, 1, 1}},
|
|
},
|
|
"two IPv4s": {
|
|
inputIPs: map[string]struct{}{"1.1.1.1": {}, "1.1.2.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 := uniqueIPsToSlice(testCase.inputIPs)
|
|
assert.ElementsMatch(t, testCase.outputIPs, outputIPs)
|
|
})
|
|
}
|
|
}
|