pg_timetable/internal/scheduler/tasks_test.go at e2f0da7c9b22a97ac17b034ee309757353f806ee · cybertec-postgresql/pg_timetable · GitHub

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
package scheduler

import (
	"context"
	"testing"

	"github.com/cybertec-postgresql/pg_timetable/internal/config"
	"github.com/cybertec-postgresql/pg_timetable/internal/log"
	"github.com/cybertec-postgresql/pg_timetable/internal/pgengine"
	"github.com/pashagolub/pgxmock/v5"
	"github.com/stretchr/testify/assert"
)

func TestExecuteTask(t *testing.T) {
	mock, err := pgxmock.NewPool() //
	a := assert.New(t)
	a.NoError(err)
	pge := pgengine.NewDB(mock, "--log-database-level=none")
	mocksch := New(pge, log.Init(config.LoggingOpts{LogLevel: "panic", LogDBLevel: "none"}))

	et := func(task string, params []string) (err error) {
		err = mocksch.executeBuiltinTask(context.TODO(), &pgengine.ChainTask{Command: task}, params)
		return
	}

	a.Error(et("foo", []string{}))

	a.Error(et("Sleep", []string{"foo"}))
	a.NoError(et("Sleep", []string{"1"}))

	a.NoError(et("NoOp", []string{}))
	a.NoError(et("NoOp", []string{"foo", "bar"}))

	a.NoError(et("Log", []string{"foo"}))

	a.Error(et("CopyFromFile", []string{"foo"}), "Invalid json")
	a.Error(et("CopyFromFile", []string{`{"sql": "COPY", "filename": "foo"}`}), "Acquire() should fail")

	a.Error(et("CopyToFile", []string{"foo"}), "Invalid json")
	a.Error(et("CopyToFile", []string{`{"sql": "COPY", "filename": "foo"}`}), "Acquire() should fail")

	a.Error(et("CopyToProgram", []string{"foo"}), "Invalid json")
	a.Error(et("CopyToProgram", []string{`{"sql": "COPY", "program": "foo"}`}), "Acquire() should fail")

	a.Error(et("CopyFromProgram", []string{"foo"}), "Invalid json")
	a.Error(et("CopyFromProgram", []string{`{"sql": "COPY", "program": "foo"}`}), "Acquire() should fail")

	a.Error(et("SendMail", []string{"foo"}), "Invalid json")
	a.Error(et("SendMail", []string{`{"ServerHost":"smtp.example.com","ServerPort":587,"Username":"user"}`}))

	a.Error(et("Download", []string{"foo"}), "Invalid json")
	a.EqualError(et("Download", []string{`{"workersnum": 0, "fileurls": [] }`}),
		"files to download are not specified", "Download with empty files should fail")
	a.Error(et("Download", []string{`{"workersnum": 0, "fileurls": ["http://foo.bar"], "destpath": "" }`}),
		"Downlod incorrect url should fail")

	a.NoError(et("Shutdown", []string{}))
}