def test_messageroute_post__invalid_paras(self): request = { "ban_id": "asdf", "author_id": 1, "sent_time": self.current_time_string, "called_out": True, } self.assertRaises( InvalidTypeError, MessageRoute().post, self.session, "Bot " + BOT_TOKEN, request, ) request = { "ban_id": 1, "sent_time": 1234, "author_id": 1, "called_out": True } self.assertRaises( InvalidTypeError, MessageRoute().post, self.session, "Bot " + BOT_TOKEN, request, ) request = { "ban_id": 1, "sent_time": "asdf", "author_id": 1, "called_out": True } self.assertRaises( ValidationError, MessageRoute().post, self.session, "Bot " + BOT_TOKEN, request, ) request = { "ban_id": 1, "sent_time": self.current_time_string, "author_id": "asdf", "called_out": True } self.assertRaises( ValidationError, MessageRoute().post, self.session, "Bot " + BOT_TOKEN, request, )
def test_messageroute_post__good_request_callout(self): request = { "ban_id": 1, "author_id": 1, "sent_time": self.current_time_string, "called_out": True, } self.assertNotEqual( self.current_time_string, self.session.query(Ban).filter_by(rowid=1).first().infracted_at, ) self.assertNotEqual( self.current_time_string, self.session.query(Ban).filter_by(rowid=1).first().calledout_at, ) MessageRoute().post(self.session, "Bot " + BOT_TOKEN, request) self.assertEqual( self.current_time_string, self.session.query(Ban).filter_by(rowid=1).first().infracted_at, ) self.assertEqual( self.current_time_string, self.session.query(Ban).filter_by(rowid=1).first().calledout_at, )
def test_messageroute_post__record_setting(self): request = { "ban_id": 1, "author_id": 1, "sent_time": self.current_time_string, } MessageRoute().post(self.session, "Bot " + BOT_TOKEN, request) self.assertEqual( timedelta(days=6).total_seconds(), self.session.query(BanRecord).filter_by( rowid=1).first().record_seconds)
def test_messageroute_post__unauthorized(self): request = { "ban_id": 3, "author_id": 1, "sent_time": self.current_time_string, } self.assertRaises( AuthenticationError, MessageRoute().post, self.session, "Bot " + "asdffdsa", request, )
def test_messageroute_post__ban_not_found(self): request = { "ban_id": 3, "author_id": 1, "sent_time": self.current_time_string, } self.assertRaises( NotFoundError, MessageRoute().post, self.session, "Bot " + BOT_TOKEN, request, )
def test_messageroute_post__new_author(self): request = { "ban_id": 1, "author_id": 1, "sent_time": self.current_time_string, } MessageRoute().post(self.session, "Bot " + BOT_TOKEN, request) self.assertEqual( 1, self.session.query(Author).filter_by( user_id=1).first().infraction_count) self.assertEqual( 1, self.session.query(AuthorInfraction).filter_by( user_id=1, ban_id=1).first().infraction_count)
def test_messageroute_post__existing_author_and_infraction(self): new_author = Author(user_id=1, infraction_count=9) new_authinf = AuthorInfraction(user_id=1, ban_id=1, infraction_count=2) self.session.add(new_author) self.session.add(new_authinf) request = { "ban_id": 1, "author_id": 1, "sent_time": self.current_time_string, } MessageRoute().post(self.session, "Bot " + BOT_TOKEN, request) self.assertEqual( 3, self.session.query(AuthorInfraction).filter_by( user_id=1, ban_id=1).first().infraction_count)