def test_fields_were_updated(client, initial_data, update_data, expected_result):
"""
Note that this does not test associations!
Setup:
Test:
:param client:
:param initial_data: Dict
:param update_data: Dict
:param expected_result: Dict
:return:
"""
# Define our data
input_fields = initial_data
post_response = create_draft_nr(client, input_fields)
# Assign the payload to new nr var
draft_nr = json.loads(post_response.data)
assert draft_nr is not None
# Take the response and edit it
patch_response = patch_nr(client, NameRequestActions.EDIT.value, draft_nr.get('id'), update_data)
patched_nr = json.loads(patch_response.data)
assert patched_nr is not None
print('PATCH Response: \n' + json.dumps(patched_nr, sort_keys=True, indent=4, separators=(',', ': ')) + '\n')
# Check data
for key, value in expected_result.items():
assert_field_equals_value(patched_nr, key, value)
def test_draft_patch_reapply(client, jwt, app):
"""
Setup:
Test:
:param client:
:param jwt:
:param app:
:return:
"""
# Define our data
input_fields = build_test_input_fields()
post_response = create_draft_nr(client, input_fields)
# Assign the payload to new nr var
draft_nr = json.loads(post_response.data)
assert draft_nr is not None
def do_reapply():
# Take the response and edit it
nr_data = {}
patch_response = patch_nr(client, NameRequestActions.REAPPLY.value,
draft_nr.get('id'), nr_data)
updated_nr = None
if patch_response.status_code == 200:
updated_nr = json.loads(patch_response.data)
assert updated_nr is not None
print('PATCH Response: \n' + json.dumps(
updated_nr, sort_keys=True, indent=4, separators=(',', ': ')) +
'\n')
return updated_nr, patch_response.status_code
# Re-apply
patched_nr, status_code = do_reapply()
# Check state
assert patched_nr.get('stateCd') == draft_nr.get('stateCd')
# Check NR number is the same because these are PATCH and call change_nr
assert_field_is_mapped(draft_nr, patched_nr, 'nrNum')
assert_field_equals_value(patched_nr, 'submitCount', 2)
# assert_field_equals_value(patched_nr, 'expirationDate', '')
# Re-apply
patched_nr, status_code = do_reapply()
assert_field_equals_value(patched_nr, 'submitCount', 3)
# Re-apply
patched_nr, status_code = do_reapply()
# The submitCount should never be greater than 3, this should now fail with a 500
assert status_code == 500
def test_draft_patch_edit_request_action_and_entity_type(client, jwt, app):
"""
This is not designed to test all combinations, for those tests see test_nr_request_combos.
Setup:
Test:
:param client:
:param jwt:
:param app:
:return:
"""
# Define our data, which is initially set to:
# test_input_fields = {
# 'request_action_cd': 'NEW'
# 'entity_type_cd': 'CR'
# 'requestTypeCd': 'CR'
# ...
# }
input_fields = build_test_input_fields()
post_response = create_draft_nr(client, input_fields)
# Assign the payload to new nr var
draft_nr = json.loads(post_response.data)
assert draft_nr is not None
# Take the response and edit it
# Change the request action
nr_data = {
'request_action_cd': 'CHG',
'entity_type_cd': 'RLC', # Limited Liability Co.
'requestTypeCd': 'CLC' # From request_type_mapping in namex.constants
}
patch_response = patch_nr(client, NameRequestActions.EDIT.value,
draft_nr.get('id'), nr_data)
patched_nr = json.loads(patch_response.data)
assert patched_nr is not None
print('PATCH Response: \n' + json.dumps(
patched_nr, sort_keys=True, indent=4, separators=(',', ': ')) + '\n')
# Check state
assert patched_nr.get('stateCd') == 'PENDING_PAYMENT'
# Check data
expected_field_values = {
'request_action_cd': 'CHG',
'entity_type_cd': 'RLC',
'requestTypeCd': 'CLC'
}
for key, value in expected_field_values.items():
assert_field_equals_value(patched_nr, key, value)
def test_draft_patch_upgrade(client, jwt, app):
"""
Setup:
Test:
:param client:
:param jwt:
:param app:
:return:
"""
# Define our data
input_fields = build_test_input_fields()
post_response = create_draft_nr(client, input_fields)
# Assign the payload to new nr var
draft_nr = json.loads(post_response.data)
assert draft_nr is not None
assert_field_equals_value(draft_nr, 'priorityCd', 'N')
# Take the response and edit it
nr_data = {}
patch_response = patch_nr(client, NameRequestActions.UPGRADE.value,
draft_nr.get('id'), nr_data)
assert patch_response.status_code == 200
patched_nr = json.loads(patch_response.data)
assert patched_nr is not None
print('PATCH Response: \n' + json.dumps(
patched_nr, sort_keys=True, indent=4, separators=(',', ': ')) + '\n')
# Check state
assert patched_nr.get('stateCd') == State.PENDING_PAYMENT
# Check NR number is the same because these are PATCH and call change_nr
assert_field_is_mapped(draft_nr, patched_nr, 'nrNum')
# Check actions (write a util for this)
# assert_field_equals_value(patched_nr, 'payment_token', '')
assert_field_equals_value(patched_nr, 'priorityCd', 'Y')
def test_draft_patch_edit_data(client, jwt, app):
"""
Test the Name Request's data fields. Excludes associations 'names' and 'applicant' - we have other tests for those.
Setup:
Test:
:param client:
:param jwt:
:param app:
:return:
"""
# Define our data
input_fields = build_test_input_fields()
post_response = create_draft_nr(client, input_fields)
# Assign the payload to new nr var
draft_nr = json.loads(post_response.data)
assert draft_nr.get('stateCd') == 'PENDING_PAYMENT'
assert draft_nr is not None
# Take the response and edit it
# Add another name to the mix
nr_data = {
'names': draft_nr.get('names'),
'applicants': draft_nr.get('applicants'),
'entity_type_cd': 'FR',
'corpNum': 'TESTCORP123',
# 'homeJurisNum': 'TESTHOME123',
'additionalInfo': 'Testing additional info',
'natureBusinessInfo': 'Testing nature of business info',
'priorityCd': 'Y',
'requestTypeCd': 'CR'
}
added_names = [{
'name': 'BLUE HERON ADVENTURE TOURS LTD.',
'choice': 2,
'designation': 'LTD.',
'name_type_cd': 'CO',
'consent_words': '',
'conflict1': 'BLUE HERON TOURS LTD.',
'conflict1_num': '0515211'
}, {
'name': 'BLUE HERON ISLAND TOURS LTD.',
'choice': 3,
'designation': 'LTD.',
'name_type_cd': 'CO',
'consent_words': '',
'conflict1': 'BLUE HERON TOURS LTD.',
'conflict1_num': '0515211'
}]
nr_data['names'].extend(added_names)
# Change the designation, we'll check to make sure it's mapped in the response
nr_data['names'][0]['designation'] = 'INC.'
# TODO: More applicant testing
# updated_applicant = {}
# nr_data['applicant'] = updated_applicant
patch_response = patch_nr(client, NameRequestActions.EDIT.value,
draft_nr.get('id'), nr_data)
patched_nr = json.loads(patch_response.data)
assert patched_nr is not None
print('PATCH Response: \n' + json.dumps(
patched_nr, sort_keys=True, indent=4, separators=(',', ': ')) + '\n')
# Check state
print('Assert that stateCd == DRAFT: ' +
str(bool(patched_nr.get('stateCd') == 'DRAFT')))
assert patched_nr.get('stateCd') == 'PENDING_PAYMENT'
# Check names
assert_names_are_mapped_correctly(nr_data.get('names'),
patched_nr.get('names'))
# Check applicant
assert_applicant_is_mapped_correctly(nr_data.get('applicants'),
patched_nr.get('applicants'))
# Check data
expected_field_values = {
'additionalInfo': 'Testing additional info',
'consentFlag': None,
'consent_dt': None,
'corpNum': 'TESTCORP123',
# 'homeJurisNum': 'TESTHOME123',
'entity_type_cd': 'FR',
'requestTypeCd': 'CR',
'request_action_cd': 'NEW',
'expirationDate': None,
'furnished': 'N',
'hasBeenReset': False,
# 'lastUpdate': None,
'natureBusinessInfo': 'Testing nature of business info',
# 'nrNum': '',
# 'nwpta': '',
# 'previousNr': '',
# 'previousRequestId': '',
# 'previousStateCd': '',
'priorityCd': 'Y',
# 'priorityDate': None,
'source': 'NAMEREQUEST',
'state': 'PENDING_PAYMENT',
'stateCd': 'PENDING_PAYMENT',
'submitCount': 1,
# 'submittedDate': None,
'submitter_userid': 'name_request_service_account',
'userId': 'name_request_service_account',
'xproJurisdiction': ''
}
for key, value in expected_field_values.items():
assert_field_equals_value(patched_nr, key, value)