name : test_scheme_registration.py # -*- coding: utf-8 -*- from __future__ import unicode_literals from typing import cast from .. import _url from .common import HyperlinkTestCase from .._url import register_scheme, URL, DecodedURL class TestSchemeRegistration(HyperlinkTestCase): def setUp(self): # type: () -> None self._orig_scheme_port_map = dict(_url.SCHEME_PORT_MAP) self._orig_no_netloc_schemes = set(_url.NO_NETLOC_SCHEMES) def tearDown(self): # type: () -> None _url.SCHEME_PORT_MAP = self._orig_scheme_port_map _url.NO_NETLOC_SCHEMES = self._orig_no_netloc_schemes def test_register_scheme_basic(self): # type: () -> None 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): # type: () -> None 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): # type: () -> None with self.assertRaises(ValueError): register_scheme("badnetlocless", uses_netloc=False, default_port=7) def test_invalid_uses_netloc(self): # type: () -> None with self.assertRaises(ValueError): register_scheme("badnetloc", uses_netloc=cast(bool, None)) with self.assertRaises(ValueError): register_scheme("badnetloc", uses_netloc=cast(bool, object())) def test_register_invalid_uses_netloc(self): # type: () -> None with self.assertRaises(ValueError): register_scheme("lol", uses_netloc=cast(bool, object())) def test_register_invalid_port(self): # type: () -> None with self.assertRaises(ValueError): register_scheme("nope", default_port=cast(bool, object())) def test_register_no_quote_plus_scheme(self): # type: () -> None register_scheme("keepplus", query_plus_is_space=False) plus_is_not_space = DecodedURL.from_text( "keepplus://example.com/?q=a+b" ) plus_is_space = DecodedURL.from_text("https://example.com/?q=a+b") assert plus_is_not_space.get("q") == ["a+b"] assert plus_is_space.get("q") == ["a b"]