def test_sanitize_json_values_handles_list_of_str(monkeypatch):
def mock_sanitizeString(some_string):
return some_string + ' sanitized'
monkeypatch.setattr(etlMods, 'sanitizeString', mock_sanitizeString)
given_list = ['item1', 'item2']
actual_list = etlMods._sanitize_json_values(given_list)
expect_list = ['item1 sanitized', 'item2 sanitized']
assert expect_list == actual_list
def test_sanitize_json_values_handles_flat_dict(monkeypatch):
def mock_sanitizeString(some_string):
return some_string + ' sanitized'
monkeypatch.setattr(etlMods, 'sanitizeString', mock_sanitizeString)
given_dict = {'key1': 'val1', 'key2': 'val2'}
actual_dict = etlMods._sanitize_json_values(given_dict)
expect_dict = {'key1': 'val1 sanitized', 'key2': 'val2 sanitized'}
assert expect_dict == actual_dict
def test_sanitize_json_values_handles_dict_containing_list(monkeypatch):
def mock_sanitizeString(some_string):
return some_string + ' sanitized'
monkeypatch.setattr(etlMods, 'sanitizeString', mock_sanitizeString)
given_dict = {'key1': 'val1', 'key2': ['item1', 'item2']}
actual_dict = etlMods._sanitize_json_values(given_dict)
expect_dict = {
'key1': 'val1 sanitized',
'key2': ['item1 sanitized', 'item2 sanitized']
}
assert expect_dict == actual_dict
def test_sanitize_json_values_handles_list_of_dict(monkeypatch):
def mock_sanitizeString(some_string):
return some_string + ' sanitized'
monkeypatch.setattr(etlMods, 'sanitizeString', mock_sanitizeString)
given_list = [{
'name': 'valuea',
'provider': 'valueb'
}, {
'name': 'aname',
'provider': 'aprovider'
}]
actual_list = etlMods._sanitize_json_values(given_list)
expect_list = [{
'name': 'valuea sanitized',
'provider': 'valueb sanitized'
}, {
'name': 'aname sanitized',
'provider': 'aprovider sanitized'
}]
assert expect_list == actual_list
def test_sanitize_json_values_does_not_over_recurse():
L = []
L.extend([L])
actual_list = etlMods._sanitize_json_values(L, recursion_limit=3)
expect_list = [[['[[...]]']]]
assert actual_list == expect_list