def test_fuzzy_should_mask_token_keys(fixture):
example = json.dumps(fixture)
masked_key = mask_values(example, ['token'], 'telepresence')
assert_that(json.loads(masked_key)['token'], equal_to('telepresence'))
assert_that(
json.loads(masked_key)['object']['token'], equal_to('telepresence')
)
def test_should_mask_multiple_keys(source, unmarshal):
masked_key = mask_values(source, ['_id', 'token'], 'telepresence')
as_json = unmarshal(masked_key)
assert_that(as_json['_id'], equal_to('telepresence'))
assert_that(as_json['token'], equal_to('telepresence'))
assert_that(as_json['object']['token'], equal_to('telepresence'))
def test_should_leave_source_unchanged_when_masking_non_existing_key(source):
masked = mask_values(source, ['_not_existing'], 'telepresence')
assert_that(masked, equal_to(source))
def test_non_existing_key(source):
"""Should leave source unchanged when masking a non-existing key"""
masked = mask_values(source, ['_not_existing'], 'telepresence')
assert_that(masked, equal_to(source))