Update tests

Signed-off-by: Peter Nied <[email protected]>
peternied · Aug 13, 2024 · 0c15414 · 0c15414
1 parent 886bd94
commit 0c15414
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/...oyment/cdk/opensearch-service-migration/lib/service-stacks/reindex-from-snapshot-stack.ts b/...oyment/cdk/opensearch-service-migration/lib/service-stacks/reindex-from-snapshot-stack.ts
@@ -90,6 +90,7 @@ export class ReindexFromSnapshotStack extends MigrationServiceCore {
             servicePolicies.push(getSecretsPolicy);
         }
 
+        const deployId = props.addOnMigrationDeployId ? props.addOnMigrationDeployId : props.defaultDeployId
         this.createService({
             serviceName: 'reindex-from-snapshot',
             taskInstanceCount: 0,
@@ -106,6 +107,7 @@ export class ReindexFromSnapshotStack extends MigrationServiceCore {
                 "RFS_TARGET_USER": targetUser,
                 "RFS_TARGET_PASSWORD": targetPassword,
                 "RFS_TARGET_PASSWORD_ARN": targetPasswordArn,
+                "TUPLE_DIR_PATH": `/shared-replayer-output/traffic-replayer-${deployId}`
             },
             ...props
         });

diff --git a/deployment/cdk/opensearch-service-migration/test/reindex-from-snapshot-stack.test.ts b/deployment/cdk/opensearch-service-migration/test/reindex-from-snapshot-stack.test.ts
@@ -128,6 +128,10 @@ describe('ReindexFromSnapshotStack Tests', () => {
       {
         Name: 'RFS_TARGET_PASSWORD_ARN',
         Value: 'arn:aws:secretsmanager:us-east-1:123456789012:secret:test-secret'
+      },
+      {
+        Name: 'TUPLE_DIR_PATH',
+        Value: '/shared-replayer-output/traffic-replayer-default'
       }
     ]);
   });
@@ -207,6 +211,10 @@ describe('ReindexFromSnapshotStack Tests', () => {
       {
         Name: 'RFS_TARGET_PASSWORD_ARN',
         Value: ''
+      },
+      {
+        Name: 'TUPLE_DIR_PATH',
+        Value: ''
       }
     ]);
   });