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
|
package etcdv3
import (
"bytes"
"errors"
"testing"
"github.com/go-kit/log"
)
// testClient is a basic implementation of Client
type testClient struct {
registerRes error // value returned when Register or Deregister is called
}
func (tc *testClient) GetEntries(prefix string) ([]string, error) {
return nil, nil
}
func (tc *testClient) WatchPrefix(prefix string, ch chan struct{}) {
}
func (tc *testClient) Register(s Service) error {
return tc.registerRes
}
func (tc *testClient) Deregister(s Service) error {
return tc.registerRes
}
func (tc *testClient) LeaseID() int64 {
return 0
}
// default service used to build registrar in our tests
var testService = Service{
Key: "testKey",
Value: "testValue",
TTL: nil,
}
// NewRegistar should return a registar with a logger using the service key and value
func TestNewRegistar(t *testing.T) {
c := Client(&testClient{nil})
buf := &bytes.Buffer{}
logger := log.NewLogfmtLogger(buf)
r := NewRegistrar(
c,
testService,
logger,
)
if err := r.logger.Log("msg", "message"); err != nil {
t.Fatal(err)
}
if want, have := "key=testKey value=testValue msg=message\n", buf.String(); want != have {
t.Errorf("\nwant: %shave: %s", want, have)
}
}
func TestRegister(t *testing.T) {
// Register log the error returned by the client or log the successful registration action
// table of test cases for method Register
var registerTestTable = []struct {
registerRes error // value returned by the client on calls to Register
log string // expected log by the registrar
}{
// test case: an error is returned by the client
{errors.New("regError"), "key=testKey value=testValue err=regError\n"},
// test case: registration successful
{nil, "key=testKey value=testValue action=register\n"},
}
for _, tc := range registerTestTable {
c := Client(&testClient{tc.registerRes})
buf := &bytes.Buffer{}
logger := log.NewLogfmtLogger(buf)
r := NewRegistrar(
c,
testService,
logger,
)
r.Register()
if want, have := tc.log, buf.String(); want != have {
t.Fatalf("want %v, have %v", want, have)
}
}
}
func TestDeregister(t *testing.T) {
// Deregister log the error returned by the client or log the successful deregistration action
// table of test cases for method Deregister
var deregisterTestTable = []struct {
deregisterRes error // value returned by the client on calls to Deregister
log string // expected log by the registrar
}{
// test case: an error is returned by the client
{errors.New("deregError"), "key=testKey value=testValue err=deregError\n"},
// test case: deregistration successful
{nil, "key=testKey value=testValue action=deregister\n"},
}
for _, tc := range deregisterTestTable {
c := Client(&testClient{tc.deregisterRes})
buf := &bytes.Buffer{}
logger := log.NewLogfmtLogger(buf)
r := NewRegistrar(
c,
testService,
logger,
)
r.Deregister()
if want, have := tc.log, buf.String(); want != have {
t.Fatalf("want %v, have %v", want, have)
}
}
}
|
