svcinit/task_wrapper_test.go at master · rrgmc/svcinit · 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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
package svcinit

import (
	"context"
	"testing"
	"testing/synctest"
	"time"

	"gotest.tools/v3/assert"
)

func TestTaskWrapper_stepCanRun(t *testing.T) {
	synctest.Test(t, func(t *testing.T) {
		testTask := BuildTask(
			WithSetup(func(ctx context.Context) error {
				return nil
			}),
			WithStart(func(ctx context.Context) error {
				return sleepContext(ctx, time.Second)
			}),
			WithStop(func(ctx context.Context) error {
				return nil
			}),
		)

		tw := newTaskWrapper(testTask)

		canStart, err := tw.checkStartStep(StepTeardown)
		assert.NilError(t, err)
		assert.Equal(t, canStart, false)
	})
}

func TestTaskWrapper_executeOrder(t *testing.T) {
	synctest.Test(t, func(t *testing.T) {
		ctx := t.Context()

		testTask := BuildTask(
			WithSetup(func(ctx context.Context) error {
				return nil
			}),
			WithStart(func(ctx context.Context) error {
				return sleepContext(ctx, time.Second)
			}),
			WithStop(func(ctx context.Context) error {
				return nil
			}),
			WithTeardown(func(ctx context.Context) error {
				return nil
			}),
		)

		tw := newTaskWrapper(testTask)

		canStart, err := tw.checkStartStep(StepSetup)
		assert.Assert(t, canStart)
		assert.NilError(t, err)
		canRun := tw.checkRunStep(StepSetup)
		assert.Assert(t, canRun)
		err = tw.run(ctx, StageDefault, StepSetup, nil, nil)
		assert.NilError(t, err)

		canStart, err = tw.checkStartStep(StepTeardown)
		assert.Assert(t, canStart)
		assert.NilError(t, err)
		canRun = tw.checkRunStep(StepTeardown)
		assert.Assert(t, canRun)
		err = tw.run(ctx, StageDefault, StepTeardown, nil, nil)
		assert.NilError(t, err)

		canStart, err = tw.checkStartStep(StepStop)
		assert.ErrorIs(t, err, ErrInvalidStepOrder)
		assert.Equal(t, false, canStart)
	})
}

func TestTaskWrapper_duplicate(t *testing.T) {
	synctest.Test(t, func(t *testing.T) {
		ctx := t.Context()

		testTask := BuildTask(
			WithSetup(func(ctx context.Context) error {
				return nil
			}),
			WithStart(func(ctx context.Context) error {
				return sleepContext(ctx, time.Second)
			}),
			WithStop(func(ctx context.Context) error {
				return nil
			}),
			WithTeardown(func(ctx context.Context) error {
				return nil
			}),
		)

		tw := newTaskWrapper(testTask)

		canStart, err := tw.checkStartStep(StepSetup)
		assert.Assert(t, canStart)
		assert.NilError(t, err)
		canRun := tw.checkRunStep(StepSetup)
		assert.Assert(t, canRun)
		err = tw.run(ctx, StageDefault, StepSetup, nil, nil)
		assert.NilError(t, err)

		canStart, err = tw.checkStartStep(StepSetup)
		assert.ErrorIs(t, err, ErrInvalidStepOrder)
		assert.Equal(t, false, canStart)
	})
}