11package rancher2
22
33import (
4- "reflect"
54 "testing"
65
76 managementClient "github.com/rancher/rancher/pkg/client/generated/management/v3"
7+ "github.com/stretchr/testify/assert"
88)
99
1010var (
@@ -127,10 +127,7 @@ func TestFlattenEventRule(t *testing.T) {
127127
128128 for _ , tc := range cases {
129129 output := flattenEventRule (tc .Input )
130- if ! reflect .DeepEqual (output , tc .ExpectedOutput ) {
131- t .Fatalf ("Unexpected output from flattener.\n Expected: %#v\n Given: %#v" ,
132- tc .ExpectedOutput , output )
133- }
130+ assert .Equal (t , tc .ExpectedOutput , output , "Unexpected output from flattener." )
134131 }
135132}
136133
@@ -148,10 +145,7 @@ func TestFlattenMetricRule(t *testing.T) {
148145
149146 for _ , tc := range cases {
150147 output := flattenMetricRule (tc .Input )
151- if ! reflect .DeepEqual (output , tc .ExpectedOutput ) {
152- t .Fatalf ("Unexpected output from flattener.\n Expected: %#v\n Given: %#v" ,
153- tc .ExpectedOutput , output )
154- }
148+ assert .Equal (t , tc .ExpectedOutput , output , "Unexpected output from flattener." )
155149 }
156150}
157151
@@ -169,10 +163,7 @@ func TestFlattenNodeRule(t *testing.T) {
169163
170164 for _ , tc := range cases {
171165 output := flattenNodeRule (tc .Input )
172- if ! reflect .DeepEqual (output , tc .ExpectedOutput ) {
173- t .Fatalf ("Unexpected output from flattener.\n Expected: %#v\n Given: %#v" ,
174- tc .ExpectedOutput , output )
175- }
166+ assert .Equal (t , tc .ExpectedOutput , output , "Unexpected output from flattener." )
176167 }
177168}
178169
@@ -190,10 +181,7 @@ func TestFlattenPodRule(t *testing.T) {
190181
191182 for _ , tc := range cases {
192183 output := flattenPodRule (tc .Input )
193- if ! reflect .DeepEqual (output , tc .ExpectedOutput ) {
194- t .Fatalf ("Unexpected output from flattener.\n Expected: %#v\n Given: %#v" ,
195- tc .ExpectedOutput , output )
196- }
184+ assert .Equal (t , tc .ExpectedOutput , output , "Unexpected output from flattener." )
197185 }
198186}
199187
@@ -211,10 +199,7 @@ func TestFlattenSystemServiceRule(t *testing.T) {
211199
212200 for _ , tc := range cases {
213201 output := flattenSystemServiceRule (tc .Input )
214- if ! reflect .DeepEqual (output , tc .ExpectedOutput ) {
215- t .Fatalf ("Unexpected output from flattener.\n Expected: %#v\n Given: %#v" ,
216- tc .ExpectedOutput , output )
217- }
202+ assert .Equal (t , tc .ExpectedOutput , output , "Unexpected output from flattener." )
218203 }
219204}
220205
@@ -232,10 +217,7 @@ func TestFlattenWorkloadRule(t *testing.T) {
232217
233218 for _ , tc := range cases {
234219 output := flattenWorkloadRule (tc .Input )
235- if ! reflect .DeepEqual (output , tc .ExpectedOutput ) {
236- t .Fatalf ("Unexpected output from flattener.\n Expected: %#v\n Given: %#v" ,
237- tc .ExpectedOutput , output )
238- }
220+ assert .Equal (t , tc .ExpectedOutput , output , "Unexpected output from flattener." )
239221 }
240222}
241223
@@ -253,10 +235,7 @@ func TestExpandEventRule(t *testing.T) {
253235
254236 for _ , tc := range cases {
255237 output := expandEventRule (tc .Input )
256- if ! reflect .DeepEqual (output , tc .ExpectedOutput ) {
257- t .Fatalf ("Unexpected output from expander.\n Expected: %#v\n Given: %#v" ,
258- tc .ExpectedOutput , output )
259- }
238+ assert .Equal (t , tc .ExpectedOutput , output , "Unexpected output from expander." )
260239 }
261240}
262241
@@ -274,10 +253,7 @@ func TestExpandMetricRule(t *testing.T) {
274253
275254 for _ , tc := range cases {
276255 output := expandMetricRule (tc .Input )
277- if ! reflect .DeepEqual (output , tc .ExpectedOutput ) {
278- t .Fatalf ("Unexpected output from expander.\n Expected: %#v\n Given: %#v" ,
279- tc .ExpectedOutput , output )
280- }
256+ assert .Equal (t , tc .ExpectedOutput , output , "Unexpected output from expander." )
281257 }
282258}
283259
@@ -295,10 +271,7 @@ func TestExpandNodeRule(t *testing.T) {
295271
296272 for _ , tc := range cases {
297273 output := expandNodeRule (tc .Input )
298- if ! reflect .DeepEqual (output , tc .ExpectedOutput ) {
299- t .Fatalf ("Unexpected output from expander.\n Expected: %#v\n Given: %#v" ,
300- tc .ExpectedOutput , output )
301- }
274+ assert .Equal (t , tc .ExpectedOutput , output , "Unexpected output from expander." )
302275 }
303276}
304277
@@ -316,10 +289,7 @@ func TestExpandPodRule(t *testing.T) {
316289
317290 for _ , tc := range cases {
318291 output := expandPodRule (tc .Input )
319- if ! reflect .DeepEqual (output , tc .ExpectedOutput ) {
320- t .Fatalf ("Unexpected output from expander.\n Expected: %#v\n Given: %#v" ,
321- tc .ExpectedOutput , output )
322- }
292+ assert .Equal (t , tc .ExpectedOutput , output , "Unexpected output from expander." )
323293 }
324294}
325295
@@ -337,10 +307,7 @@ func TestExpandSystemServiceRule(t *testing.T) {
337307
338308 for _ , tc := range cases {
339309 output := expandSystemServiceRule (tc .Input )
340- if ! reflect .DeepEqual (output , tc .ExpectedOutput ) {
341- t .Fatalf ("Unexpected output from expander.\n Expected: %#v\n Given: %#v" ,
342- tc .ExpectedOutput , output )
343- }
310+ assert .Equal (t , tc .ExpectedOutput , output , "Unexpected output from expander." )
344311 }
345312}
346313
@@ -358,9 +325,6 @@ func TestExpandWorkloadRule(t *testing.T) {
358325
359326 for _ , tc := range cases {
360327 output := expandWorkloadRule (tc .Input )
361- if ! reflect .DeepEqual (output , tc .ExpectedOutput ) {
362- t .Fatalf ("Unexpected output from expander.\n Expected: %#v\n Given: %#v" ,
363- tc .ExpectedOutput , output )
364- }
328+ assert .Equal (t , tc .ExpectedOutput , output , "Unexpected output from expander." )
365329 }
366330}
0 commit comments