class TestBSFilterValidator(unittest.TestCase): def setUp(self): self.bsfilter_dao = Mock() self.user = User(id=sentinel.user_id) self.funckey = FuncKey(destination=BSFilterDestination(filter_member_id=sentinel.filter_member_id)) self.validator = BSFilterValidator(self.bsfilter_dao) def test_when_func_key_does_not_have_bsfilter_destination_then_validation_passes(self): funckey = FuncKey(destination=CustomDestination(exten='1234')) self.validator.validate(self.user, funckey) def test_when_user_is_not_member_of_a_filter_then_raises_error(self): self.bsfilter_dao.find_all_by_member_id.return_value = [] assert_that(calling(self.validator.validate).with_args(self.user, self.funckey), raises(ResourceError)) def test_when_user_is_member_of_a_filter_then_validation_passes(self): filter_member = FilterMember(id=None, member_id=sentinel.user_id, role='boss') self.bsfilter_dao.find_all_by_member_id.return_value = [filter_member] self.validator.validate(self.user, self.funckey)
def setUp(self): self.bsfilter_dao = Mock() self.user = User(id=sentinel.user_id) self.funckey = FuncKey(destination=BSFilterDestination(filter_member_id=sentinel.filter_member_id)) self.validator = BSFilterValidator(self.bsfilter_dao)