def print_message_callback(ch, method, properties, body, partial_files_directory=Config.PARTIAL_FILES_DIRECTORY):
try:
generate_print_row(body, partial_files_directory)
except Exception as e:
handle_message_error(body, e, ch, method.delivery_tag, properties)
return
ch.basic_ack(delivery_tag=method.delivery_tag)
def test_action_type_mappings(cleanup_test_files, action_type):
# Given
json_body = json.dumps({
"actionType": action_type.value,
# Use a hardcoded pack code just to test all the action types, we don't care here if it's correct
"packCode": "P_IC_ICL1",
"batchId": "1",
"batchQuantity": 1,
"uac": "test_uac",
"qid": "test_qid",
"uacWales": "test_wales_uac",
"qidWales": "test_wales_qid",
"caseRef": "test_caseref", # NB: ignored
"fieldCoordinatorId": "test_qm_coordinator_id",
"addressLine1": "123 Fake Street",
"addressLine2": "Duffryn",
"townName": "Newport",
"postcode": "NPXXXX",
"organisationName": "ONS",
"fieldOfficerId": "012345678"
})
if ACTION_TYPE_TO_PRINT_TEMPLATE[action_type] == PrintTemplate.QM_QUESTIONNAIRE_TEMPLATE:
expected_row = ('test_uac|test_qid|test_wales_uac|test_wales_qid|test_qm_coordinator_id|'
'|||123 Fake Street|Duffryn||Newport|NPXXXX|P_IC_ICL1|ONS|012345678\n')
else:
expected_row = ('test_uac|test_caseref||||123 Fake Street|Duffryn||Newport|NPXXXX|'
'P_IC_ICL1|test_qid|ONS|test_qm_coordinator_id|012345678\n')
# When
generate_print_row(json_body, cleanup_test_files.partial_files)
# Then
generated_print_file = cleanup_test_files.partial_files.joinpath(f'{action_type.value}.P_IC_ICL1.1.1')
assert generated_print_file.read_text() == expected_row
def test_generate_print_row_valid_ICL1E(cleanup_test_files):
# Given
json_body = json.dumps({
"actionType": "ICL1E",
"batchId": "1",
"batchQuantity": 3,
"uac": "test_uac",
"caseRef": "test_caseref",
"addressLine1": "123 Fake Street",
"addressLine2": "Duffryn",
"townName": "Newport",
"postcode": "NPXXXX",
"packCode": "P_IC_ICL1",
"qid": "",
"organisationName": "",
"fieldCoordinatorId": "",
"fieldOfficerId": ""
})
# When
generate_print_row(json_body, cleanup_test_files.partial_files)
# Then
generated_print_file = cleanup_test_files.partial_files.joinpath('ICL1E.P_IC_ICL1.1.3')
assert generated_print_file.read_text() == ('test_uac|test_caseref||||123 Fake Street'
'|Duffryn||Newport|NPXXXX|P_IC_ICL1||||\n')
def test_generate_print_row_valid_SPG_IC14(cleanup_test_files):
# Given
json_body = json.dumps({
"actionType": "SPG_IC14",
"batchId": "1",
"batchQuantity": 3,
"uac": "test_uac",
"qid": "test_qid",
"uacWales": "test_wales_uac",
"qidWales": "test_wales_qid",
"fieldCoordinatorId": "test_qm_coordinator_id",
"addressLine1": "123 Fake Street",
"addressLine2": "Duffryn",
"townName": "Newport",
"postcode": "NPXXXX",
"packCode": "D_FDCE_H2",
"organisationName": "ONS",
"fieldOfficerId": "012345678"
})
# When
generate_print_row(json_body, cleanup_test_files.partial_files)
# Then
generated_print_file = cleanup_test_files.partial_files.joinpath('SPG_IC14.D_FDCE_H2.1.3')
assert generated_print_file.read_text() == (
'test_uac|test_qid|test_wales_uac|test_wales_qid|test_qm_coordinator_id||||123 Fake Street|Duffryn||Newport'
'|NPXXXX|D_FDCE_H2|ONS|012345678\n')
def test_generate_print_row_valid_SPG_IC12(cleanup_test_files):
# Given
json_body = json.dumps({
"actionType": "SPG_IC12",
"batchId": "1",
"batchQuantity": 3,
"uac": "test_uac",
"caseRef": "test_caseref",
"addressLine1": "123 Fake Street",
"addressLine2": "Duffryn",
"townName": "Newport",
"postcode": "NPXXXX",
"packCode": "P_ICCE_ICL2B",
"qid": "02000000005555",
"organisationName": "ONS",
"fieldCoordinatorId": "123456789",
"fieldOfficerId": "012345678"
})
# When
generate_print_row(json_body, cleanup_test_files.partial_files)
# Then
generated_print_file = cleanup_test_files.partial_files.joinpath('SPG_IC12.P_ICCE_ICL2B.1.3')
assert generated_print_file.read_text() == ('test_uac|test_caseref||||123 Fake Street'
'|Duffryn||Newport|NPXXXX|P_ICCE_ICL2B|02000000005555'
'|ONS|123456789|012345678\n')
def test_pack_code_mappings(cleanup_test_files, pack_code):
# Given
json_body = json.dumps({
"packCode": pack_code,
# Use a hardcoded action type just to test all the pack codes, we don't care here if it's correct
"actionType": "ICHHQE",
"batchId": "1",
"batchQuantity": 1,
"uac": "test_uac",
"qid": "test_qid",
"uacWales": "test_wales_uac",
"qidWales": "test_wales_qid",
"caseRef": "test_caseref", # NB: ignored
"fieldCoordinatorId": "test_qm_coordinator_id",
"addressLine1": "123 Fake Street",
"addressLine2": "Duffryn",
"townName": "Newport",
"postcode": "NPXXXX",
"organisationName": "ONS",
"fieldOfficerId": "012345678"
})
# When
generate_print_row(json_body, cleanup_test_files.partial_files)
# Then
generated_print_file = cleanup_test_files.partial_files.joinpath(f'ICHHQE.{pack_code}.1.1')
assert generated_print_file.read_text() == (
'test_uac|test_qid|test_wales_uac|test_wales_qid|test_qm_coordinator_id|'
f'|||123 Fake Street|Duffryn||Newport|NPXXXX|{pack_code}|ONS|012345678\n')
def test_generate_print_row_valid_ICHHQE(cleanup_test_files):
# Given
json_body = json.dumps({
"actionType": "ICHHQE",
"batchId": "1",
"batchQuantity": 3,
"uac": "test_uac",
"qid": "test_qid",
"uacWales": "test_wales_uac",
"qidWales": "test_wales_qid",
"caseRef": "test_caseref", # NB: ignored
"fieldCoordinatorId": "test_qm_coordinator_id",
"addressLine1": "123 Fake Street",
"addressLine2": "Duffryn",
"townName": "Newport",
"postcode": "NPXXXX",
"packCode": "P_IC_H1",
"organisationName": "ONS",
"fieldOfficerId": "012345678"
})
# When
generate_print_row(json_body, cleanup_test_files.partial_files)
# Then
generated_print_file = cleanup_test_files.partial_files.joinpath('ICHHQE.P_IC_H1.1.3')
assert generated_print_file.read_text() == (
'test_uac|test_qid|test_wales_uac|test_wales_qid|test_qm_coordinator_id|'
'|||123 Fake Street|Duffryn||Newport|NPXXXX|P_IC_H1|ONS|012345678\n')
def test_generate_print_row_invalid_supplier(cleanup_test_files):
# Given
json_body = json.dumps({
"packCode": pack_code,
"supplier": "NOT_A_VALID_SUPPLIER",
"batchId": "1",
"batchQuantity": 3,
"row": "test|example",
})
# When/Then
with pytest.raises(MalformedMessageError):
generate_print_row(json_body, cleanup_test_files.partial_files,
TestConfig.SUPPLIERS_CONFIG)
def test_generate_print_row_invalid_action_type(cleanup_test_files):
# Given
json_body = json.dumps({
"actionType": "NOT_A_VALID_ACTION_TYPE",
"batchId": "1",
"batchQuantity": 3,
"uac": "test_uac",
"caseRef": "c",
"addressLine1": "123 Fake Street",
"addressLine2": "Duffryn",
"townName": "Newport",
"postcode": "NPXXXX",
"packCode": "PC_ICL_1"
})
# When/Then
with pytest.raises(MalformedMessageError):
generate_print_row(json_body, cleanup_test_files.partial_files)
def test_generate_print_row(cleanup_test_files):
# Given
row = 'test|data|example'
json_body = json.dumps({
"packCode": pack_code,
"printSupplier": supplier,
"batchId": "1",
"batchQuantity": 1,
"row": row
})
# When
generate_print_row(json_body, cleanup_test_files.partial_files,
TestConfig.SUPPLIERS_CONFIG)
# Then
generated_print_file = cleanup_test_files.partial_files.joinpath(
f'{supplier}.{pack_code}.1.1')
assert generated_print_file.read_text() == (row + '\n')
