Skip to content

Commit 852dba8

Browse files
committed
test record processor amendments
1 parent 5134e22 commit 852dba8

File tree

3 files changed

+7
-18
lines changed

3 files changed

+7
-18
lines changed

recordprocessor/tests/utils_for_recordprocessor_tests/generic_setup_and_teardown.py

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
"""Generic setup and teardown for recordprocessor tests"""
22

33
from unittest.mock import patch
4-
from boto3 import client as boto3_client
54

65
from tests.utils_for_recordprocessor_tests.mock_environment_variables import (
76
BucketNames,
@@ -13,13 +12,7 @@
1312
# Ensure environment variables are mocked before importing from src files
1413
with patch.dict("os.environ", MOCK_ENVIRONMENT_DICT):
1514
from clients import REGION_NAME
16-
from constants import (
17-
AuditTableKeys,
18-
AUDIT_TABLE_QUEUE_NAME_GSI,
19-
AUDIT_TABLE_FILENAME_GSI,
20-
AUDIT_TABLE_NAME,
21-
AUDIT_TABLE_NAME,
22-
)
15+
from constants import AuditTableKeys, AUDIT_TABLE_QUEUE_NAME_GSI, AUDIT_TABLE_FILENAME_GSI, AUDIT_TABLE_NAME
2316

2417

2518
class GenericSetUp:
@@ -87,7 +80,6 @@ def __init__(self, s3_client=None, firehose_client=None, sqs_client=None, dynamo
8780
},
8881
],
8982
)
90-
print(f"set up: {AUDIT_TABLE_NAME}")
9183

9284

9385
class GenericTearDown:
@@ -113,5 +105,4 @@ def __init__(self, s3_client=None, firehose_client=None, sqs_client=None, dynamo
113105
sqs_client.delete_queue(QueueUrl=Sqs.TEST_QUEUE_URL)
114106

115107
if dynamodb_client:
116-
print(f"TEARTHISDOWN: {AUDIT_TABLE_NAME}")
117108
dynamodb_client.delete_table(TableName=AUDIT_TABLE_NAME)

recordprocessor/tests/utils_for_recordprocessor_tests/utils_for_recordprocessor_tests.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,6 @@
11
"""Utils for the recordprocessor tests"""
22

3-
from csv import DictReader
43
from io import StringIO
5-
from tests.utils_for_recordprocessor_tests.values_for_recordprocessor_tests import REGION_NAME
64
from tests.utils_for_recordprocessor_tests.mock_environment_variables import BucketNames, Firehose, Kinesis
75
from tests.utils_for_recordprocessor_tests.values_for_recordprocessor_tests import MockFileDetails, FileDetails
86
from boto3.dynamodb.types import TypeDeserializer

recordprocessor/tests/utils_for_recordprocessor_tests/values_for_recordprocessor_tests.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -187,15 +187,15 @@ def __init__(self, vaccine_type: str, supplier: str, ods_code: str, file_number:
187187
class MockFileDetails:
188188
"""Class containing mock file details for use in tests"""
189189

190-
ravs_rsv_1 = FileDetails("RAVS", "RSV", "X26", file_number=1)
191-
ravs_rsv_2 = FileDetails("RAVS", "RSV", "X26", file_number=2)
192-
ravs_rsv_3 = FileDetails("RAVS", "RSV", "X26", file_number=3)
193-
ravs_rsv_4 = FileDetails("RAVS", "RSV", "X26", file_number=4)
194-
ravs_rsv_5 = FileDetails("RAVS", "RSV", "X26", file_number=5)
190+
ravs_rsv_1 = FileDetails("RSV", "RAVS", "X26", file_number=1)
191+
ravs_rsv_2 = FileDetails("RSV", "RAVS", "X26", file_number=2)
192+
ravs_rsv_3 = FileDetails("RSV", "RAVS", "X26", file_number=3)
193+
ravs_rsv_4 = FileDetails("RSV", "RAVS", "X26", file_number=4)
194+
ravs_rsv_5 = FileDetails("RSV", "RAVS", "X26", file_number=5)
195195
rsv_ravs = FileDetails("RSV", "RAVS", "X26")
196196
rsv_emis = FileDetails("RSV", "EMIS", "8HK48")
197197
flu_emis = FileDetails("FLU", "EMIS", "YGM41")
198-
ravs_flu = FileDetails("RAVS", "FLU", "X26")
198+
ravs_flu = FileDetails("FLU", "RSV", "X26")
199199

200200

201201
class UnorderedFieldDictionaries:

0 commit comments

Comments
 (0)