test: Fix ConsolidationState() test failure in state testing (#1957)

pkg/controllers/state/cluster.go

@@ -339,12 +339,17 @@ func (c *Cluster) ConsolidationState() time.Time {
 func (c *Cluster) Reset() {
 	c.mu.Lock()
 	defer c.mu.Unlock()
+	c.clusterState = time.Time{}
+	c.unsyncedStartTime = time.Time{}
 	c.nodes = map[string]*StateNode{}
 	c.nodeNameToProviderID = map[string]string{}
 	c.nodeClaimNameToProviderID = map[string]string{}
 	c.bindings = map[types.NamespacedName]string{}
 	c.antiAffinityPods = sync.Map{}
 	c.daemonSetPods = sync.Map{}
+	c.podAcks = sync.Map{}
+	c.podsSchedulingAttempted = sync.Map{}
+	c.podsSchedulableTimes = sync.Map{}
 }
 
 func (c *Cluster) GetDaemonSetPod(daemonset *appsv1.DaemonSet) *corev1.Pod {

pkg/controllers/state/suite_test.go

@@ -21,6 +21,7 @@ import (
 	"fmt"
 	"math/rand"
 	"net"
+	"sync"
 	"testing"
 	"time"
 
@@ -1443,7 +1444,7 @@ var _ = Describe("Consolidated State", func() {
 		cluster.MarkUnconsolidated()
 		Expect(cluster.ConsolidationState()).ToNot(Equal(state))
 	})
-	It("should update the consolidated value when consolidation timeout (5m) has passed and state hasn't changed", func() {
+	It("should update the consolidated value when state timeout (5m) has passed and state hasn't changed", func() {
 		state := cluster.ConsolidationState()
 
 		fakeClock.Step(time.Minute)
@@ -1468,12 +1469,16 @@ var _ = Describe("Consolidated State", func() {
 var _ = Describe("Data Races", func() {
 	It("should ensure that calling Synced() is valid while making updates to Nodes", func() {
 		cancelCtx, cancel := context.WithCancel(ctx)
+		var wg sync.WaitGroup
 		DeferCleanup(func() {
 			cancel()
+			wg.Wait()
 		})
 
 		// Keep calling Synced for the entirety of this test
+		wg.Add(1)
 		go func() {
+			defer wg.Done()
 			for {
 				_ = cluster.Synced(ctx)
 				if cancelCtx.Err() != nil {
@@ -1493,12 +1498,16 @@ var _ = Describe("Data Races", func() {
 	})
 	It("should ensure that calling Synced() is valid while making updates to NodeClaims", func() {
 		cancelCtx, cancel := context.WithCancel(ctx)
+		var wg sync.WaitGroup
 		DeferCleanup(func() {
 			cancel()
+			wg.Wait()
 		})
 
 		// Keep calling Synced for the entirety of this test
+		wg.Add(1)
 		go func() {
+			defer wg.Done()
 			for {
 				_ = cluster.Synced(ctx)
 				if cancelCtx.Err() != nil {