mirror-webhookd/pkg/hook/test/job_test.go

package test

import (
	"log/slog"
	"os"
	"strconv"
	"testing"

	"github.com/ncarlier/webhookd/pkg/assert"
	"github.com/ncarlier/webhookd/pkg/hook"
)

func printJobMessages(job *hook.Job) {
	go func() {
		for {
			msg, open := <-job.MessageChan
			if !open {
				break
			}
			slog.Info(string(msg))
		}
	}()
}

func TestHookJob(t *testing.T) {
	req := &hook.Request{
		Name:    "test_simple",
		Script:  "../test/test_simple.sh",
		Method:  "GET",
		Payload: "{\"foo\": \"bar\"}",
		Args: []string{
			"name=foo",
			"user_agent=test",
		},
		Timeout:   5,
		OutputDir: os.TempDir(),
	}
	job, err := hook.NewHookJob(req)
	assert.Nil(t, err, "")
	assert.NotNil(t, job, "")
	printJobMessages(job)
	err = job.Run()
	assert.Nil(t, err, "")
	assert.Equal(t, job.Status(), hook.Success, "")
	assert.Equal(t, job.Logs("notify:"), "OK\n", "")

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

func TestWorkRunnerWithError(t *testing.T) {
	req := &hook.Request{
		Name:      "test_error",
		Script:    "../test/test_error.sh",
		Method:    "POST",
		Payload:   "",
		Args:      []string{},
		Timeout:   5,
		OutputDir: os.TempDir(),
	}
	job, err := hook.NewHookJob(req)
	assert.Nil(t, err, "")
	assert.NotNil(t, job, "")
	printJobMessages(job)
	err = job.Run()
	assert.NotNil(t, err, "")
	assert.Equal(t, job.Status(), hook.Error, "")
	assert.Equal(t, "exit status 1", err.Error(), "")
	assert.Equal(t, 1, job.ExitCode(), "")
}

func TestWorkRunnerWithTimeout(t *testing.T) {
	req := &hook.Request{
		Name:      "test_timeout",
		Script:    "../test/test_timeout.sh",
		Method:    "POST",
		Payload:   "",
		Args:      []string{},
		Timeout:   1,
		OutputDir: os.TempDir(),
	}
	job, err := hook.NewHookJob(req)
	assert.Nil(t, err, "")
	assert.NotNil(t, job, "")
	printJobMessages(job)
	err = job.Run()
	assert.NotNil(t, err, "")
	assert.Equal(t, job.Status(), hook.Error, "")
	assert.Equal(t, "signal: killed", err.Error(), "")
}