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

#2040)

Co-authored-by: Jonathan Innis <[email protected]>

.github/actions/install-deps/action.yaml

@@ -16,7 +16,7 @@ runs:
     # Root path permission workaround for caching https://github.com/actions/cache/issues/845#issuecomment-1252594999
     - run: sudo chown "$USER" /usr/local
       shell: bash
-    - uses: actions/cache@0c45773b623bea8c8e75f6c82b208c3cf94ea4f9 # v4.0.2
+    - uses: actions/cache@d4323d4df104b026a6aa633fdb11d772146be0bf # v4.2.2
       id: cache-toolchain
       with:
         path: |

pkg/controllers/state/cluster.go

@@ -339,6 +339,7 @@ func (c *Cluster) ConsolidationState() time.Time {
 func (c *Cluster) Reset() {
 	c.mu.Lock()
 	defer c.mu.Unlock()
+	c.clusterState = time.Time{}
 	c.nodes = map[string]*StateNode{}
 	c.nodeNameToProviderID = map[string]string{}
 	c.nodeClaimNameToProviderID = map[string]string{}

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 {