missingnotice_test: Test article query

Issue #64 (#64)
This commit is contained in:
2018-09-18 12:34:14 +02:00
parent dbcc2717d7
commit 99adad873e

View File

@@ -37,7 +37,22 @@ class TestMissingNoticeBot(unittest.TestCase):
""" """
Test class MissingNoticeBot Test class MissingNoticeBot
""" """
pass
@mock.patch( 'sqlalchemy.engine.Engine.execute',
return_value=( { "page_title": b"a", },
{ "page_title": b"b", },
{ "page_title": b"c", },
{ "page_title": b"d", }, ) )
def test_get_categorized_articles(self, execute_mock):
"""
Test method get_categorized_articles()
"""
self.assertFalse(execute_mock.called)
result = MissingNoticeBot.get_categorized_articles()
self.assertTrue(execute_mock.called)
self.assertEqual(result, ["a", "b", "c", "d"] )
if __name__ == '__main__': if __name__ == '__main__':