From 3a4ef4d70ed6e9970088169c7cf7afeba940736b Mon Sep 17 00:00:00 2001
From: JordanBrockopp <Jordan_Brockopp@cargill.com>
Date: Thu, 29 Jun 2023 11:12:21 -0500
Subject: [PATCH] fix(database): steps

---
 database/integration_test.go | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/database/integration_test.go b/database/integration_test.go
index 5ee54e016..f450c0bd1 100644
--- a/database/integration_test.go
+++ b/database/integration_test.go
@@ -262,21 +262,34 @@ func testSteps(t *testing.T, db Interface, builds []*library.Build) {
 	}
 	counter++
 
+	expected := map[string]float64{
+		"#init":                  1,
+		"target/vela-git:v0.3.0": 1,
+	}
 	images, err := db.ListStepImageCount()
 	if err != nil {
 		t.Errorf("unable to list step image count: %v", err)
 	}
-	if len(images) != len(steps) {
-		t.Errorf("ListStepImageCount() is %v, want %v", len(images), len(steps))
+	if !reflect.DeepEqual(images, expected) {
+		pretty.Ldiff(t, images, expected)
+		t.Errorf("ListStepImageCount() is %v, want %v", images, expected)
 	}
 	counter++
 
+	expected = map[string]float64{
+		"pending": 1,
+		"failure": 0,
+		"killed":  0,
+		"running": 1,
+		"success": 0,
+	}
 	statuses, err := db.ListStepStatusCount()
 	if err != nil {
 		t.Errorf("unable to list step status count: %v", err)
 	}
-	if len(statuses) != len(steps) {
-		t.Errorf("ListStepStatusCount() is %v, want %v", len(images), len(steps))
+	if !reflect.DeepEqual(statuses, expected) {
+		pretty.Ldiff(t, statuses, expected)
+		t.Errorf("ListStepStatusCount() is %v, want %v", statuses, expected)
 	}
 	counter++