import webbrowser import unittest import os import sys import subprocess from unittest import mock from test import support URL = 'http://www.example.com' CMD_NAME = 'test' class PopenMock(mock.MagicMock): def poll(self): return 0 def wait(self, seconds=None): return 0 class CommandTestMixin: def _test(self, meth, *, args=[URL], kw={}, options, arguments): """Given a web browser instance method name along with arguments and keywords for same (which defaults to the single argument URL), creates a browser instance from the class pointed to by self.browser, calls the indicated instance method with the indicated arguments, and compares the resulting options and arguments passed to Popen by the browser instance against the 'options' and 'args' lists. Options are compared in a position independent fashion, and the arguments are compared in sequence order to whatever is left over after removing the options. """ popen = PopenMock() support.patch(self, subprocess, 'Popen', popen) browser = self.browser_class(name=CMD_NAME) getattr(browser, meth)(*args, **kw) popen_args = subprocess.Popen.call_args[0][0] self.assertEqual(popen_args[0], CMD_NAME) popen_args.pop(0) for option in options: self.assertIn(option, popen_args) popen_args.pop(popen_args.index(option)) self.assertEqual(popen_args, arguments) class GenericBrowserCommandTest(CommandTestMixin, unittest.TestCase): browser_class = webbrowser.GenericBrowser def test_open(self): self._test('open', options=[], arguments=[URL]) class BackgroundBrowserCommandTest(CommandTestMixin, unittest.TestCase): browser_class = webbrowser.BackgroundBrowser def test_open(self): self._test('open', options=[], arguments=[URL]) class ChromeCommandTest(CommandTestMixin, unittest.TestCase): browser_class = webbrowser.Chrome def test_open(self): self._test('open', options=[], arguments=[URL]) def test_open_with_autoraise_false(self): self._test('open', kw=dict(autoraise=False), options=[], arguments=[URL]) def test_open_new(self): self._test('open_new', options=['--new-window'], arguments=[URL]) def test_open_new_tab(self): self._test('open_new_tab', options=[], arguments=[URL]) class MozillaCommandTest(CommandTestMixin, unittest.TestCase): browser_class = webbrowser.Mozilla def test_open(self): self._test('open', options=[], arguments=[URL]) def test_open_with_autoraise_false(self): self._test('open', kw=dict(autoraise=False), options=[], arguments=[URL]) def test_open_new(self): self._test('open_new', options=[], arguments=['-new-window', URL]) def test_open_new_tab(self): self._test('open_new_tab', options=[], arguments=['-new-tab', URL]) class NetscapeCommandTest(CommandTestMixin, unittest.TestCase): browser_class = webbrowser.Netscape def test_open(self): self._test('open', options=['-raise', '-remote'], arguments=['openURL({})'.format(URL)]) def test_open_with_autoraise_false(self): self._test('open', kw=dict(autoraise=False), options=['-noraise', '-remote'], arguments=['openURL({})'.format(URL)]) def test_open_new(self): self._test('open_new', options=['-raise', '-remote'], arguments=['openURL({},new-window)'.format(URL)]) def test_open_new_tab(self): self._test('open_new_tab', options=['-raise', '-remote'], arguments=['openURL({},new-tab)'.format(URL)]) class GaleonCommandTest(CommandTestMixin, unittest.TestCase): browser_class = webbrowser.Galeon def test_open(self): self._test('open', options=['-n'], arguments=[URL]) def test_open_with_autoraise_false(self): self._test('open', kw=dict(autoraise=False), options=['-noraise', '-n'], arguments=[URL]) def test_open_new(self): self._test('open_new', options=['-w'], arguments=[URL]) def test_open_new_tab(self): self._test('open_new_tab', options=['-w'], arguments=[URL]) class OperaCommandTest(CommandTestMixin, unittest.TestCase): browser_class = webbrowser.Opera def test_open(self): self._test('open', options=[], arguments=[URL]) def test_open_with_autoraise_false(self): self._test('open', kw=dict(autoraise=False), options=[], arguments=[URL]) def test_open_new(self): self._test('open_new', options=['--new-window'], arguments=[URL]) def test_open_new_tab(self): self._test('open_new_tab', options=[], arguments=[URL]) class ELinksCommandTest(CommandTestMixin, unittest.TestCase): browser_class = webbrowser.Elinks def test_open(self): self._test('open', options=['-remote'], arguments=['openURL({})'.format(URL)]) def test_open_with_autoraise_false(self): self._test('open', options=['-remote'], arguments=['openURL({})'.format(URL)]) def test_open_new(self): self._test('open_new', options=['-remote'], arguments=['openURL({},new-window)'.format(URL)]) def test_open_new_tab(self): self._test('open_new_tab', options=['-remote'], arguments=['openURL({},new-tab)'.format(URL)]) class BrowserRegistrationTest(unittest.TestCase): def setUp(self): # Ensure we don't alter the real registered browser details self._saved_tryorder = webbrowser._tryorder webbrowser._tryorder = [] self._saved_browsers = webbrowser._browsers webbrowser._browsers = {} def tearDown(self): webbrowser._tryorder = self._saved_tryorder webbrowser._browsers = self._saved_browsers def _check_registration(self, preferred): class ExampleBrowser: pass expected_tryorder = [] expected_browsers = {} self.assertEqual(webbrowser._tryorder, expected_tryorder) self.assertEqual(webbrowser._browsers, expected_browsers) webbrowser.register('Example1', ExampleBrowser) expected_tryorder = ['Example1'] expected_browsers['example1'] = [ExampleBrowser, None] self.assertEqual(webbrowser._tryorder, expected_tryorder) self.assertEqual(webbrowser._browsers, expected_browsers) instance = ExampleBrowser() if preferred is not None: webbrowser.register('example2', ExampleBrowser, instance, preferred=preferred) else: webbrowser.register('example2', ExampleBrowser, instance) if preferred: expected_tryorder = ['example2', 'Example1'] else: expected_tryorder = ['Example1', 'example2'] expected_browsers['example2'] = [ExampleBrowser, instance] self.assertEqual(webbrowser._tryorder, expected_tryorder) self.assertEqual(webbrowser._browsers, expected_browsers) def test_register(self): self._check_registration(preferred=False) def test_register_default(self): self._check_registration(preferred=None) def test_register_preferred(self): self._check_registration(preferred=True) class ImportTest(unittest.TestCase): def test_register(self): webbrowser = support.import_fresh_module('webbrowser') self.assertIsNone(webbrowser._tryorder) self.assertFalse(webbrowser._browsers) class ExampleBrowser: pass webbrowser.register('Example1', ExampleBrowser) self.assertTrue(webbrowser._tryorder) self.assertEqual(webbrowser._tryorder[-1], 'Example1') self.assertTrue(webbrowser._browsers) self.assertIn('example1', webbrowser._browsers) self.assertEqual(webbrowser._browsers['example1'], [ExampleBrowser, None]) def test_get(self): webbrowser = support.import_fresh_module('webbrowser') self.assertIsNone(webbrowser._tryorder) self.assertFalse(webbrowser._browsers) with self.assertRaises(webbrowser.Error): webbrowser.get('fakebrowser') self.assertIsNotNone(webbrowser._tryorder) def test_synthesize(self): webbrowser = support.import_fresh_module('webbrowser') name = os.path.basename(sys.executable).lower() webbrowser.register(name, None, webbrowser.GenericBrowser(name)) webbrowser.get(sys.executable) def test_environment(self): webbrowser = support.import_fresh_module('webbrowser') try: browser = webbrowser.get().name except (webbrowser.Error, AttributeError) as err: self.skipTest(str(err)) with support.EnvironmentVarGuard() as env: env["BROWSER"] = browser webbrowser = support.import_fresh_module('webbrowser') webbrowser.get() def test_environment_preferred(self): webbrowser = support.import_fresh_module('webbrowser') try: webbrowser.get() least_preferred_browser = webbrowser.get(webbrowser._tryorder[-1]).name except (webbrowser.Error, AttributeError, IndexError) as err: self.skipTest(str(err)) with support.EnvironmentVarGuard() as env: env["BROWSER"] = least_preferred_browser webbrowser = support.import_fresh_module('webbrowser') self.assertEqual(webbrowser.get().name, least_preferred_browser) with support.EnvironmentVarGuard() as env: env["BROWSER"] = sys.executable webbrowser = support.import_fresh_module('webbrowser') self.assertEqual(webbrowser.get().name, sys.executable) if __name__=='__main__': unittest.main()