Server : nginx/1.18.0 System : Linux localhost 6.14.3-x86_64-linode168 #1 SMP PREEMPT_DYNAMIC Mon Apr 21 19:47:55 EDT 2025 x86_64 User : www-data ( 33) PHP Version : 8.0.16 Disable Function : pcntl_alarm,pcntl_fork,pcntl_waitpid,pcntl_wait,pcntl_wifexited,pcntl_wifstopped,pcntl_wifsignaled,pcntl_wifcontinued,pcntl_wexitstatus,pcntl_wtermsig,pcntl_wstopsig,pcntl_signal,pcntl_signal_get_handler,pcntl_signal_dispatch,pcntl_get_last_error,pcntl_strerror,pcntl_sigprocmask,pcntl_sigwaitinfo,pcntl_sigtimedwait,pcntl_exec,pcntl_getpriority,pcntl_setpriority,pcntl_async_signals,pcntl_unshare, Directory :  /usr/lib/python3/dist-packages/hyperlink/test/

Upload File : current_dir [ Writeable ] document_root [ Writeable ]

# -*- coding: utf-8 -*-
from __future__ import unicode_literals


from .. import _url
from .common import HyperlinkTestCase
from .._url import register_scheme, URL


class TestSchemeRegistration(HyperlinkTestCase):

    def setUp(self):
        self._orig_scheme_port_map = dict(_url.SCHEME_PORT_MAP)
        self._orig_no_netloc_schemes = set(_url.NO_NETLOC_SCHEMES)

    def tearDown(self):
        _url.SCHEME_PORT_MAP = self._orig_scheme_port_map
        _url.NO_NETLOC_SCHEMES = self._orig_no_netloc_schemes

    def test_register_scheme_basic(self):
        register_scheme('deltron', uses_netloc=True, default_port=3030)

        u1 = URL.from_text('deltron://example.com')
        assert u1.scheme == 'deltron'
        assert u1.port == 3030
        assert u1.uses_netloc is True

        # test netloc works even when the original gives no indication
        u2 = URL.from_text('deltron:')
        u2 = u2.replace(host='example.com')
        assert u2.to_text() == 'deltron://example.com'

        # test default port means no emission
        u3 = URL.from_text('deltron://example.com:3030')
        assert u3.to_text() == 'deltron://example.com'

        register_scheme('nonetron', default_port=3031)
        u4 = URL(scheme='nonetron')
        u4 = u4.replace(host='example.com')
        assert u4.to_text() == 'nonetron://example.com'

    def test_register_no_netloc_scheme(self):
        register_scheme('noloctron', uses_netloc=False)
        u4 = URL(scheme='noloctron')
        u4 = u4.replace(path=("example", "path"))
        assert u4.to_text() == 'noloctron:example/path'

    def test_register_no_netloc_with_port(self):
        with self.assertRaises(ValueError):
            register_scheme('badnetlocless', uses_netloc=False, default_port=7)

    def test_invalid_uses_netloc(self):
        with self.assertRaises(ValueError):
            register_scheme('badnetloc', uses_netloc=None)
        with self.assertRaises(ValueError):
            register_scheme('badnetloc', uses_netloc=object())

    def test_register_invalid_uses_netloc(self):
        with self.assertRaises(ValueError):
            register_scheme('lol', uses_netloc=lambda: 'nope')

    def test_register_invalid_port(self):
        with self.assertRaises(ValueError):
            register_scheme('nope', default_port=lambda: 'lol')