Pavel Patsey
87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
import pytest
|
|
|
|
from vpn_manager.cidr4_merge.cidr4_merger import (
|
|
Cidr4MergerError,
|
|
cidr4_to_node,
|
|
find_parent,
|
|
make_cidr4,
|
|
sort_nodes,
|
|
)
|
|
|
|
|
|
def test_true():
|
|
assert True
|
|
|
|
|
|
def test_cidr4_to_node():
|
|
assert cidr4_to_node("4.78.139.0/24") == (72256256, 24)
|
|
assert cidr4_to_node("0.0.0.0/32") == (0, 32)
|
|
|
|
assert cidr4_to_node("23.234.30.0/24") == (401219072, 24)
|
|
assert cidr4_to_node("172.217.0.0/19") == (2899902464, 19)
|
|
assert cidr4_to_node("23.225.141.0/24") == (400657664, 24)
|
|
assert cidr4_to_node("31.13.94.0/23") == (520969728, 23)
|
|
|
|
assert cidr4_to_node("0.0.0.0/2") == (0, 2)
|
|
assert cidr4_to_node("64.0.0.0/2") == (1073741824, 2)
|
|
assert cidr4_to_node("128.0.0.0/2") == (2147483648, 2)
|
|
assert cidr4_to_node("192.0.0.0/2") == (3221225472, 2)
|
|
|
|
|
|
def test_make_cidr4():
|
|
assert make_cidr4(72256256, 24) == "4.78.139.0/24"
|
|
assert make_cidr4(0, 32) == "0.0.0.0/32"
|
|
|
|
assert make_cidr4(401219072, 24) == "23.234.30.0/24"
|
|
assert make_cidr4(2899902464, 19) == "172.217.0.0/19"
|
|
assert make_cidr4(400657664, 24) == "23.225.141.0/24"
|
|
assert make_cidr4(520969728, 23) == "31.13.94.0/23"
|
|
|
|
|
|
def test_sort_nodes():
|
|
assert sort_nodes(
|
|
[
|
|
(401219072, 24),
|
|
(2899902464, 19),
|
|
(400657664, 24),
|
|
(520969728, 23),
|
|
]
|
|
) == [
|
|
(400657664, 24),
|
|
(401219072, 24),
|
|
(520969728, 23),
|
|
(2899902464, 19),
|
|
]
|
|
|
|
|
|
def test_find_parent():
|
|
assert find_parent((0, 2), (1073741824, 2)) == (0, 1)
|
|
assert find_parent((2147483648, 2), (3221225472, 2)) == (2147483648, 1)
|
|
assert find_parent((0, 2), (3221225472, 2)) == (0, 0)
|
|
assert find_parent((1, 32), (6, 32)) == (0, 29)
|
|
|
|
|
|
def test_find_parent__with_exception():
|
|
with pytest.raises(Exception) as exc_info:
|
|
find_parent((0, 32), (0, 29))
|
|
assert (
|
|
str(exc_info.value)
|
|
== "Error! Trying to find common parent of network and subnet! parent_node=(0, 29), a=(0, 32), b=(0, 29)."
|
|
)
|
|
assert exc_info.type is Cidr4MergerError
|
|
with pytest.raises(Exception) as exc_info:
|
|
find_parent((0, 1), (1073741824, 2))
|
|
assert (
|
|
str(exc_info.value)
|
|
== "Error! Trying to find common parent of network and subnet! parent_node=(0, 1), a=(0, 1), b=(1073741824, 2)."
|
|
)
|
|
assert exc_info.type is Cidr4MergerError
|
|
|
|
with pytest.raises(Exception) as exc_info:
|
|
find_parent((0, 0), (3221225472, 2))
|
|
assert (
|
|
str(exc_info.value)
|
|
== "Error! Trying to find common parent of network and subnet! parent_node=(0, 0), a=(0, 0), b=(3221225472, 2)."
|
|
)
|
|
assert exc_info.type is Cidr4MergerError
|