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
|
package dnsutils
import (
"github.com/stretchr/testify/assert"
"net"
"testing"
)
func TestOrderSRV(t *testing.T) {
srv11 := net.SRV{
Target: "t11",
Port: 1234,
Priority: 1,
Weight: 100,
}
srv12 := net.SRV{
Target: "t12",
Port: 1234,
Priority: 1,
Weight: 100,
}
srv13 := net.SRV{
Target: "t13",
Port: 1234,
Priority: 1,
Weight: 20,
}
srv21 := net.SRV{
Target: "t21",
Port: 1234,
Priority: 2,
Weight: 1,
}
addrs := []*net.SRV{
&srv11, &srv21, &srv12, &srv13,
}
count, orderedSRV := orderSRV(addrs)
assert.Equal(t, len(addrs), count, "Index not the expected size")
assert.Equal(t, len(addrs), len(orderedSRV), "orderedSRV not the expected size")
assert.Equal(t, uint16(2), orderedSRV[4].Priority, "Priority order not as expected")
}
|
