def test_remote_ip_invalis_ips_count():
ips = [ip_address('10.10.10.10'), ip_address('20.20.20.20')]
trusted = parse_trusted_list([['40.40.40.40'], ['20.20.20.20']])
with pytest.raises(IncorrectIPCount) as ctx:
remote_ip(trusted, ips)
assert ctx.value.expected == 3
assert ctx.value.actual == [
IPv4Address('10.10.10.10'),
IPv4Address('20.20.20.20')
]
def test_remote_ip_not_trusted_ip():
ips = [
ip_address('10.10.10.10'),
ip_address('20.20.20.20'),
ip_address('30.30.30.30')
]
trusted = parse_trusted_list([['40.40.40.40'], ['20.20.20.20']])
with pytest.raises(UntrustedIP) as ctx:
remote_ip(trusted, ips)
assert ctx.value.trusted == [ip_address('40.40.40.40')]
assert ctx.value.ip == ip_address('10.10.10.10')
def test_remote_ip_ok():
ips = [
ip_address('10.10.10.10'),
ip_address('20.20.20.20'),
ip_address('30.30.30.30')
]
trusted = parse_trusted_list([['10.10.0.0/16'], ['20.20.20.20']])
assert ips[-1] == remote_ip(trusted, ips)
def test_remote_with_ellipsis():
ips = [
ip_address('10.10.10.10'),
ip_address('20.20.20.20'),
ip_address('30.30.30.30')
]
trusted = parse_trusted_list([['10.10.0.0/16'], ...])
assert ips[-2] == remote_ip(trusted, ips)
def test_remote_ip_no_trusted():
ip = ip_address('10.10.10.10')
assert ip == remote_ip([], [ip])
