72 lines
2.5 KiB
Python
72 lines
2.5 KiB
Python
# -*- 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
|
|
|
|
|
|
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()))
|