def test__get_faker_anonymize_category_not_exist(self):
"""Test _get_faker with a category that don't exist"""
# Run & assert
transformer = Mock()
transformer.anonymize = 'SuP3R-P1Th0N-P0w3R'
with self.assertRaises(ValueError):
CategoricalTransformer._get_faker(transformer)
def test__get_faker_anonymize_list_type(self):
"""Test _get_faker when anonymize is a list with two elements"""
# Run
transformer = Mock()
transformer.anonymize = ['credit_card_number', 'visa']
faker_method = CategoricalTransformer._get_faker(transformer)
fake_value = faker_method()
# Asserts
assert isinstance(fake_value, str)
assert len(fake_value) == 16
def test__get_faker_anonymize_not_tuple_or_list(self):
"""Test _get_faker when anonymize is neither a typle or a list"""
# Run
transformer = Mock()
transformer.anonymize = 'email'
result = CategoricalTransformer._get_faker(transformer)
# Asserts
self.assertEqual(
result.__name__,
'faker',
"Expected faker function"
)
def test__get_faker_anonymize_list(self):
"""Test _get_faker when anonymize is a list"""
# Run
transformer = Mock()
transformer.anonymize = ['email']
result = CategoricalTransformer._get_faker(transformer)
# Asserts
self.assertEqual(
result.__name__,
'faker',
"Expected faker function"
)
def test__get_faker_anonymize_tuple(self):
"""Test _get_faker when anonymize is a tuple"""
# Setup
# Run
transformer = Mock()
transformer.anonymize = ('email',)
result = CategoricalTransformer._get_faker(transformer)
# Asserts
self.assertEqual(
result.__name__,
'faker',
"Expected faker function"
)
