test

Author: yangw-dev

torchci/clickhouse_queries/oss_ci_util_metadata/params.json

@@ -1,7 +1,10 @@
 {
-  "workflowId": "UInt64",
-  "jobId": "UInt64",
-  "runAttempt": "UInt32",
-  "repo": "String",
-  "type": "String"
+  "params": {
+    "workflowId": "UInt64",
+    "jobId": "UInt64",
+    "runAttempt": "UInt32",
+    "repo": "String",
+    "type": "String"
+  },
+  "tests":[]
 }

torchci/clickhouse_queries/oss_ci_util_ts/params.json

@@ -1,7 +1,10 @@
 {
-  "workflowId": "UInt64",
-  "jobId": "UInt64",
-  "runAttempt": "UInt32",
-  "repo": "String",
-  "type": "String"
+  "params": {
+    "workflowId": "UInt64",
+    "jobId": "UInt64",
+    "runAttempt": "UInt32",
+    "repo": "String",
+    "type": "String"
+  },
+  "tests":[]
 }

torchci/lib/utilization/fetchUtilization.test.ts

@@ -270,10 +270,25 @@ describe("fetchUtilization", () => {
     // Assert query sent to clickhouse
     expect(mockQuery).toHaveBeenCalledTimes(2);
     const firstQuery = mockQuery.mock.calls[0][0];
+    expect(firstQuery.query_params).toEqual({
+      workflowId: "1234",
+      jobId: "2345",
+      runAttempt: "1",
+      repo: "pytorch/pytorch",
+      type: "utilization",
+    });
+
     expect(
       firstQuery.query.includes("oss_ci_utilization_metadata")
     ).toBeTruthy();
     const secondQuery = mockQuery.mock.calls[1][0];
+    expect(secondQuery.query_params).toEqual({
+      workflowId: "1234",
+      jobId: "2345",
+      runAttempt: "1",
+      repo: "pytorch/pytorch",
+      type: "utilization",
+    });
     expect(secondQuery.query.includes("oss_ci_time_series")).toBeTruthy();
   });
 });