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_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_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_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()
replies = db_helper.reply(journalist, source, 2)
test_reply = Reply.query.first()
test_reply.__repr__()
