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
111
112
113
|
// Copyright (c) 2018-2021, Sylabs Inc. All rights reserved.
// This software is licensed under a 3-clause BSD license. Please consult the
// LICENSE.md file distributed with the sources of this project regarding your
// rights to use or distribute this software.
package starter
import (
"context"
"testing"
"github.com/sylabs/singularity/v4/internal/pkg/runtime/engine"
"github.com/sylabs/singularity/v4/internal/pkg/test"
)
// TODO: actually we can't really test Master function which is
// part of the main function, as it exits, it would require mock at
// some point and that would make code more complex than necessary.
// createContainer and startContainer are quickly tested and only
// cover case with bad socket file descriptors or non socket
// file descriptor (stderr).
//nolint:dupl
func TestCreateContainer(t *testing.T) {
test.DropPrivilege(t)
defer test.ResetPrivilege(t)
var fatal error
fatalChan := make(chan error, 1)
tests := []struct {
name string
rpcSocket int
containerPid int
engine *engine.Engine
shallPass bool
}{
{
name: "nil engine; bad rpcSocket",
rpcSocket: -1,
containerPid: -1,
engine: nil,
shallPass: false,
},
{
name: "nil engine; wrong socket",
rpcSocket: 2,
containerPid: -1,
engine: nil,
shallPass: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
go createContainer(context.Background(), tt.rpcSocket, tt.containerPid, tt.engine, fatalChan)
// createContainer is creating a separate thread and we sync with that
// thread through a channel similarly to the createContainer function itself,
// as well as the Master function.
// For more details, please refer to the master_linux.go code.
fatal = <-fatalChan
if tt.shallPass && fatal != nil {
t.Fatalf("test %s expected to succeed but failed: %s", tt.name, fatal)
} else if !tt.shallPass && fatal == nil {
t.Fatalf("test %s expected to fail but succeeded", tt.name)
}
})
}
}
//nolint:dupl
func TestStartContainer(t *testing.T) {
test.DropPrivilege(t)
defer test.ResetPrivilege(t)
var fatal error
fatalChan := make(chan error, 1)
tests := []struct {
name string
masterSocket int
containerPid int
engine *engine.Engine
shallPass bool
}{
{
name: "nil engine; bad masterSocket",
masterSocket: -1,
containerPid: -1,
engine: nil,
shallPass: false,
},
{
name: "nil engine; wrong socket",
masterSocket: 2,
containerPid: -1,
engine: nil,
shallPass: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
go startContainer(context.Background(), tt.masterSocket, tt.containerPid, tt.engine, fatalChan)
fatal = <-fatalChan
if tt.shallPass && fatal != nil {
t.Fatalf("test %s expected to succeed but failed: %s", tt.name, fatal)
} else if !tt.shallPass && fatal == nil {
t.Fatalf("test %s expected to fail but succeeded", tt.name)
}
})
}
}
|
