Skip to content

Commit b799626

Browse files
committed
fix unit_test
Signed-off-by: Suleiman Dibirov <[email protected]>
1 parent b05399e commit b799626

File tree

2 files changed

+15
-21
lines changed

2 files changed

+15
-21
lines changed

pkg/skaffold/deploy/helm/util.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ func VerifyNoCycles(graph map[string][]string) error {
6060
// ensuring that releases are deployed after their dependencies.
6161
func calculateDeploymentOrder(graph map[string][]string) ([]string, error) {
6262
visited := make(map[string]bool)
63-
var order []string
63+
order := make([]string, 0)
6464

6565
var visit func(node string) error
6666
visit = func(node string) error {

pkg/skaffold/deploy/helm/util_test.go

Lines changed: 14 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@ import (
66

77
"github.com/GoogleContainerTools/skaffold/v2/pkg/skaffold/schema/latest"
88
"github.com/GoogleContainerTools/skaffold/v2/testutil"
9-
"github.com/stretchr/testify/require"
109
)
1110

1211
func TestBuildDependencyGraph(t *testing.T) {
@@ -65,12 +64,11 @@ func TestBuildDependencyGraph(t *testing.T) {
6564
graph, err := BuildDependencyGraph(test.releases)
6665

6766
if test.shouldErr {
68-
require.Error(t, err)
67+
t.CheckError(true, err)
6968
return
7069
}
7170

72-
require.NoError(t, err)
73-
require.Equal(t, test.expected, graph)
71+
t.CheckErrorAndDeepEqual(false, err, test.expected, graph)
7472
})
7573
}
7674
}
@@ -137,10 +135,9 @@ func TestVerifyNoCycles(t *testing.T) {
137135
err := VerifyNoCycles(test.graph)
138136

139137
if test.shouldErr {
140-
require.Error(t, err)
141-
require.Contains(t, err.Error(), "cycle detected")
138+
t.CheckErrorContains("cycle detected", err)
142139
} else {
143-
require.NoError(t, err)
140+
t.RequireNoError(err)
144141
}
145142
})
146143
}
@@ -199,24 +196,22 @@ func TestCalculateDeploymentOrder(t *testing.T) {
199196
}
200197

201198
for _, test := range tests {
202-
t.Run(test.description, func(t *testing.T) {
199+
testutil.Run(t, test.description, func(t *testutil.T) {
203200
order, err := calculateDeploymentOrder(test.graph)
204201

205202
if test.shouldErr {
206-
require.Error(t, err)
203+
t.CheckError(true, err)
207204
return
208205
}
209206

210-
require.NoError(t, err)
211-
require.Equal(t, len(test.expected), len(order), "deployment order length mismatch")
207+
t.CheckErrorAndDeepEqual(false, err, test.expected, order)
212208

213209
// Verify order satisfies dependencies
214210
installed := make(map[string]bool)
215211
for _, release := range order {
216212
// Check all dependencies are installed
217213
for _, dep := range test.graph[release] {
218-
require.True(t, installed[dep],
219-
"release %s deployed before dependency %s", release, dep)
214+
t.CheckTrue(installed[dep])
220215
}
221216
installed[release] = true
222217
}
@@ -297,14 +292,13 @@ func TestGroupReleasesByLevel(t *testing.T) {
297292
}
298293

299294
for _, test := range tests {
300-
t.Run(test.description, func(t *testing.T) {
295+
testutil.Run(t, test.description, func(t *testutil.T) {
301296
levels := groupReleasesByLevel(test.order, test.graph)
302297

303-
require.Equal(t, len(test.expected), len(levels), "number of levels mismatch")
298+
t.CheckDeepEqual(len(test.expected), len(levels))
304299

305300
for level, releases := range test.expected {
306-
require.ElementsMatch(t, releases, levels[level],
307-
"releases at level %d don't match", level)
301+
t.CheckDeepEqual(releases, levels[level])
308302
}
309303

310304
// Verify level assignments are correct
@@ -314,9 +308,9 @@ func TestGroupReleasesByLevel(t *testing.T) {
314308
for _, dep := range test.graph[release] {
315309
for depLevel, depReleases := range levels {
316310
if slices.Contains(depReleases, dep) {
317-
require.Less(t, depLevel, level,
318-
"dependency %s at level %d >= release %s at level %d",
319-
dep, depLevel, release, level)
311+
if depLevel >= level {
312+
t.Errorf("dependency %s at level %d >= release %s at level %d", dep, depLevel, release, level)
313+
}
320314
}
321315
}
322316
}

0 commit comments

Comments
 (0)