10 // The testing file contains test helpers that you can use outside of
11 // this package for making it easier to test plugins themselves.
13 // TestConn is a helper function for returning a client and server
14 // net.Conn connected to each other.
15 func TestConn(t *testing.T) (net.Conn, net.Conn) {
16 // Listen to any local port. This listener will be closed
17 // after a single connection is established.
18 l, err := net.Listen("tcp", "127.0.0.1:0")
20 t.Fatalf("err: %s", err)
23 // Start a goroutine to accept our client connection
24 var serverConn net.Conn
25 doneCh := make(chan struct{})
30 serverConn, err = l.Accept()
32 t.Fatalf("err: %s", err)
36 // Connect to the server
37 clientConn, err := net.Dial("tcp", l.Addr().String())
39 t.Fatalf("err: %s", err)
42 // Wait for the server side to acknowledge it has connected
45 return clientConn, serverConn
48 // TestRPCConn returns a rpc client and server connected to each other.
49 func TestRPCConn(t *testing.T) (*rpc.Client, *rpc.Server) {
50 clientConn, serverConn := TestConn(t)
52 server := rpc.NewServer()
53 go server.ServeConn(serverConn)
55 client := rpc.NewClient(clientConn)
59 // TestPluginRPCConn returns a plugin RPC client and server that are connected
60 // together and configured.
61 func TestPluginRPCConn(t *testing.T, ps map[string]Plugin) (*RPCClient, *RPCServer) {
62 // Create two net.Conns we can use to shuttle our control connection
63 clientConn, serverConn := TestConn(t)
65 // Start up the server
66 server := &RPCServer{Plugins: ps, Stdout: new(bytes.Buffer), Stderr: new(bytes.Buffer)}
67 go server.ServeConn(serverConn)
69 // Connect the client to the server
70 client, err := NewRPCClient(clientConn, ps)
72 t.Fatalf("err: %s", err)