class TestUserHasNoVoicemail(unittest.TestCase):
def setUp(self):
self.dao = Mock()
self.validator = UserHasNoVoicemail(self.dao)
def test_given_user_has_no_voicemail_then_validation_passes(self):
model = UserVoicemail(user_id=1, voicemail_id=2)
self.dao.find_by_user_id.return_value = None
self.validator.validate(model)
def test_given_user_has_a_voicemail_then_validation_passes(self):
model = UserVoicemail(user_id=1, voicemail_id=2)
self.dao.find_by_user_id.return_value = model
assert_that(
calling(self.validator.validate).with_args(model),
raises(ResourceError))
def setUp(self):
self.dao = Mock()
self.validator = UserHasNoVoicemail(self.dao)
