def test_dehydrate(dehydrate_spec_mock): # TODO: find out how to undecorate function with mock dehydrate_spec_mock.return_value = list(zip(DUMMY_SPECS, ('Tummy', 10))) dehydrator = Dehydrator() dehydrator.specs = DUMMY_SPECS dehydrator.dehydrate(obj=DUMMY_OBJECT) dehydrate_spec_mock.has_calls( call(DUMMY_OBJECT, DUMMY_SPECS[0]), call(DUMMY_OBJECT, DUMMY_SPECS[1]), )
def test_dehydrate_none_with_fallback_factory(): fallback = lambda obj: {'id': '', 'name': ''} dehydrator = Dehydrator(specs=DUMMY_SPECS, empty=fallback) result1 = dehydrator.dehydrate(None) result2 = dehydrator.dehydrate(None) assert result1 == result2 == {'id': '', 'name': ''} assert result1 is not result2
def test_dehydrate_none_with_fallback(): fallback_object = object() dehydrator = Dehydrator(specs=DUMMY_SPECS, empty=fallback_object) result = dehydrator.dehydrate(None) assert result is fallback_object
def test_dehydrate_none(): dehydrator = Dehydrator(specs=DUMMY_SPECS) result = dehydrator.dehydrate(None) assert result is None