pkg/test/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

package test

import (
	"os"
	"path/filepath"
	"strconv"

	"tunnel/pkg/client"
	"tunnel/pkg/server"
)

func getSocketPath() string {
	s := "tunnel.test." + strconv.Itoa(os.Getpid())
	return filepath.Join(os.TempDir(), s)
}

func newClientServer() (*client.Client, *server.Server, error) {
	socket := getSocketPath()

	s, err := server.New(socket)
	if err != nil {
		return nil, nil, err
	}

	go s.Serve()

	c, err := client.New(socket)
	if err != nil {
		s.Stop()
		return nil, nil, err
	}

	return c, s, nil
}