Merge pull request #13 from ezzy1337/hacktoberfest/add-unit-tests

Hacktoberfest/add unit tests

Author: marazt

.gitignore

@@ -57,3 +57,4 @@ target/
 
 # Idea
 .idea
+.python-version

mapper/object_mapper.py

@@ -107,7 +107,7 @@ def create_map(self, type_from, type_to, mapping=None):
             self.mappings[key_from] = {}
             self.mappings[key_from][key_to] = mapping
 
-    def map(self, from_obj, to_type, ignore_case=False, allow_none=False):
+    def map(self, from_obj, to_type, ignore_case=False, allow_none=False, excluded=None):
         """Method for creating target object instance
 
         :param from_obj: source object to be mapped from
@@ -127,10 +127,15 @@ def map(self, from_obj, to_type, ignore_case=False, allow_none=False):
         key_from = from_obj.__class__.__name__
         key_to = to_type.__name__
 
-        def not_private(s): return not s.startswith('_')
+        def not_private(s):
+            return not s.startswith('_')
+
+        def not_excluded(s):
+            return not (excluded and s in excluded)
 
         from_obj_attributes = getmembers(from_obj, lambda a: not isroutine(a))
-        from_obj_dict = {k: v for k, v in from_obj_attributes if not_private(k)}
+        from_obj_dict = {k: v for k, v in from_obj_attributes
+                         if not_private(k) and not_excluded(k)}
 
         to_obj_attributes = getmembers(inst, lambda a: not isroutine(a))
         to_obj_dict = {k: v for k, v in to_obj_attributes if not_private(k)}

tests/object_mapper_test.py

@@ -300,4 +300,21 @@ def __setattr__(self, name, value):
         self.assertTrue(isinstance(result, ToCustomDirClass), "Target types must be same")
         self.assertEqual(result.name, from_class.name, "Name mapping must be equal")
         self.assertEqual(result.date, from_class.date, "Date mapping must be equal")
-        self.assertNotIn("surname", dir(result), "To class must not contain surname")
+        self.assertNotIn("surname", dir(result), "To class must not contain surname")
+
+    def test_mapping_excluded_field(self):
+        """Test mapping with excluded fields"""
+        #Arrange
+        from_class = FromTestClass()
+        mapper = ObjectMapper()
+        mapper.create_map(FromTestClass, ToTestClass)
+
+        #Act
+        result = mapper.map(FromTestClass(), ToTestClass, excluded=['date'])
+
+        #Assert
+        print(result)      
+        self.assertTrue(isinstance(result, ToTestClass), "Type must be ToTestClass")
+        self.assertEqual(result.name, from_class.name, "Name mapping must be equal")
+        self.assertEqual(result.date, '', "Date mapping must be equal")
+        self.assertNotIn("surname", result.__dict__, "To class must not contain surname")