57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
# Copyright 2018 New Vector Ltd
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
from synapse.util.stringutils import parse_and_validate_server_name, parse_server_name
|
|
|
|
from tests import unittest
|
|
|
|
|
|
class ServerNameTestCase(unittest.TestCase):
|
|
def test_parse_server_name(self) -> None:
|
|
test_data = {
|
|
"localhost": ("localhost", None),
|
|
"my-example.com:1234": ("my-example.com", 1234),
|
|
"1.2.3.4": ("1.2.3.4", None),
|
|
"[0abc:1def::1234]": ("[0abc:1def::1234]", None),
|
|
"1.2.3.4:1": ("1.2.3.4", 1),
|
|
"[0abc:1def::1234]:8080": ("[0abc:1def::1234]", 8080),
|
|
":80": ("", 80),
|
|
"": ("", None),
|
|
}
|
|
|
|
for i, o in test_data.items():
|
|
self.assertEqual(parse_server_name(i), o)
|
|
|
|
def test_validate_bad_server_names(self) -> None:
|
|
test_data = [
|
|
"", # empty
|
|
"localhost:http", # non-numeric port
|
|
"1234]", # smells like ipv6 literal but isn't
|
|
"[1234",
|
|
"[1.2.3.4]",
|
|
"underscore_.com",
|
|
"percent%65.com",
|
|
"newline.com\n",
|
|
".empty-label.com",
|
|
"1234:5678:80", # too many colons
|
|
":80",
|
|
]
|
|
for i in test_data:
|
|
try:
|
|
parse_and_validate_server_name(i)
|
|
self.fail(
|
|
"Expected parse_and_validate_server_name('%s') to throw" % (i,)
|
|
)
|
|
except ValueError:
|
|
pass
|