Break out random workspace resource

code-asher · code-asher · commit 990dfb5d29e0 · 2023-10-18T10:39:38.000-08:00
This will be used in a test.
diff --git a/src/test/groovy/DataGen.groovy b/src/test/groovy/DataGen.groovy
@@ -31,10 +31,8 @@ class DataGen {
         )
     }
 
-    static Workspace workspace(String name, Map<String, String> agents = [:]) {
-        UUID wsId = UUID.randomUUID()
-        UUID ownerId = UUID.randomUUID()
-        List<WorkspaceResource> resources = agents.collect{ agentName, agentId -> new WorkspaceResource(
+    static WorkspaceResource resource(String agentName, String agentId){
+        return new WorkspaceResource(
                 UUID.randomUUID(),      // id
                 new Date().toInstant(), // created_at
                 UUID.randomUUID(),      // job_id
@@ -70,7 +68,13 @@ class DataGen {
                 )),
                 null,                   // metadata
                 0,                      // daily_cost
-        )}
+        )
+    }
+
+    static Workspace workspace(String name, Map<String, String> agents = [:]) {
+        UUID wsId = UUID.randomUUID()
+        UUID ownerId = UUID.randomUUID()
+        List<WorkspaceResource> resources = agents.collect{ resource(it.key, it.value)}
         return new Workspace(
                 wsId,
                 new Date().toInstant(), // created_at
