Skip to content

Commit 07be76a

Browse files
committed
VED-901: Update tests.
1 parent 06eee29 commit 07be76a

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

lambdas/filenameprocessor/tests/test_lambda_handler.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -288,8 +288,8 @@ def test_lambda_handler_extended_attributes_success(self, mock_get_redis_client)
288288
archived_obj = s3_client.get_object(Bucket=BucketNames.SOURCE, Key=archived_key)
289289
self.assertIsNotNone(archived_obj)
290290

291-
# Also verify file copied to DPS destination bucket under dps_destination/<file_key>
292-
dps_key = f"dps_destination/{test_cases[0].file_key}"
291+
# Also verify file copied to DPS destination bucket under generic/EXTENDED_ATTRIBUTES_DAILY_1/<file_key>
292+
dps_key = f"generic/EXTENDED_ATTRIBUTES_DAILY_1/{test_cases[0].file_key}"
293293
copied_obj = s3_client.get_object(Bucket=BucketNames.DPS_DESTINATION, Key=dps_key)
294294
self.assertIsNotNone(copied_obj)
295295

@@ -463,7 +463,7 @@ def test_lambda_handler_extended_attributes_extension_checks(self, mock_get_redi
463463
# Ensure processed path hit by checking archive move in source bucket
464464
s3_client.get_object(Bucket=BucketNames.SOURCE, Key=f"extended-attributes-archive/{csv_key}")
465465
# And verify copy to DPS destination
466-
s3_client.get_object(Bucket=BucketNames.DPS_DESTINATION, Key=f"dps_destination/{csv_key}")
466+
s3_client.get_object(Bucket=BucketNames.DPS_DESTINATION, Key=f"generic/EXTENDED_ATTRIBUTES_DAILY_1/{csv_key}")
467467

468468
# .DAT accepted
469469
dat_key = MockFileDetails.extended_attributes_file.file_key[:-3] + "dat"
@@ -474,7 +474,7 @@ def test_lambda_handler_extended_attributes_extension_checks(self, mock_get_redi
474474
):
475475
lambda_handler(self.make_event([self.make_record(dat_key)]), None)
476476
s3_client.get_object(Bucket=BucketNames.SOURCE, Key=f"extended-attributes-archive/{dat_key}")
477-
s3_client.get_object(Bucket=BucketNames.DPS_DESTINATION, Key=f"dps_destination/{dat_key}")
477+
s3_client.get_object(Bucket=BucketNames.DPS_DESTINATION, Key=f"generic/EXTENDED_ATTRIBUTES_DAILY_1/{dat_key}")
478478

479479
# Invalid extension fails
480480
bad_ext_key = csv_key[:-3] + "txt"

0 commit comments

Comments
 (0)