Mock out logging

DankCity · Dec 3, 2016 · 122c655 · 122c655
1 parent 3d5876a
commit 122c655
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -1,19 +1,21 @@
-from logging import Logger
 from configparser import ConfigParser
 from unittest.mock import patch
 
 from dankbot import cli
 
 
+@patch('dankbot.cli.logging.getLogger')
+@patch('dankbot.cli.RotatingFileHandler')
 @patch('dankbot.cli.DankBot')
-def test_main(dankbot):
+def test_main(dankbot, _, gl_mock):
+    gl_mock.return_value = gl_mock
     dankbot.return_value = dankbot
 
     cli.main()
 
     assert dankbot.called
     assert isinstance(dankbot.call_args[0][0], ConfigParser)
-    assert isinstance(dankbot.call_args[0][1], Logger)
+    assert dankbot.call_args[0][1], gl_mock
     assert dankbot.find_and_post_memes.called