fix test

Signed-off-by: Tim Vaillancourt <[email protected]>
vitessio · Dec 9, 2024 · 0f50b80 · 0f50b80
1 parent 6a54868
commit 0f50b80
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/go/vt/vtorc/process/health_test.go b/go/vt/vtorc/process/health_test.go
@@ -26,21 +26,23 @@ func TestHealthTest(t *testing.T) {
 	defer func() {
 		FirstDiscoveryCycleComplete.Store(false)
 		ThisNodeHealth = &NodeHealth{}
+		ResetLastHealthCheckCache()
 	}()
 
 	require.Zero(t, ThisNodeHealth.LastReported)
-	require.False(t, ThisNodeHealth.Healthy)
 
 	ThisNodeHealth = &NodeHealth{}
-	health, discoveredOnce := HealthTest()
+	health, err := HealthTest()
+	require.NoError(t, err)
 	require.False(t, health.Healthy)
-	require.False(t, discoveredOnce)
+	require.False(t, health.DiscoveredOnce)
 	require.NotZero(t, ThisNodeHealth.LastReported)
 
 	ThisNodeHealth = &NodeHealth{}
 	FirstDiscoveryCycleComplete.Store(true)
-	health, discoveredOnce = HealthTest()
+	health, err = HealthTest()
+	require.NoError(t, err)
 	require.True(t, health.Healthy)
-	require.True(t, discoveredOnce)
+	require.True(t, health.DiscoveredOnce)
 	require.NotZero(t, ThisNodeHealth.LastReported)
 }