Add opts tests

Author: MrMino

tests/test_opts.py

@@ -0,0 +1,51 @@
+# Copyright (c) 2012-2016 Marc Abramowitz and ipdb development team
+#
+# This file is part of ipdb.
+# Redistributable under the revised BSD license
+# https://opensource.org/licenses/BSD-3-Clause
+
+import unittest
+from unittest.mock import patch
+from getopt import GetoptError
+from ipdb.__main__ import main
+
+
+@patch('ipdb.__main__.debugger_cls')
+class OptsTest(unittest.TestCase):
+    def set_argv(self, *argv):
+        argv_patch = patch('ipdb.__main__.sys.argv', argv)
+        argv_patch.start()
+        self.addCleanup(argv_patch.stop)
+
+    @patch('ipdb.__main__.sys.version_info', (3, 7))
+    def test_debug_module_script(self, debugger_cls):
+        module_name = 'my_buggy_module'
+        self.set_argv('ipdb', '-m', module_name)
+
+        main()
+
+        debugger = debugger_cls.return_value
+        debugger._runmodule.assert_called_once_with(module_name)
+
+    @patch('ipdb.__main__.os.path.exists')
+    def test_debug_script(self, exists, debugger_cls):
+        script_name = 'my_buggy_script'
+        self.set_argv('ipdb', script_name)
+
+        main()
+
+        debugger = debugger_cls.return_value
+        debugger._runscript.assert_called_once_with(script_name)
+
+    def test_option_m_fallback_on_py36(self, debugger_cls):
+        self.set_argv('ipdb', '-m', 'my.module')
+        with patch('ipdb.__main__.sys.version_info', (3, 6)):
+            with self.assertRaises(GetoptError):
+                main()
+
+        with patch('ipdb.__main__.sys.version_info', (3, 7)):
+            self.set_argv('ipdb', '-m', 'my.module')
+            try:
+                main()
+            except GetoptError:
+                self.fail("GetoptError raised unexpectedly.")