added more eternal_blue tests

code/Test/test_EternalBlue.py

@@ -1,11 +1,24 @@
 import Test.ID2TAttackTest as Test
+import ID2TLib.TestLibrary as Lib
 
 sha_default = 'c707492a0493efcf46a569c91fe77685286402ddfdff3c79e64157b3324dc9f6'
+sha_ips_not_in_pcap = '55d74bc906edc6b81a04a38539a0521228ee09146ff507cc19f6142a386bc2eb'
+sha_multiple_params = '118745f3588a862b7a3f9e5e66e568742db58277084f4783cfc3b41cff8350d3'
 
 # TODO: improve coverage
 
 
 class UnitTestEternalBlue(Test.ID2TAttackTest):
 
-    def test_eternal_blue_default(self):
+    def test_eternalblue_default(self):
         self.checksum_test([['EternalBlueExploit']], sha_default)
+
+    def test_eternalblue_ips_not_in_pcap(self):
+        self.checksum_test([['EternalBlueExploit', 'ip.src=1.1.1.1', 'ip.dst=2.2.2.2']], sha_ips_not_in_pcap)
+
+    def test_eternalblue_multiple_params(self):
+        ip_src = 'ip.src='+Lib.test_pcap_ips[0]
+        ip_dst = 'ip.dst='+Lib.test_pcap_ips[1]
+        self.checksum_test([['EternalBlueExploit', ip_src, ip_dst, 'mac.src=00:0C:21:1C:60:61',
+                             'mac.dst=04:0C:32:2C:63:62', 'port.src=1337', 'port.dst=42']], sha_multiple_params)
+

code/Test/test_Utility.py

@@ -27,7 +27,6 @@ class TestUtility(unittest.TestCase):
         cipps = [(5, 1), (10, 2), (15, 3)]
         self.assertEqual(Utility.get_interval_pps(cipps, 30), 3)
 
-    # Errors if empty list and result bad if only one list
     def test_get_nth_random_element_equal_no(self):
         letters = ["A", "B", "C"]
         numbers = [1, 2, 3]