Merge pull request #477 from spiral/feature/worker_watcher_testsv2.0.0-beta8

feat(worker_watcher): Add stack tests

1 files changed, 129 insertions, 0 deletions

diff --git a/pkg/worker_watcher/stack_test.go b/pkg/worker_watcher/stack_test.go
new file mode 100644
index 00000000..86af2043
--- /dev/null
+++ b/pkg/worker_watcher/stack_test.go
@@ -0,0 +1,129 @@
+package worker_watcher //nolint:golint,stylecheck
+import (
+	"context"
+	"os/exec"
+	"testing"
+	"time"
+
+	"github.com/spiral/roadrunner/v2/interfaces/worker"
+	workerImpl "github.com/spiral/roadrunner/v2/pkg/worker"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestNewWorkersStack(t *testing.T) {
+	stack := NewWorkersStack(0)
+	assert.Equal(t, uint64(0), stack.actualNumOfWorkers)
+	assert.Equal(t, []worker.BaseProcess{}, stack.workers)
+}
+
+func TestStack_Push(t *testing.T) {
+	stack := NewWorkersStack(1)
+
+	w, err := workerImpl.InitBaseWorker(&exec.Cmd{})
+	assert.NoError(t, err)
+
+	stack.Push(w)
+	assert.Equal(t, uint64(1), stack.actualNumOfWorkers)
+}
+
+func TestStack_Pop(t *testing.T) {
+	stack := NewWorkersStack(1)
+	cmd := exec.Command("php", "../tests/client.php", "echo", "pipes")
+
+	w, err := workerImpl.InitBaseWorker(cmd)
+	assert.NoError(t, err)
+
+	stack.Push(w)
+	assert.Equal(t, uint64(1), stack.actualNumOfWorkers)
+
+	_, _ = stack.Pop()
+	assert.Equal(t, uint64(0), stack.actualNumOfWorkers)
+}
+
+func TestStack_FindAndRemoveByPid(t *testing.T) {
+	stack := NewWorkersStack(1)
+	cmd := exec.Command("php", "../tests/client.php", "echo", "pipes")
+	w, err := workerImpl.InitBaseWorker(cmd)
+	assert.NoError(t, err)
+
+	assert.NoError(t, w.Start())
+
+	stack.Push(w)
+	assert.Equal(t, uint64(1), stack.actualNumOfWorkers)
+
+	stack.FindAndRemoveByPid(w.Pid())
+	assert.Equal(t, uint64(0), stack.actualNumOfWorkers)
+}
+
+func TestStack_IsEmpty(t *testing.T) {
+	stack := NewWorkersStack(1)
+	cmd := exec.Command("php", "../tests/client.php", "echo", "pipes")
+
+	w, err := workerImpl.InitBaseWorker(cmd)
+	assert.NoError(t, err)
+
+	stack.Push(w)
+	assert.Equal(t, uint64(1), stack.actualNumOfWorkers)
+
+	assert.Equal(t, false, stack.IsEmpty())
+}
+
+func TestStack_Workers(t *testing.T) {
+	stack := NewWorkersStack(1)
+	cmd := exec.Command("php", "../tests/client.php", "echo", "pipes")
+	w, err := workerImpl.InitBaseWorker(cmd)
+	assert.NoError(t, err)
+	assert.NoError(t, w.Start())
+
+	stack.Push(w)
+
+	wrks := stack.Workers()
+	assert.Equal(t, 1, len(wrks))
+	assert.Equal(t, w.Pid(), wrks[0].Pid())
+}
+
+func TestStack_Reset(t *testing.T) {
+	stack := NewWorkersStack(1)
+	cmd := exec.Command("php", "../tests/client.php", "echo", "pipes")
+	w, err := workerImpl.InitBaseWorker(cmd)
+	assert.NoError(t, err)
+	assert.NoError(t, w.Start())
+
+	stack.Push(w)
+	assert.Equal(t, uint64(1), stack.actualNumOfWorkers)
+	stack.Reset()
+	assert.Equal(t, uint64(0), stack.actualNumOfWorkers)
+}
+
+func TestStack_Destroy(t *testing.T) {
+	stack := NewWorkersStack(1)
+	cmd := exec.Command("php", "../tests/client.php", "echo", "pipes")
+	w, err := workerImpl.InitBaseWorker(cmd)
+	assert.NoError(t, err)
+	assert.NoError(t, w.Start())
+
+	stack.Push(w)
+	stack.Destroy(context.Background())
+	assert.Equal(t, uint64(0), stack.actualNumOfWorkers)
+}
+
+func TestStack_DestroyWithWait(t *testing.T) {
+	stack := NewWorkersStack(2)
+	cmd := exec.Command("php", "../tests/client.php", "echo", "pipes")
+	w, err := workerImpl.InitBaseWorker(cmd)
+	assert.NoError(t, err)
+	assert.NoError(t, w.Start())
+
+	stack.Push(w)
+	stack.Push(w)
+	assert.Equal(t, uint64(2), stack.actualNumOfWorkers)
+
+	go func() {
+		wrk, _ := stack.Pop()
+		time.Sleep(time.Second * 3)
+		stack.Push(wrk)
+	}()
+	time.Sleep(time.Second)
+	stack.Destroy(context.Background())
+	assert.Equal(t, uint64(0), stack.actualNumOfWorkers)
+}