Update UTs

Checkmarx · Oct 2, 2024 · 2791bd9 · 2791bd9
1 parent 0a1d217
commit 2791bd9
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/engine/engine_test.go b/engine/engine_test.go
@@ -2,6 +2,7 @@ package engine
 
 import (
 	"fmt"
+	"github.com/stretchr/testify/assert"
 	"sync"
 	"testing"
 
@@ -159,11 +160,10 @@ func TestSecrets(t *testing.T) {
 
 			s := <-secretsChan
 
-			if s == nil && secret.ShouldFind {
-				t.Errorf("secret \"%s\" not found", secret.Name)
-			}
-			if s != nil && !secret.ShouldFind {
-				t.Errorf("should not find")
+			if secret.ShouldFind {
+				assert.Equal(t, s.LineContent, secret.Content)
+			} else {
+				assert.Nil(t, s)
 			}
 		})
 	}