Add excluded_types attribute to faker unique proxy

Author: pvk-developer

rdt/transformers/pii/anonymizer.py

@@ -196,6 +196,9 @@ def reset_randomization(self):
 
     def _function(self):
         """Return the result of calling the ``faker`` function."""
+        if not hasattr(self.faker.unique, '_excluded_types'):
+            setattr(self.faker.unique, '_excluded_types', ())  # The default is an empty tuple
+
         try:
             if self.cardinality_rule in {'unique', 'match', 'scale'}:
                 faker_attr = self.faker.unique

tests/unit/transformers/pii/test_anonymizer.py

@@ -101,7 +101,7 @@ def test__function_cardinality_rule_none(self):
             - The returned function, when called, has to call the `faker.<function_name>` function
               with the provided kwargs.
         """
-        # setup
+        # Setup
         instance = Mock()
         function = Mock()
         unique_function = Mock()
@@ -147,7 +147,7 @@ def test__function_cardinality_rule_unique(self):
             - The returned function, when called, has to call the ``faker.unique.<function_name>``
               function with the provided kwargs.
         """
-        # setup
+        # Setup
         instance = Mock()
         function = Mock()
         unique_function = Mock()
@@ -169,7 +169,7 @@ def test__function_cardinality_rule_unique(self):
 
     def test__function_cardinality_rule_match(self):
         """Test it when 'cardinality_rule' is 'match'."""
-        # setup
+        # Setup
         instance = Mock()
         function = Mock()
         unique_function = Mock()
@@ -191,7 +191,7 @@ def test__function_cardinality_rule_match(self):
 
     def test__function_cardinality_rule_missing_attribute(self):
         """Test it when ``cardinality_rule`` attribute is missing."""
-        # setup
+        # Setup
         instance = Mock()
         function = Mock()
         unique_function = Mock()
@@ -214,11 +214,36 @@ def test__function_cardinality_rule_missing_attribute(self):
 
     def test__function_with_iterables_return(self):
         """Test that ``_function`` returns the values of the iterable."""
-        # setup
+        # Setup
+        instance = Mock()
+        instance.cardinality_rule = None
+        function = Mock()
+        function.return_value = ('value_1', 'value_2')
+
+        instance.faker.number = function
+        instance.function_name = 'number'
+        instance.function_kwargs = {'type': 'int'}
+
+        # Run
+        result = AnonymizedFaker._function(instance)
+
+        # Assert
+        function.assert_called_once_with(type='int')
+        assert result == 'value_1, value_2'
+
+    def test__function_sets__excluded_types(self):
+        """Test that ``_function`` will set the `_excluded_types` attribute to `faker`."""
+
+        # Setup
+        class Dummy:
+            pass
+
         instance = Mock()
         instance.cardinality_rule = None
         function = Mock()
         function.return_value = ('value_1', 'value_2')
+        instance.faker.unique = Dummy()
+        assert hasattr(instance.faker.unique, '_excluded_types') is False
 
         instance.faker.number = function
         instance.function_name = 'number'
@@ -230,6 +255,7 @@ def test__function_with_iterables_return(self):
         # Assert
         function.assert_called_once_with(type='int')
         assert result == 'value_1, value_2'
+        assert hasattr(instance.faker.unique, '_excluded_types')
 
     @patch('rdt.transformers.pii.anonymizer.importlib')
     @patch('rdt.transformers.pii.anonymizer.warnings')