Fix unit test

Signed-off-by: droctothorpe <[email protected]>
kubeflow · Aug 15, 2024 · fbf5f36 · fbf5f36
1 parent e7e57b9
commit fbf5f36
Showing 1 changed file with 12 additions and 9 deletions.
diff --git a/frontend/server/workflow-helper.test.ts b/frontend/server/workflow-helper.test.ts
@@ -147,6 +147,17 @@ describe('workflow-helper', () => {
         apiVersion: 'argoproj.io/v1alpha1',
         kind: 'Workflow',
         status: {
+          artifactRepositoryRef: {
+            artifactRepository: { archiveLogs: true },
+            s3: {
+              accessKeySecret: { key: 'accessKey', name: 'accessKeyName' },
+              bucket: 'bucket',
+              endpoint: 'minio-service.kubeflow',
+              insecure: true,
+              key: 'prefix/workflow-name/workflow-name-abc/some-artifact.csv',
+              secretKeySecret: { key: 'secretKey', name: 'secretKeyName' },
+            },
+          },
           nodes: {
             'workflow-name-abc': {
               outputs: {
@@ -163,16 +174,8 @@ describe('workflow-helper', () => {
                     },
                   },
                   {
-                    archiveLogs: true,
                     name: 'main.log',
-                    s3: {
-                      accessKeySecret: { key: 'accessKey', name: 'accessKeyName' },
-                      bucket: 'bucket',
-                      endpoint: 'minio-service.kubeflow',
-                      insecure: true,
-                      key: 'prefix/workflow-name/workflow-name-abc/main.log',
-                      secretKeySecret: { key: 'secretKey', name: 'secretKeyName' },
-                    },
+                    s3: { key: 'prefix/workflow-name/workflow-name-abc/main.log' },
                   },
                 ],
               },