test: fix test

SpecterOps · Oct 2, 2024 · 4982db6 · 4982db6
1 parent 54ea4db
commit 4982db6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/cmd/api/src/database/datapipestatus_test.go b/cmd/api/src/database/datapipestatus_test.go
@@ -35,12 +35,15 @@ func TestDatapipeStatus(t *testing.T) {
 	require.Nil(t, err)
 	require.Equal(t, model.DatapipeStatusIdle, status.Status)
 
+	oldAnalysisTime := status.LastAnalysisRunAt
+
 	err = db.SetDatapipeStatus(testCtx, model.DatapipeStatusAnalyzing, false)
 	require.Nil(t, err)
 
 	status, err = db.GetDatapipeStatus(testCtx)
 	require.Nil(t, err)
 	require.Equal(t, model.DatapipeStatusAnalyzing, status.Status)
+	require.True(t, oldAnalysisTime.Before(status.LastAnalysisRunAt))
 
 	// when `SetDatapipeStatus` is called with `true` for the `updateAnalysisTime` parameter, assert that the time is no longer null
 	require.True(t, status.LastCompleteAnalysisAt.IsZero())

diff --git a/cmd/api/src/database/parameters_test.go b/cmd/api/src/database/parameters_test.go
@@ -132,7 +132,7 @@ func TestParameters_GetAllConfigurationParameter(t *testing.T) {
 	)
 	parameters, err := dbInst.GetAllConfigurationParameters(testCtx)
 	require.Nil(t, err)
-	require.Len(t, parameters, 6)
+	require.Len(t, parameters, 5)
 	for _, parameter := range parameters {
 		require.True(t, parameter.IsValidKey(parameter.Key))
 	}