slightly improved network functionality and added tests for network

2022-05-19 11:55:38 -06:00
parent 40f14876cc
commit ec5563f2f0
3 changed files with 100 additions and 17 deletions
--- a/tests/network_tests/init.py
+++ b/tests/network_tests/init.py
@@ -0,0 +1,70 @@
+import ipaddress
+
+from network.net_range import MinerNetworkRange
+from network import MinerNetwork
+
+
+def test_network():
+    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",
+    ]
+    miner_net_1 = "192.168.1.0"
+    miner_net_1_mask = 29
+
+    #######################
+    # Network Range Tests #
+    #######################
+
+    net_range_1 = list(MinerNetworkRange(net_range_str).hosts())
+
+    net_range_2 = list(MinerNetworkRange(net_range_list).hosts())
+
+    correct_net_range = [
+        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"),
+    ]
+
+    assert net_range_1 == correct_net_range
+    assert net_range_2 == correct_net_range
+
+    print("Network Range test succeeded.")
+
+    #######################
+    # Miner Network Tests #
+    #######################
+
+    miner_net = list(
+        MinerNetwork(miner_net_1, mask=miner_net_1_mask).get_network().hosts()
+    )
+
+    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__":
+    test_network()