60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
import ipaddress
|
|
import unittest
|
|
|
|
from network.net_range import MinerNetworkRange
|
|
from network import MinerNetwork
|
|
|
|
|
|
class NetworkTest(unittest.TestCase):
|
|
def test_net_range(self):
|
|
net_range_str = "192.168.1.29, 192.168.1.40-192.168.1.43, 192.168.1.60"
|
|
net_range_list = [
|
|
"192.168.1.29",
|
|
"192.168.1.40",
|
|
"192.168.1.41",
|
|
"192.168.1.42",
|
|
"192.168.1.43",
|
|
"192.168.1.60",
|
|
]
|
|
|
|
net_1 = list(MinerNetworkRange(net_range_str).hosts())
|
|
net_2 = list(MinerNetworkRange(net_range_list).hosts())
|
|
|
|
correct_net = [
|
|
ipaddress.IPv4Address("192.168.1.29"),
|
|
ipaddress.IPv4Address("192.168.1.40"),
|
|
ipaddress.IPv4Address("192.168.1.41"),
|
|
ipaddress.IPv4Address("192.168.1.42"),
|
|
ipaddress.IPv4Address("192.168.1.43"),
|
|
ipaddress.IPv4Address("192.168.1.60"),
|
|
]
|
|
|
|
self.assertTrue(net_1 == correct_net)
|
|
self.assertTrue(net_2 == correct_net)
|
|
|
|
def test_net(self):
|
|
net_1_str = "192.168.1.0"
|
|
net_1_mask = 29
|
|
|
|
net_1 = list(MinerNetwork(net_1_str, mask=net_1_mask).get_network().hosts())
|
|
|
|
net_2 = list(
|
|
MinerNetwork("192.168.1.1-192.168.1.5, 192.168.1.6").get_network().hosts()
|
|
)
|
|
|
|
correct_net = [
|
|
ipaddress.IPv4Address("192.168.1.1"),
|
|
ipaddress.IPv4Address("192.168.1.2"),
|
|
ipaddress.IPv4Address("192.168.1.3"),
|
|
ipaddress.IPv4Address("192.168.1.4"),
|
|
ipaddress.IPv4Address("192.168.1.5"),
|
|
ipaddress.IPv4Address("192.168.1.6"),
|
|
]
|
|
|
|
self.assertTrue(net_1 == correct_net)
|
|
self.assertTrue(net_2 == correct_net)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|