Scheduler: fix panic in nodedb benchmark (#4120)

Signed-off-by: Robert Smith <robertdavidsmith@yahoo.com>
armadaproject · Jan 3, 2025 · 3a3fc36 · 3a3fc36
1 parent 1ddd71b
commit 3a3fc36
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/internal/scheduler/nodedb/nodedb_test.go b/internal/scheduler/nodedb/nodedb_test.go
@@ -727,7 +727,8 @@ func benchmarkUpsert(nodes []*internaltypes.Node, b *testing.B) {
 	for i, node := range nodes {
 		err = nodeDb.CreateAndInsertWithJobDbJobsWithTxn(txn, nil, node)
 		require.NoError(b, err)
-		entry, err := nodeDb.GetNode(node.GetId())
+		entry, err := nodeDb.GetNodeWithTxn(txn, node.GetId())
+		require.NotNil(b, entry)
 		require.NoError(b, err)
 		entries[i] = entry
 	}