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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
package controlsvc_test
import (
"context"
"errors"
"testing"
"time"
"github.com/ansible/receptor/pkg/controlsvc"
"github.com/ansible/receptor/pkg/controlsvc/mock_controlsvc"
"go.uber.org/mock/gomock"
)
func TestPingInitFromString(t *testing.T) {
pingCommandType := controlsvc.PingCommandType{}
initFromStringTestCases := []struct {
name string
expectedError bool
errorMessage string
input string
}{
{
name: "no ping target 1",
expectedError: true,
errorMessage: "no ping target",
input: "",
},
{
name: "single param - pass",
expectedError: false,
errorMessage: "",
input: "one",
},
}
for _, testCase := range initFromStringTestCases {
t.Run(testCase.name, func(t *testing.T) {
_, err := pingCommandType.InitFromString(testCase.input)
CheckExpectedError(testCase.expectedError, testCase.errorMessage, t, err)
})
}
}
type InitFromJSONTestCase struct {
name string
expectedError bool
errorMessage string
input map[string]interface{}
}
func BuildInitFromJSONTestCases(name string, expectedError bool, errorMessage string, input map[string]interface{}) InitFromJSONTestCase {
return InitFromJSONTestCase{
name: name,
expectedError: expectedError,
errorMessage: errorMessage,
input: input,
}
}
func TestPingInitFromJSON(t *testing.T) {
pingCommandType := controlsvc.PingCommandType{}
initFromJSONTestCases := []InitFromJSONTestCase{
BuildInitFromJSONTestCases("no ping target 2", true, "no ping target", map[string]interface{}{}),
BuildInitFromJSONTestCases("ping target must be string", true, "ping target must be string", map[string]interface{}{"target": 7}),
BuildInitFromJSONTestCases("three params - pass", false, "", map[string]interface{}{"target": "some target"}),
}
for _, testCase := range initFromJSONTestCases {
t.Run(testCase.name, func(t *testing.T) {
_, err := pingCommandType.InitFromJSON(testCase.input)
CheckExpectedError(testCase.expectedError, testCase.errorMessage, t, err)
})
}
}
func TestPingControlFunc(t *testing.T) {
pingCommand := controlsvc.PingCommand{}
ctrl := gomock.NewController(t)
mockNetceptor := mock_controlsvc.NewMockNetceptorForControlsvc(ctrl)
mockControlFunc := mock_controlsvc.NewMockControlFuncOperations(ctrl)
controlFuncTestCases := []struct {
name string
expectedError bool
errorMessage string
expectedCalls func()
}{
{
name: "ping error",
expectedError: true,
errorMessage: "terminated ping",
expectedCalls: func() {
mockNetceptor.EXPECT().Ping(gomock.Any(), gomock.Any(), gomock.Any()).Return(time.Second, "", errors.New("terminated ping"))
mockNetceptor.EXPECT().MaxForwardingHops()
},
},
{
name: "control func pass",
errorMessage: "",
expectedError: false,
expectedCalls: func() {
mockNetceptor.EXPECT().Ping(gomock.Any(), gomock.Any(), gomock.Any()).Return(time.Second, "", nil)
mockNetceptor.EXPECT().MaxForwardingHops()
},
},
}
for _, testCase := range controlFuncTestCases {
t.Run(testCase.name, func(t *testing.T) {
testCase.expectedCalls()
cfr, _ := pingCommand.ControlFunc(context.Background(), mockNetceptor, mockControlFunc)
err, ok := cfr["Error"]
if testCase.expectedError && testCase.errorMessage != err {
t.Errorf("expected: %s , received: %s", testCase.errorMessage, err)
}
if !testCase.expectedError && ok {
t.Error(cfr["Error"])
}
})
}
}
|
