forked from shibukawa/tobubus
-
Notifications
You must be signed in to change notification settings - Fork 1
/
util_test.go
49 lines (43 loc) · 1.11 KB
/
util_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
package tobubus
import (
"github.com/shibukawa/localsocket"
"github.com/shibukawa/mockconn"
"net"
"testing"
)
type testStruct struct {
args []string
result string
}
func (ts *testStruct) TestMethod(arg string) string {
ts.args = []string{arg}
return ts.result
}
func (ts *testStruct) testMethod(arg string) string {
ts.args = []string{arg}
return ts.result
}
func newPluginForTest(pipeName, id string, t *testing.T) (*Plugin, *mockconn.Conn) {
socket := mockconn.New(t)
return &Plugin{
pipeName: pipeName,
id: id,
socket: socket,
objectMap: make(map[string]*Proxy),
sessions: newSessionManager(incrementStrategy),
}, socket
}
func newHostForTest(pipeName string) *Host {
server := localsocket.NewLocalServer(pipeName)
host := &Host{
server: server,
sessions: newSessionManager(incrementStrategy),
pluginReservedSpaces: make(map[string]net.Conn),
localObjectMap: make(map[string]*Proxy),
sockets: make(map[string]net.Conn),
}
server.SetOnConnectionCallback(func(socket net.Conn) {
go host.listenAndServeTo(socket)
})
return host
}