CAT_TOM/pyasic
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updated network test to use unittest

Browse Source
This commit is contained in:
UpstreamData
2022-05-25 11:49:28 -06:00
parent 9eb2259aae
commit 6fbd9faffd
2 changed files with 45 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/__init__.py
View File

@@ -1,9 +1,7 @@
from tests.network_tests import test_network import unittest
from tests.network_tests import NetworkTest
def test_all():
test_network()
if __name__ == "__main__": if __name__ == "__main__":
test_all() unittest.main()

95
tests/network_tests/__init__.py
View File

@@ -1,70 +1,59 @@
import ipaddress import ipaddress
import unittest
from network.net_range import MinerNetworkRange from network.net_range import MinerNetworkRange
from network import MinerNetwork from network import MinerNetwork
def test_network(): class NetworkTest(unittest.TestCase):
net_range_str = "192.168.1.29, 192.168.1.40-192.168.1.43, 192.168.1.60" def test_net_range(self):
net_range_list = [ net_range_str = "192.168.1.29, 192.168.1.40-192.168.1.43, 192.168.1.60"
"192.168.1.29", net_range_list = [
"192.168.1.40", "192.168.1.29",
"192.168.1.41", "192.168.1.40",
"192.168.1.42", "192.168.1.41",
"192.168.1.43", "192.168.1.42",
"192.168.1.60", "192.168.1.43",
] "192.168.1.60",
miner_net_1 = "192.168.1.0" ]
miner_net_1_mask = 29
####################### net_1 = list(MinerNetworkRange(net_range_str).hosts())
# Network Range Tests # net_2 = list(MinerNetworkRange(net_range_list).hosts())
#######################
net_range_1 = list(MinerNetworkRange(net_range_str).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"),
]
net_range_2 = list(MinerNetworkRange(net_range_list).hosts()) self.assertTrue(net_1 == correct_net)
self.assertTrue(net_2 == correct_net)
correct_net_range = [ def test_net(self):
ipaddress.IPv4Address("192.168.1.29"), net_1_str = "192.168.1.0"
ipaddress.IPv4Address("192.168.1.40"), net_1_mask = 29
ipaddress.IPv4Address("192.168.1.41"),
ipaddress.IPv4Address("192.168.1.42"),
ipaddress.IPv4Address("192.168.1.43"),
ipaddress.IPv4Address("192.168.1.60"),
]
assert net_range_1 == correct_net_range net_1 = list(MinerNetwork(net_1_str, mask=net_1_mask).get_network().hosts())
assert net_range_2 == correct_net_range
print("Network Range test succeeded.") net_2 = list(
MinerNetwork("192.168.1.1-192.168.1.5, 192.168.1.6").get_network().hosts()
)
####################### correct_net = [
# Miner Network Tests # 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"),
]
miner_net = list( self.assertTrue(net_1 == correct_net)
MinerNetwork(miner_net_1, mask=miner_net_1_mask).get_network().hosts() self.assertTrue(net_2 == correct_net)
)
miner_net_str = 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"),
]
assert miner_net == correct_net
assert miner_net_str == correct_net
print("Miner Network test succeeded.")
if __name__ == "__main__": if __name__ == "__main__":
test_network() unittest.main()