-
Notifications
You must be signed in to change notification settings - Fork 2
/
client_test.go
71 lines (59 loc) · 1.74 KB
/
client_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
package main
import (
"bytes"
"io"
"testing"
arbor "github.com/arborchat/arbor-go"
"github.com/arborchat/muscadine/archive"
"github.com/arborchat/muscadine/types"
"github.com/onsi/gomega"
)
func bufConnector(address string) (io.ReadWriteCloser, error) {
return arbor.NoopRWCloser(new(bytes.Buffer)), nil
}
// TestNetClientInvalid checks that the NetClient constructor rejects invalid constructor
// parameters.
func TestNetClientInvalid(t *testing.T) {
g := gomega.NewGomegaWithT(t)
address := "localhost:7777"
history, err := archive.NewManager(".")
if err != nil {
t.Skip(err)
}
nc, err := NewNetClient(address, "username", nil)
g.Expect(nc).To(gomega.BeNil())
g.Expect(err).ToNot(gomega.BeNil())
nc, err = NewNetClient(address, "", history)
g.Expect(nc).To(gomega.BeNil())
g.Expect(err).ToNot(gomega.BeNil())
nc, err = NewNetClient("", "username", history)
g.Expect(nc).To(gomega.BeNil())
g.Expect(err).ToNot(gomega.BeNil())
}
// TestNetClient checks that the basic operations of a NetClient function as expected.
func TestNetClient(t *testing.T) {
g := gomega.NewGomegaWithT(t)
address := "localhost:7777"
history, err := archive.NewManager(".")
if err != nil {
t.Skip(err)
}
times := 0
timesDisconnected := make(chan int)
nc, err := NewNetClient(address, "username", history)
g.Expect(err).To(gomega.BeNil())
nc.SetConnector(bufConnector)
nc.OnDisconnect(func(client types.Connection) {
times++
timesDisconnected <- times
})
nc.OnReceive(func(m *arbor.ChatMessage) {
})
err = nc.Connect()
g.Expect(err).To(gomega.BeNil())
err = nc.Disconnect()
g.Expect(err).To(gomega.BeNil())
err = nc.Connect()
g.Expect(err).To(gomega.BeNil())
g.Eventually(func() int { return <-timesDisconnected }).Should(gomega.Equal(1))
}