Cleaned up the tests

code/src/packet_capturing.py

@@ -106,8 +106,7 @@ def get_dict():
 	return syn_counts
 
 def dict_clear():
-	global syn_counts
-	syn_counts = defaultdict(int)
+	syn_counts.clear()
 
 def main():
 	print("Starting packet capture...")

code/test/test_packet_capturing.py

 import unittest
 
 from scapy.layers.inet import IP, TCP
-from unittest.mock import patch
+from unittest.mock import patch, MagicMock
 from scapy.layers.l2 import Ether
 
-from src.packet_capturing import packet_handler, get_dict, dict_clear
-
-EXPECTED_EMPTY_DICT = "Expected empty dict"
-EXPECTED_VALUE_DICT = "Expected packet's IP in the dict"
+from src.packet_capturing import packet_handler, get_dict, dict_clear, syn_flood_detection, port_check, ip_spoofing, \
+    syn_fin, null_packet
 
 class TestPacketCapturing(unittest.TestCase):
 
@@ -18,55 +16,44 @@ class TestPacketCapturing(unittest.TestCase):
         # Create a dummy packet
         packet = IP(src = "100.84.6.141", dst = "192.168.1.1") / TCP(dport = 80, sport = 12345, flags = "S")
         # Show detailed information about the package
-        # print(packet.show())
+        print(packet.show())
 
         # Mockito checks if print was called
         with patch('builtins.print') as mock_print:
             packet_handler(packet)
             mock_print.assert_any_call("Packet captured: 100.84.6.141 -> 192.168.1.1")
 
-        self.assertEqual({'100.84.6.141' : 1}, get_dict() , EXPECTED_VALUE_DICT)
-
     def test_ip_spoofing(self):
         # Starts with 169.254, it's a suspicious packet
         packet = IP(src = "169.254.6.141", dst = "192.168.1.1")
 
         with patch('builtins.print') as mock_print:
-            packet_handler(packet)
+            ip_spoofing(0, packet[IP].src)
             mock_print.assert_any_call("Possible IP spoofing using private networks detected.")
 
-        # Has no TCP flags, therefore dict must be empty
-        self.assertEqual({}, get_dict(), EXPECTED_EMPTY_DICT)
-
     def test_syn_fin(self):
         # Has S and F flags, which is also known as SYN-FIN, which is a malicious packet signature
         # Ether addresses are placeholders
         packet = IP(src = "100.84.6.141", dst = "192.168.1.1") / TCP(dport = 80, flags = "SF") / Ether(src="00:11:22:33:44:55", dst="ff:ff:ff:ff:ff:ff")
 
         with patch('builtins.print') as mock_print:
-            packet_handler(packet)
+            syn_fin(packet)
             mock_print.assert_any_call("Malicious packet detected: SYN-FIN combination.")
 
-        self.assertEqual({}, get_dict(), EXPECTED_EMPTY_DICT)
-
     def test_null_packet(self):
         packet = IP(src = "100.84.6.141", dst = "192.168.1.1") / TCP(dport = 80, flags = 0) / Ether(src="00:11:22:33:44:55", dst="ff:ff:ff:ff:ff:ff")
 
         with patch('builtins.print') as mock_print:
-            packet_handler(packet)
+            null_packet(packet)
             mock_print.assert_any_call("Malicious null packet found.")
 
-        self.assertEqual(get_dict(), {}, EXPECTED_EMPTY_DICT)
-
     def test_port_dest_check(self):
         packet = IP(src="100.84.6.141", dst="192.168.1.1") / TCP(dport = 0, sport = 200, flags="F") / Ether(src="00:11:22:33:44:55", dst="ff:ff:ff:ff:ff:ff")
 
         with patch('builtins.print') as mock_print:
-            packet_handler(packet)
+            port_check(packet)
             mock_print.assert_any_call("Illegal packet with source or destination port 0.")
 
-        self.assertEqual(get_dict(), {}, EXPECTED_EMPTY_DICT)
-
     def test_port_src_check(self):
         packet = IP(src="100.84.6.141", dst="192.168.1.1") / TCP(dport=80, sport=0, flags="F") / Ether(src="00:11:22:33:44:55", dst="ff:ff:ff:ff:ff:ff")
 
@@ -74,7 +61,12 @@ class TestPacketCapturing(unittest.TestCase):
             packet_handler(packet)
             mock_print.assert_any_call("Illegal packet with source or destination port 0.")
 
-        self.assertEqual(get_dict(), {}, EXPECTED_EMPTY_DICT)
+    def test_syn_flood_detection(self):
+        packet = IP(src="100.84.6.141", dst="192.168.1.1") / TCP(dport=80, sport=0, flags="S")
+        for i in range (101):
+            syn_flood_detection(packet)
+
+        self.assertEqual({"100.84.6.141": 101}, get_dict(), "Expected the packet's IP with 101 entries")
 
 if __name__ == '__main__':
     unittest.main()