mirror of
https://github.com/ncarlier/webhookd.git
synced 2025-04-06 20:31:31 +00:00
07fbb6ee3a
BREAKING CHANGES for configuration - rename `-log-dir` to `-hook-log-dir` - add `-hook-log-output` - add `-log-level` - remove `-debug` close #44
73 lines
1.9 KiB
Go
73 lines
1.9 KiB
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", "out")
|
|
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(), "")
|
|
}
|