webhookd/pkg/worker/test/work_runner_test.go

package test

import (
	"os"
	"strconv"
	"testing"

	"github.com/ncarlier/webhookd/pkg/assert"
	"github.com/ncarlier/webhookd/pkg/logger"
	"github.com/ncarlier/webhookd/pkg/model"
	"github.com/ncarlier/webhookd/pkg/worker"
)

func printWorkMessages(work *model.WorkRequest) {
	go func() {
		for {
			msg, open := <-work.MessageChan
			if !open {
				break
			}
			logger.Info.Println(string(msg))
		}
	}()
}

func TestWorkRunner(t *testing.T) {
	logger.Init("debug")
	script := "./test_simple.sh"
	args := []string{
		"name=foo",
		"user_agent=test",
	}
	payload := "{\"foo\": \"bar\"}"
	work := model.NewWorkRequest("test", script, payload, os.TempDir(), args, 5)
	assert.NotNil(t, work, "")
	printWorkMessages(work)
	err := worker.Run(work)
	assert.Nil(t, err, "")
	assert.Equal(t, work.Status, model.Success, "")
	assert.Equal(t, work.GetLogContent("notify:"), "OK\n", "")

	// Test that we can retrieve log file afterward
	id := strconv.FormatUint(work.ID, 10)
	logFile, err := worker.RetrieveLogFile(id, "test", os.TempDir())
	defer logFile.Close()
	assert.Nil(t, err, "Log file should exists")
	assert.NotNil(t, logFile, "Log file should be retrieve")
}

func TestWorkRunnerWithError(t *testing.T) {
	logger.Init("debug")
	script := "./test_error.sh"
	work := model.NewWorkRequest("test", script, "", os.TempDir(), []string{}, 5)
	assert.NotNil(t, work, "")
	printWorkMessages(work)
	err := worker.Run(work)
	assert.NotNil(t, err, "")
	assert.Equal(t, work.Status, model.Error, "")
	assert.Equal(t, "exit status 1", err.Error(), "")
}

func TestWorkRunnerWithTimeout(t *testing.T) {
	logger.Init("debug")
	script := "./test_timeout.sh"
	work := model.NewWorkRequest("test", script, "", os.TempDir(), []string{}, 1)
	assert.NotNil(t, work, "")
	printWorkMessages(work)
	err := worker.Run(work)
	assert.NotNil(t, err, "")
	assert.Equal(t, work.Status, model.Error, "")
	assert.Equal(t, "signal: killed", err.Error(), "")
}