refactor(dispatcherrm): split fetch HPC resources out (determined-ai#778

) [DET-9168]
determined-ai · Apr 16, 2024 · 76c94fc · 76c94fc
1 parent ab60c99
commit 76c94fc
Show file tree

Hide file tree

Showing 5 changed files with 532 additions and 473 deletions.
diff --git a/master/internal/rm/dispatcherrm/dispatcher_agents_test.go b/master/internal/rm/dispatcherrm/dispatcher_agents_test.go
@@ -6,7 +6,6 @@ package dispatcherrm
 import (
 	"context"
 	"testing"
-	"time"
 
 	"github.com/determined-ai/determined/master/internal/db"
 	"github.com/determined-ai/determined/master/pkg/actor"
@@ -23,16 +22,13 @@ func TestDispatcherAgents(t *testing.T) {
 	_, _ = db.Bun().NewDelete().Model(&dispatcherState{}).Where("id=0").Exec(context.TODO())
 
 	m := &dispatcherResourceManager{
-		resourceDetails: hpcResourceDetailsCache{
-			lastSample: hpcResources{
-				Nodes: []hpcNodeDetails{
-					{Name: "Node A"},
-					{Name: "Node B"},
-					{Name: "Node C"},
-				},
+		hpcDetailsCache: makeTestHpcDetailsCache(&hpcResources{
+			Nodes: []hpcNodeDetails{
+				{Name: "Node A"},
+				{Name: "Node B"},
+				{Name: "Node C"},
 			},
-			sampleTime: time.Now(),
-		},
+		}),
 		dbState: *newDispatcherState(),
 	}
 	_, err := m.disableAgent("Node Z")