From 99adad873e387637d6de582071fb635e43e8c0d6 Mon Sep 17 00:00:00 2001 From: Jonathan Golder Date: Tue, 18 Sep 2018 12:34:14 +0200 Subject: [PATCH] missingnotice_test: Test article query Issue #64 (https://git.golderweb.de/wiki/jogobot-red/issues/64) --- tests/missingnotice_tests.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/tests/missingnotice_tests.py b/tests/missingnotice_tests.py index 7cbe286..2e3c1b4 100644 --- a/tests/missingnotice_tests.py +++ b/tests/missingnotice_tests.py @@ -37,7 +37,22 @@ class TestMissingNoticeBot(unittest.TestCase): """ 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__':