| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336 |
- import webbrowser
- import unittest
- import os
- import sys
- import subprocess
- from unittest import mock
- from test import support
- from test.support import import_helper
- from test.support import os_helper
- if not support.has_subprocess_support:
- raise unittest.SkipTest("test webserver requires subprocess")
- 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 = import_helper.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 = import_helper.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 = import_helper.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 = import_helper.import_fresh_module('webbrowser')
- try:
- browser = webbrowser.get().name
- except webbrowser.Error as err:
- self.skipTest(str(err))
- with os_helper.EnvironmentVarGuard() as env:
- env["BROWSER"] = browser
- webbrowser = import_helper.import_fresh_module('webbrowser')
- webbrowser.get()
- def test_environment_preferred(self):
- webbrowser = import_helper.import_fresh_module('webbrowser')
- try:
- webbrowser.get()
- least_preferred_browser = webbrowser.get(webbrowser._tryorder[-1]).name
- except (webbrowser.Error, IndexError) as err:
- self.skipTest(str(err))
- with os_helper.EnvironmentVarGuard() as env:
- env["BROWSER"] = least_preferred_browser
- webbrowser = import_helper.import_fresh_module('webbrowser')
- self.assertEqual(webbrowser.get().name, least_preferred_browser)
- with os_helper.EnvironmentVarGuard() as env:
- env["BROWSER"] = sys.executable
- webbrowser = import_helper.import_fresh_module('webbrowser')
- self.assertEqual(webbrowser.get().name, sys.executable)
- if __name__=='__main__':
- unittest.main()
|