def test_build_value(resolve_target_mock): handler = SimpleHandler(dehydrator=DUMMY_OBJECT, spec=DUMMY_SPEC) handler.build_value(obj=DUMMY_OBJECT) == DUMMY_OBJECT resolve_target_mock.assert_called_once_with( obj=DUMMY_OBJECT, target=DUMMY_TARGET)
def test_build_value(resolve_target_mock): handler = SimpleHandler(dehydrator=DUMMY_OBJECT, spec=DUMMY_SPEC) handler.build_value(obj=DUMMY_OBJECT) == DUMMY_OBJECT resolve_target_mock.assert_called_once_with(obj=DUMMY_OBJECT, target=DUMMY_TARGET)
def test_post_hook(): dehydrator = stub(post_handle_value=lambda obj: obj.upper()) handler = SimpleHandler(dehydrator=dehydrator, spec=DUMMY_SPEC) handler.resolve_target = Mock(return_value='iron_man') assert handler.build_value(obj=DUMMY_OBJECT) == 'IRON_MAN'
def test_post_hook(): dehydrator = stub(post_handle_value=lambda obj: obj.upper()) handler = SimpleHandler(dehydrator=dehydrator, spec=DUMMY_SPEC) handler.resolve_target = Mock(return_value='iron_man') assert handler.build_value(obj=DUMMY_OBJECT) == 'IRON_MAN'