49 lines
1 KiB
Python
49 lines
1 KiB
Python
|
import unittest
|
||
|
|
|
||
|
|
from .exceptions import InvalidURI
|
||
|
|
from .uri import *
|
||
|
|
|
||
|
|
|
||
|
|
VALID_URIS = [
|
||
|
|
(
|
||
|
|
'ws://localhost/',
|
||
|
|
(False, 'localhost', 80, '/', None),
|
||
|
|
),
|
||
|
|
(
|
||
|
|
'wss://localhost/',
|
||
|
|
(True, 'localhost', 443, '/', None),
|
||
|
|
),
|
||
|
|
(
|
||
|
|
'ws://localhost/path?query',
|
||
|
|
(False, 'localhost', 80, '/path?query', None),
|
||
|
|
),
|
||
|
|
(
|
||
|
|
'WS://LOCALHOST/PATH?QUERY',
|
||
|
|
(False, 'localhost', 80, '/PATH?QUERY', None),
|
||
|
|
),
|
||
|
|
(
|
||
|
|
'ws://user:pass@localhost/',
|
||
|
|
(False, 'localhost', 80, '/', ('user', 'pass')),
|
||
|
|
),
|
||
|
|
]
|
||
|
|
|
||
|
|
INVALID_URIS = [
|
||
|
|
'http://localhost/',
|
||
|
|
'https://localhost/',
|
||
|
|
'ws://localhost/path#fragment',
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
class URITests(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_success(self):
|
||
|
|
for uri, parsed in VALID_URIS:
|
||
|
|
with self.subTest(uri=uri):
|
||
|
|
self.assertEqual(parse_uri(uri), parsed)
|
||
|
|
|
||
|
|
def test_error(self):
|
||
|
|
for uri in INVALID_URIS:
|
||
|
|
with self.subTest(uri=uri):
|
||
|
|
with self.assertRaises(InvalidURI):
|
||
|
|
parse_uri(uri)
|