mirror of
https://github.com/peeringdb/peeringdb.git
synced 2024-05-11 05:55:09 +00:00
5eb925e319
* fix next redirect when using U2F 2FA auth (#1191) * Added self identifier to API * fix migrations hierarchy after merging in previous support branch * campus object Co-authored-by: Stefan Pratter <stefan@20c.com> * fix out of bound error message add city / country to campus view * fix tests * relock poetry * linting * linting * fix docs regen * regen docs * linting * refactor self entity view to support carrier and campus object types and also make it easier to support additional object types in the future * remove debug message --------- Co-authored-by: Gajanan Patil <dipaksavaliya.python@gmail.com>
126 lines
3.8 KiB
Python
126 lines
3.8 KiB
Python
import ipaddress
|
|
|
|
import pytest
|
|
import pytest_filedata
|
|
|
|
from peeringdb_server.inet import RdapNotFoundError, renumber_ipaddress
|
|
|
|
|
|
def test_rdap_asn_lookup(rdap):
|
|
asn = rdap.get_asn(63311)
|
|
assert asn.data
|
|
assert asn.name
|
|
assert asn.emails
|
|
assert asn.org_name
|
|
assert asn.org_address
|
|
|
|
|
|
def test_rdap_asn_lookup_not_found(rdap):
|
|
with pytest.raises(RdapNotFoundError):
|
|
rdap.get_asn(65535)
|
|
|
|
|
|
def test_rdap_asn_lookup_not_found(rdap):
|
|
with pytest.raises(RdapNotFoundError):
|
|
rdap.get_asn(9999999)
|
|
|
|
|
|
def test_mocker(rdap):
|
|
with pytest_filedata.RequestsData("rdap"):
|
|
asn = rdap.get_asn(63311)
|
|
|
|
|
|
@pytest_filedata.RequestsData("rdap")
|
|
def test_arin0(rdap):
|
|
asn = rdap.get_asn(63311)
|
|
assert asn.emails == ["neteng@20c.com"]
|
|
|
|
|
|
def test_recurse_contacts(rdap):
|
|
asn = rdap.get_asn(3333)
|
|
assert rdap == asn._rdapc
|
|
assert len(asn.emails) > 1
|
|
assert len(rdap.history) > 1
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"ip,old_prefix,new_prefix,valid",
|
|
[
|
|
# Test successes
|
|
("206.41.110.48", "206.41.110.0/24", "206.41.111.0/24", "206.41.111.48"),
|
|
("206.41.110.48", "206.41.110.0/25", "206.41.111.0/24", "206.41.111.48"),
|
|
("206.41.110.48", "206.41.110.0/25", "206.41.111.0/25", "206.41.111.48"),
|
|
("206.41.116.101", "206.41.116.0/23", "206.42.116.0/23", "206.42.116.101"),
|
|
("206.41.116.101", "206.41.116.0/23", "206.42.116.0/24", "206.42.116.101"),
|
|
("206.41.116.101", "206.41.116.0/23", "206.42.116.0/25", "206.42.116.101"),
|
|
(
|
|
"2001:504:41:110::20",
|
|
"2001:504:41:110::/64",
|
|
"2001:504:41:111::/64",
|
|
"2001:504:41:111::20",
|
|
),
|
|
(
|
|
"2001:504:41:110::20",
|
|
"2001:504:41:110::/64",
|
|
"2001:504:42:110::/60",
|
|
"2001:504:42:110::20",
|
|
),
|
|
(
|
|
"2001:504:41:110::20",
|
|
"2001:504:41:110::/64",
|
|
"2001:504:42::/48",
|
|
"2001:504:42:110::20",
|
|
),
|
|
# Test failures
|
|
("2001:504:41:110::20", "206.41.110.0/24", "206.41.111.0/24", False),
|
|
("2001:504:41:110::20", "2001:504:41:110::/64", "206.41.111.0/24", False),
|
|
("206.41.110.48", "206.41.110.0/25", "206.41.111.128/25", False),
|
|
("206.41.116.101", "206.41.116.0/23", "206.41.110.0/23", False),
|
|
],
|
|
)
|
|
def test_renumber_ipaddress(ip, old_prefix, new_prefix, valid):
|
|
ip = ipaddress.ip_address(ip)
|
|
old_prefix = ipaddress.ip_network(old_prefix)
|
|
new_prefix = ipaddress.ip_network(new_prefix)
|
|
|
|
if valid:
|
|
renumbered = renumber_ipaddress(ip, old_prefix, new_prefix)
|
|
assert renumbered.compressed == valid
|
|
else:
|
|
with pytest.raises(ValueError):
|
|
renumber_ipaddress(ip, old_prefix, new_prefix)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"input_str,compressed",
|
|
[
|
|
("2001:0db8::0001", "2001:db8::1"),
|
|
("2001:db8:0:0:0:0:2:1", "2001:db8::2:1"),
|
|
("2001:db8::0:1", "2001:db8::1"),
|
|
("2001:7f8:f2:e1::af21:3376:1", "2001:7f8:f2:e1:0:af21:3376:1"),
|
|
("2001:db8::1:1:1:1:1", "2001:db8:0:1:1:1:1:1"),
|
|
("2001:db8:0:0:1:0:0:1", "2001:db8::1:0:0:1"),
|
|
("2001:7F8:F2:E1:0:AF21:3376:1", "2001:7f8:f2:e1:0:af21:3376:1"),
|
|
],
|
|
ids=[
|
|
"4.1 handling leading zeros",
|
|
"4.2.1 shorten as much as possible (1)",
|
|
"4.2.1 shorten as much as possible (2)",
|
|
"4.2.2 handling one 16-bit 0 field (1)",
|
|
"4.2.2 handling one 16-bit 0 field (2)",
|
|
"4.2.3 choice in placement of ::",
|
|
"4.3 lowercase",
|
|
],
|
|
)
|
|
def test_ipaddress6_compression(input_str, compressed):
|
|
"""
|
|
Testing if the ipaddress string formatting
|
|
is compliant with RFC 5952
|
|
https://tools.ietf.org/html/rfc5952#section-4
|
|
|
|
Ids of parameters denote which rule in the spec
|
|
the test is demonstrating
|
|
"""
|
|
ipv6 = ipaddress.ip_address(input_str)
|
|
assert str(ipv6) == compressed
|