-
Notifications
You must be signed in to change notification settings - Fork 2
/
socket_test.go
116 lines (103 loc) · 2.31 KB
/
socket_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
115
116
package wpaclient
import (
"os"
"strings"
"testing"
)
func TestConnDial(t *testing.T) {
tests := []struct {
name string
addr string
expErr bool
clean func(s *socket)
skip bool
}{
{name: "success0"},
{name: "success1"},
{name: "fail max socket count", expErr: true, skip: socketType == "UDP",
clean: func(s *socket) {
for i := 0; i < 2; i++ {
os.Remove(localSocket(i))
}
}},
{name: "check file", skip: socketType == "UDP",
clean: func(s *socket) {
s.close()
if _, err := os.Stat(localSocket(0)); !os.IsNotExist(err) {
t.Errorf("Expect Close() to remove socket file, it exists")
}
}},
{name: "not exist", addr: "not_exist", expErr: true},
{name: "closed", expErr: true},
}
ts, close := newTestServer(t)
defer close()
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
if tt.skip {
t.Skip()
}
if tt.addr == "" {
tt.addr = ts.addr()
}
if tt.name == "closed" {
if socketType == "UDP" {
tt.addr = "127.0.0.1:1"
} else {
fn := "/tmp/dkfhnli23yr98ndshfw2asdfkajjfo3eiu"
tt.addr = fn
os.Create(fn)
defer os.Remove(fn)
}
}
soc, err := dial(tt.addr)
if err != nil && !tt.expErr {
t.Errorf("Dial not expect an error, got %v", err)
}
if tt.expErr && err == nil {
t.Error("Dial expect an error, got <nil>")
}
if tt.clean != nil {
tt.clean(soc)
}
})
}
}
func TestSocketExecute(t *testing.T) {
ts, close := newTestServer(t)
defer close()
soc, err := dial(ts.addr())
if err != nil {
t.Fatalf("Dial failed, %v", err)
}
defer soc.close()
tests := []struct {
name string
fn func() error
err string
}{
{name: "success", fn: func() error {
_, err := soc.execute([]byte(CmdPing))
return err
}},
{name: "receive fail", err: "use of closed network connection",
fn: func() error {
soc.close()
_, err := soc.receive()
return err
}},
{name: "send fail", err: "use of closed network connection",
fn: func() error {
_, err := soc.execute([]byte(CmdPing))
return err
}},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
err := tt.fn()
if (tt.err == "" && err != nil) || (err != nil && !strings.Contains(err.Error(), tt.err)) {
t.Errorf("Execute expect error %v, got %v", tt.err, err)
}
})
}
}