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
130
131
132
133
134
135
136
137
138
139
|
package net
import (
"io/ioutil"
"log"
"testing"
"github.com/miekg/dns"
"github.com/twstrike/gotk3adapter/glib_mock"
"github.com/twstrike/coyim/i18n"
. "gopkg.in/check.v1"
)
func Test(t *testing.T) { TestingT(t) }
func init() {
log.SetOutput(ioutil.Discard)
i18n.InitLocalization(&glib_mock.Mock{})
}
type DNSXmppSuite struct{}
var _ = Suite(&DNSXmppSuite{})
func (s *DNSXmppSuite) Test_createCName_createsAValidCnameForAService(c *C) {
ret := createCName("foo", "bar", "bax.com")
c.Assert(ret, Equals, "_foo._bar.bax.com.")
}
func (s *DNSXmppSuite) Test_convertAnswerToSRV_returnsNilForNonSRVEntry(c *C) {
cn := new(dns.CNAME)
res := convertAnswerToSRV(cn)
c.Assert(res, IsNil)
}
func (s *DNSXmppSuite) Test_convertAnswerToSRV_returnsAValidNetSRV(c *C) {
srv := new(dns.SRV)
srv.Target = "foo.com"
srv.Port = 123
srv.Priority = 5
srv.Weight = 42
res := convertAnswerToSRV(srv)
c.Assert(res, Not(IsNil))
c.Assert(res.Target, Equals, "foo.com")
c.Assert(res.Port, Equals, uint16(123))
c.Assert(res.Priority, Equals, uint16(5))
c.Assert(res.Weight, Equals, uint16(42))
}
func (s *DNSXmppSuite) Test_convertAnswersToSRV_convertsAnswers(c *C) {
cn := new(dns.CNAME)
srv := new(dns.SRV)
srv.Target = "foo2.com"
in := make([]dns.RR, 2)
in[0] = cn
in[1] = srv
res := convertAnswersToSRV(in)
c.Assert(res, HasLen, 1)
c.Assert(res[0].Target, Equals, "foo2.com")
}
func (s *DNSXmppSuite) Test_msgSRV_createsMessage(c *C) {
res := msgSRV("foo.com")
c.Assert(res.Question[0].Name, Equals, "foo.com")
c.Assert(res.Question[0].Qtype, Equals, dns.TypeSRV)
}
func (s *DNSXmppSuite) Test_convertAnswersToSRV_sortsByPriority(c *C) {
srv1 := &dns.SRV{
Target: "foo1.com",
Priority: 5,
Weight: 1,
}
srv2 := &dns.SRV{
Target: "foo2.com",
Priority: 3,
Weight: 1,
}
srv3 := &dns.SRV{
Target: "foo3.com",
Priority: 6,
Weight: 1,
}
srv4 := &dns.SRV{
Target: "foo4.com",
Priority: 1,
Weight: 1,
}
in := []dns.RR{
srv1,
srv2,
srv3,
srv4,
}
res := convertAnswersToSRV(in)
c.Assert(res[0].Target, Equals, "foo4.com")
c.Assert(res[1].Target, Equals, "foo2.com")
c.Assert(res[2].Target, Equals, "foo1.com")
c.Assert(res[3].Target, Equals, "foo3.com")
}
func (s *DNSXmppSuite) Test_convertAnswersToSRV_sortsByWeightIfPriotityIsTheSame(c *C) {
srv1 := &dns.SRV{
Target: "foo1.com",
Priority: 1,
Weight: 5,
}
srv2 := &dns.SRV{
Target: "foo2.com",
Priority: 1,
Weight: 3,
}
srv3 := &dns.SRV{
Target: "foo3.com",
Priority: 1,
Weight: 6,
}
srv4 := &dns.SRV{
Target: "foo4.com",
Priority: 1,
Weight: 1,
}
in := []dns.RR{
srv1,
srv2,
srv3,
srv4,
}
res := convertAnswersToSRV(in)
c.Assert(res[0].Target, Equals, "foo4.com")
c.Assert(res[1].Target, Equals, "foo2.com")
c.Assert(res[2].Target, Equals, "foo1.com")
c.Assert(res[3].Target, Equals, "foo3.com")
}
|
