def test_in_set_invalid():
# Given
in_set_validator = validators.in_set({'a'}, label=test_label)
# When, then raises
with pytest.raises(validators.Invalid):
in_set_validator('abc')
class RefusalProcessor(Processor):
file_prefix = Config.BULK_REFUSAL_FILE_PREFIX
routing_key = Config.REFUSAL_EVENT_ROUTING_KEY
exchange = Config.EVENTS_EXCHANGE
bucket_name = Config.BULK_REFUSAL_BUCKET_NAME
project_id = Config.BULK_REFUSAL_PROJECT_ID
schema = {
"case_id": [is_uuid(), case_exists_by_id()],
"refusal_type": [in_set({"HARD_REFUSAL", "EXTRAORDINARY_REFUSAL"}, label='refusal types')]
}
def build_event_messages(self, row):
return [{
"event": {
"type": "REFUSAL_RECEIVED",
"source": "RM_BULK_REFUSAL_PROCESSOR",
"channel": "RM",
"dateTime": datetime.utcnow().isoformat() + 'Z',
"transactionId": str(uuid.uuid4())
},
"payload": {
"refusal": {
"type": row['refusal_type'],
"collectionCase": {
"id": row['case_id']
}
}
}
}]
def test_in_set_valid():
# Given
in_set_validator = validators.in_set({'a', 'b', 'c'}, label=test_label)
# When
in_set_validator('a')
in_set_validator('b')
in_set_validator('c')
class NonComplianceProcessor(Processor):
file_prefix = Config.BULK_NON_COMPLIANCE_FILE_PREFIX
routing_key = Config.BULK_NON_COMPLIANCE_ROUTING_KEY
exchange = ''
bucket_name = Config.BULK_NON_COMPLIANCE_BUCKET_NAME
project_id = Config.BULK_NON_COMPLIANCE_PROJECT_ID
schema = {
"CASE_ID": [is_uuid(), hh_case_exists_by_id()],
"NC_STATUS":
[in_set({"NCL", "NCF", "NCFW"}, label='non-compliance status')],
"FIELDCOORDINATOR_ID": [
max_length(10),
no_padding_whitespace(),
no_pipe_character(),
alphanumeric_plus_hyphen_field_values_ignore_empty_strings()
],
"FIELDOFFICER_ID": [
max_length(13),
no_padding_whitespace(),
no_pipe_character(),
alphanumeric_plus_hyphen_field_values_ignore_empty_strings()
]
}
def build_event_messages(self, row):
return [{
"event": {
"type": "SELECTED_FOR_NON_COMPLIANCE",
"source": "NON_COMPLIANCE",
"channel": "NC",
"dateTime": datetime.utcnow().isoformat() + 'Z',
"transactionId": str(uuid.uuid4())
},
"payload": {
"collectionCase": {
"id": row['CASE_ID'],
"fieldCoordinatorId": row['FIELDCOORDINATOR_ID'],
"fieldOfficerId": row['FIELDOFFICER_ID'],
"nonComplianceStatus": row['NC_STATUS'],
}
}
}]
class AddressUpdateProcessor(Processor):
file_prefix = Config.BULK_ADDRESS_UPDATE_FILE_PREFIX
routing_key = Config.BULK_ADDRESS_UPDATE_ROUTING_KEY
exchange = ''
bucket_name = Config.BULK_ADDRESS_UPDATE_BUCKET_NAME
project_id = Config.BULK_ADDRESS_UPDATE_PROJECT_ID
schema = {
'CASE_ID': [mandatory(), case_exists_by_id()],
'UPRN': [
max_length(13),
numeric(),
no_padding_whitespace(),
cant_be_deleted(),
mandatory_after_update('uprn')
],
'ESTAB_UPRN': [
max_length(13),
numeric(),
no_padding_whitespace(),
cant_be_deleted(),
mandatory_after_update('estab_uprn')
],
'ESTAB_TYPE':
[mandatory(),
in_set(Config.ESTAB_TYPES, label='ESTAB_TYPE')],
'ABP_CODE': [
max_length(6),
no_padding_whitespace(),
no_pipe_character(),
cant_be_deleted(),
mandatory_after_update('abp_code')
],
'ORGANISATION_NAME': [
max_length(60),
no_padding_whitespace(),
no_pipe_character(),
check_delete_keyword()
],
'ADDRESS_LINE1': [
max_length(60),
no_padding_whitespace(),
no_pipe_character(),
cant_be_deleted(),
mandatory_after_update('address_line1')
],
'ADDRESS_LINE2': [
max_length(60),
no_padding_whitespace(),
no_pipe_character(),
check_delete_keyword()
],
'ADDRESS_LINE3': [
max_length(60),
no_padding_whitespace(),
no_pipe_character(),
check_delete_keyword()
],
'TOWN_NAME': [
max_length(30),
no_padding_whitespace(),
no_pipe_character(),
cant_be_deleted(),
mandatory_after_update('town_name')
],
'POSTCODE': [
max_length(8),
no_padding_whitespace(),
cant_be_deleted(),
alphanumeric_postcode(),
no_pipe_character(),
mandatory_after_update('postcode')
],
'LATITUDE': [
mandatory(),
latitude_longitude(max_scale=7, max_precision=9),
cant_be_deleted(),
no_padding_whitespace(),
no_pipe_character(),
latitude_longitude_range()
],
'LONGITUDE': [
mandatory(),
latitude_longitude(max_scale=7, max_precision=8),
cant_be_deleted(),
no_padding_whitespace(),
no_pipe_character(),
latitude_longitude_range()
],
'OA': [
mandatory(),
max_length(9),
no_padding_whitespace(),
no_pipe_character(),
cant_be_deleted()
],
'LSOA': [
mandatory(),
max_length(9),
no_padding_whitespace(),
no_pipe_character(),
cant_be_deleted()
],
'MSOA': [
mandatory(),
max_length(9),
no_padding_whitespace(),
no_pipe_character(),
cant_be_deleted()
],
'LAD': [
mandatory(),
max_length(9),
no_padding_whitespace(),
no_pipe_character(),
cant_be_deleted()
],
'HTC_WILLINGNESS': [
in_set({'0', '1', '2', '3', '4', '5'}, label='HTC_WILLINGNESS'),
mandatory_after_update('htc_willingness')
],
'HTC_DIGITAL': [
in_set({'0', '1', '2', '3', '4', '5'}, label='HTC_DIGITAL'),
mandatory_after_update('htc_digital')
],
'TREATMENT_CODE':
[mandatory(),
in_set(Config.TREATMENT_CODES, label='TREATMENT_CODE')],
'FIELDCOORDINATOR_ID': [
mandatory(),
max_length(10),
no_padding_whitespace(),
no_pipe_character(),
cant_be_deleted()
],
'FIELDOFFICER_ID': [
mandatory(),
max_length(13),
no_padding_whitespace(),
no_pipe_character(),
cant_be_deleted()
],
'CE_EXPECTED_CAPACITY':
[numeric(), max_length(4),
no_padding_whitespace()],
'CE_SECURE': [
optional_in_set({'0', '1'}, label='CE_SECURE'),
no_padding_whitespace()
],
'PRINT_BATCH': [
no_padding_whitespace(),
numeric_2_digit_or_delete(),
check_delete_keyword()
]
}
def build_event_messages(self, row):
event_message = {
"event": {
"type": "RM_CASE_UPDATED",
"source": "RM_BULK_ADDRESS_UPDATE_PROCESSOR",
"channel": address_resolution,
"dateTime": datetime.utcnow().isoformat() + 'Z',
"transactionId": str(uuid.uuid4())
},
"payload": {
"rmCaseUpdated": {
# Set the mandatory values up front
'caseId': row['CASE_ID'],
'treatmentCode': row['TREATMENT_CODE'],
'estabType': row['ESTAB_TYPE'],
'oa': row['OA'],
'lsoa': row['LSOA'],
'msoa': row['MSOA'],
'lad': row['LAD'],
'fieldCoordinatorId': row['FIELDCOORDINATOR_ID'],
'fieldOfficerId': row['FIELDOFFICER_ID'],
'latitude': row['LATITUDE'],
'longitude': row['LONGITUDE'],
}
}
}
# Set the optional values if present
event_message = set_optional_values_if_present(row, event_message)
# Set the deletable fields
event_message = set_deleteable_fields(row, event_message)
return [event_message]