def test_get_one_or_else_multiple_results(self, mock): journo_1, _ = db_helper.init_journalist() journo_2, _ = db_helper.init_journalist() with mock.patch('logger') as mock_logger: get_one_or_else(Journalist.query, mock_logger, mock) mock_logger.error.assert_called() # Not specifying very long log line mock.assert_called_with(500)
def test_get_one_or_else_returns_one(self, mock): new_journo, _ = db_helper.init_journalist() query = Journalist.query.filter( Journalist.username == new_journo.username) with mock.patch('logger') as mock_logger: selected_journo = get_one_or_else(query, mock_logger, mock) self.assertEqual(new_journo, selected_journo)
def test_throttle_login(self): journalist, _ = db_helper.init_journalist() for _ in range(Journalist._MAX_LOGIN_ATTEMPTS_PER_PERIOD): Journalist.throttle_login(journalist) with self.assertRaises(LoginThrottledException): Journalist.throttle_login(journalist)
def test_journalist_string_representation(self): test_journalist, _ = db_helper.init_journalist() test_journalist.__repr__()
def test_reply_string_representation(self): journalist, _ = db_helper.init_journalist() source, _ = db_helper.init_source() db_helper.reply(journalist, source, 2) test_reply = Reply.query.first() test_reply.__repr__()
def test_reply_string_representation(self): journalist, _ = db_helper.init_journalist() source, _ = db_helper.init_source() replies = db_helper.reply(journalist, source, 2) test_reply = Reply.query.first() test_reply.__repr__()