Skip to content

Commit 5a388cf

Browse files
author
Derek Drummond
committed
Added unit test for exlcuded kwarg.
1 parent 7ef3810 commit 5a388cf

File tree

1 file changed

+18
-1
lines changed

1 file changed

+18
-1
lines changed

tests/object_mapper_test.py

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -300,4 +300,21 @@ def __setattr__(self, name, value):
300300
self.assertTrue(isinstance(result, ToCustomDirClass), "Target types must be same")
301301
self.assertEqual(result.name, from_class.name, "Name mapping must be equal")
302302
self.assertEqual(result.date, from_class.date, "Date mapping must be equal")
303-
self.assertNotIn("surname", dir(result), "To class must not contain surname")
303+
self.assertNotIn("surname", dir(result), "To class must not contain surname")
304+
305+
def test_mapping_excluded_field(self):
306+
"""Test mapping with excluded fields"""
307+
#Arrange
308+
from_class = FromTestClass()
309+
mapper = ObjectMapper()
310+
mapper.create_map(FromTestClass, ToTestClass)
311+
312+
#Act
313+
result = mapper.map(FromTestClass(), ToTestClass, excluded=['date'])
314+
315+
#Assert
316+
print(result)
317+
self.assertTrue(isinstance(result, ToTestClass), "Type must be ToTestClass")
318+
self.assertEqual(result.name, from_class.name, "Name mapping must be equal")
319+
self.assertEqual(result.date, '', "Date mapping must be equal")
320+
self.assertNotIn("surname", result.__dict__, "To class must not contain surname")

0 commit comments

Comments
 (0)