summaryrefslogtreecommitdiff
path: root/server_test.go
blob: cfc75265873505dedc0c2f0895591e0fd40b7a48 (plain)
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
package roadrunner

import (
	"testing"
	"github.com/stretchr/testify/assert"
	"runtime"
	"time"
)

func TestServer_PipesEcho(t *testing.T) {
	srv := NewServer(&ServerConfig{
		Command: "php php-src/tests/client.php echo pipes",
		Relay:   "pipes",
		Pool: &Config{
			NumWorkers:      uint64(runtime.NumCPU()),
			AllocateTimeout: time.Second,
			DestroyTimeout:  time.Second,
		},
	}, nil)
	defer srv.Stop()

	assert.NoError(t, srv.Start())

	res, err := srv.Exec(&Payload{Body: []byte("hello")})

	assert.NoError(t, err)
	assert.NotNil(t, res)
	assert.NotNil(t, res.Body)
	assert.Nil(t, res.Context)

	assert.Equal(t, "hello", res.String())
}

func TestServer_SocketEcho(t *testing.T) {
	srv := NewServer(&ServerConfig{
		Command:      "php php-src/tests/client.php echo tcp",
		Relay:        "tcp://:9007",
		RelayTimeout: 10 * time.Second,
		Pool: &Config{
			NumWorkers:      uint64(runtime.NumCPU()),
			AllocateTimeout: time.Second,
			DestroyTimeout:  time.Second,
		},
	}, nil)
	defer srv.Stop()

	assert.NoError(t, srv.Start())

	res, err := srv.Exec(&Payload{Body: []byte("hello")})

	assert.NoError(t, err)
	assert.NotNil(t, res)
	assert.NotNil(t, res.Body)
	assert.Nil(t, res.Context)

	assert.Equal(t, "hello", res.String())
}

func TestServer_Reconfigure(t *testing.T) {
	srv := NewServer(&ServerConfig{
		Command: "php php-src/tests/client.php echo pipes",
		Relay:   "pipes",
		Pool: &Config{
			NumWorkers:      1,
			AllocateTimeout: time.Second,
			DestroyTimeout:  time.Second,
		},
	}, nil)
	defer srv.Stop()

	assert.NoError(t, srv.Start())
	assert.Len(t, srv.Workers(), 1)

	err := srv.Reconfigure(&ServerConfig{
		Command: "php php-src/tests/client.php echo pipes",
		Relay:   "pipes",
		Pool: &Config{
			NumWorkers:      2,
			AllocateTimeout: time.Second,
			DestroyTimeout:  time.Second,
		},
	})
	assert.NoError(t, err)

	assert.Len(t, srv.Workers(), 2)
}

func TestServer_Reset(t *testing.T) {
	srv := NewServer(&ServerConfig{
		Command: "php php-src/tests/client.php echo pipes",
		Relay:   "pipes",
		Pool: &Config{
			NumWorkers:      1,
			AllocateTimeout: time.Second,
			DestroyTimeout:  time.Second,
		},
	}, nil)
	defer srv.Stop()

	assert.NoError(t, srv.Start())
	assert.Len(t, srv.Workers(), 1)

	pid := *srv.Workers()[0].Pid
	assert.NoError(t, srv.Reset())
	assert.Len(t, srv.Workers(), 1)
	assert.NotEqual(t, pid, srv.Workers()[0].Pid)
}