This repository has been archived by the owner on May 1, 2024. It is now read-only.
/
zkregistry_test.go
114 lines (85 loc) · 2.82 KB
/
zkregistry_test.go
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
package serviceregistry_test
import (
"github.com/koofr/go-netutils"
. "github.com/koofr/go-serviceregistry"
"github.com/koofr/go-zkutils"
zk "github.com/koofr/gozk"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"strconv"
)
var _ = Describe("Zkregistry", func() {
var s *zkutils.TestServer
var r *ZkRegistry
var z *zk.Conn
BeforeEach(func() {
port, err := netutils.UnusedPort()
Expect(err).NotTo(HaveOccurred())
s, err = zkutils.NewTestServer(port)
Expect(err).NotTo(HaveOccurred())
if err != nil {
return
}
zz, session, err := zk.Dial("localhost:"+strconv.Itoa(port), 5e9)
Expect(err).NotTo(HaveOccurred())
Expect((<-session).State).To(Equal(zk.STATE_CONNECTED))
z = zz
rr, err := NewZkRegistry("localhost:" + strconv.Itoa(port))
Expect(err).NotTo(HaveOccurred())
r = rr
})
AfterEach(func() {
z.Close()
if r != nil {
r.Close()
}
s.Stop()
})
It("should register service", func() {
stat, err := z.Exists("/services/myservice/proto")
Expect(err).NotTo(HaveOccurred())
Expect(stat).To(BeNil())
err = r.Register("myservice", "proto", "localhost:1234")
Expect(err).NotTo(HaveOccurred())
stat, err = z.Exists("/services/myservice/proto")
Expect(err).NotTo(HaveOccurred())
Expect(stat).NotTo(BeNil())
children, _, err := z.Children("/services/myservice/proto")
Expect(err).NotTo(HaveOccurred())
Expect(children).To(HaveLen(1))
data, _, err := z.Get("/services/myservice/proto/" + children[0])
Expect(err).NotTo(HaveOccurred())
Expect(data).To(Equal("localhost:1234"))
err = r.Register("myservice", "proto", "localhost:1235")
Expect(err).NotTo(HaveOccurred())
children, _, err = z.Children("/services/myservice/proto")
Expect(err).NotTo(HaveOccurred())
Expect(children).To(HaveLen(2))
})
It("should remove service when registry closes", func() {
err := r.Register("myservice", "proto", "localhost:1234")
Expect(err).NotTo(HaveOccurred())
children, _, err := z.Children("/services/myservice/proto")
Expect(err).NotTo(HaveOccurred())
Expect(children).To(HaveLen(1))
r.Close()
children, _, err = z.Children("/services/myservice/proto")
Expect(err).NotTo(HaveOccurred())
Expect(children).To(HaveLen(0))
})
It("should get service", func() {
servers, err := r.Get("myservice", "proto")
Expect(err).NotTo(HaveOccurred())
Expect(servers).To(Equal([]string{}))
err = r.Register("myservice", "proto", "localhost:1234")
Expect(err).NotTo(HaveOccurred())
servers, err = r.Get("myservice", "proto")
Expect(err).NotTo(HaveOccurred())
Expect(servers).To(Equal([]string{"localhost:1234"}))
err = r.Register("myservice", "proto", "localhost:1235")
Expect(err).NotTo(HaveOccurred())
servers, err = r.Get("myservice", "proto")
Expect(err).NotTo(HaveOccurred())
Expect(servers).To(Equal([]string{"localhost:1234", "localhost:1235"}))
})
})