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
|
//go:build !no_workceptor
// +build !no_workceptor
package workceptor
import (
"context"
"os"
"testing"
"github.com/ansible/receptor/pkg/netceptor"
)
func newCommandWorker(_ BaseWorkUnitForWorkUnit, w *Workceptor, unitID string, workType string) WorkUnit {
cw := &commandUnit{
BaseWorkUnitForWorkUnit: &BaseWorkUnit{
status: StatusFileData{
ExtraData: &CommandExtraData{},
},
},
command: "echo",
baseParams: "foo",
allowRuntimeParams: true,
}
cw.BaseWorkUnitForWorkUnit.Init(w, unitID, workType, FileSystem{}, nil)
return cw
}
func TestWorkceptorJson(t *testing.T) {
tmpdir, err := os.MkdirTemp(os.TempDir(), "receptor-test-*")
if err != nil {
t.Fatal(err)
}
defer os.RemoveAll(tmpdir)
nc := netceptor.New(context.TODO(), "test")
w, err := New(context.Background(), nc, tmpdir)
if err != nil {
t.Fatal(err)
}
err = w.RegisterWorker("command", newCommandWorker, false)
if err != nil {
t.Fatal(err)
}
cw, err := w.AllocateUnit("command", "", make(map[string]string))
if err != nil {
t.Fatal(err)
}
cw.UpdateFullStatus(func(status *StatusFileData) {
ed, ok := status.ExtraData.(*CommandExtraData)
if !ok {
t.Fatal("ExtraData type assertion failed")
}
ed.Pid = 12345
})
err = cw.Save()
if err != nil {
t.Fatal(err)
}
cw2 := newCommandWorker(nil, w, cw.ID(), "command")
err = cw2.Load()
if err != nil {
t.Fatal(err)
}
ed2, ok := cw2.Status().ExtraData.(*CommandExtraData)
if !ok {
t.Fatal("ExtraData type assertion failed")
}
if ed2.Pid != 12345 {
t.Fatal("PID did not make it through")
}
}
|
