3
0

test_DDoSAttack.py 2.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. import unittest.mock as mock
  2. import ID2TLib.TestLibrary as Lib
  3. import Test.ID2TAttackTest as Test
  4. sha_basic_ddos = '33dfb7a6f4b32be3409cf088e5a2e179e98456637d192e7541a6f16c286c2bc2'
  5. sha_num_attackers_ddos = '64a76befad280dac48c39ba9e3258a7768c13aeea3203044900fd8ac49e1e6ea'
  6. sha_dest_mac_length_zero_ddos = '75c6aaf84f3ede7b740233f2e6e7f78a4d1887b1e3b5137942c3e2095abad5f6'
  7. sha_mss_none_ddos = '33dfb7a6f4b32be3409cf088e5a2e179e98456637d192e7541a6f16c286c2bc2'
  8. sha_one_attacker_ddos = '4e9dc3c4ff09f9fa52fb2f6cd6260dcfbe338dfdbf4df100c1170da913547dcb'
  9. sha_ip_range_ddos = '87639c6ecb652afdc393ed810efb0d6fee1032629d4f00d774589fba90e89204'
  10. sha_port_range_ddos = 'c141c32cb78755e20477518c1165eb4564af88b1a95025925f6c90039de35359'
  11. class UnitTestDDoS(Test.ID2TAttackTest):
  12. @mock.patch('ID2TLib.Utility.get_attacker_config', side_effect=Lib.get_attacker_config)
  13. def test_ddos_basic(self, mock_get_attacker_config):
  14. self.checksum_test([['DDoSAttack']],
  15. sha_basic_ddos)
  16. @mock.patch('ID2TLib.Utility.get_attacker_config', side_effect=Lib.get_attacker_config)
  17. def test_ddos_num_attackers(self, mock_get_attacker_config):
  18. self.checksum_test([['DDoSAttack', 'attackers.count=5']],
  19. sha_num_attackers_ddos)
  20. @mock.patch('ID2TLib.Utility.get_attacker_config', side_effect=Lib.get_attacker_config)
  21. @mock.patch('Core.Statistics.Statistics.get_mac_address', return_value=[])
  22. def test_ddos_dest_mac_length_zero(self, mock_dest_mac, mock_get_attacker_config):
  23. self.checksum_test([['DDoSAttack']], sha_dest_mac_length_zero_ddos)
  24. @mock.patch('ID2TLib.Utility.get_attacker_config', side_effect=Lib.get_attacker_config)
  25. @mock.patch('Core.Statistics.Statistics.get_most_used_mss', return_value=None)
  26. def test_ddos_mss_none(self, mock_mss, mock_get_attacker_config):
  27. self.checksum_test([['DDoSAttack']], sha_mss_none_ddos)
  28. @mock.patch('ID2TLib.Utility.get_attacker_config', side_effect=Lib.get_attacker_config)
  29. def test_ddos_one_attacker(self, mock_get_attacker_config):
  30. self.checksum_test([['DDoSAttack', 'ip.src=1.1.1.1']],
  31. sha_one_attacker_ddos)
  32. @mock.patch('ID2TLib.Utility.get_attacker_config', side_effect=Lib.get_attacker_config)
  33. def test_ddos_ip_range(self, mock_get_attacker_config):
  34. self.checksum_test([['DDoSAttack', 'ip.src=1.1.1.1-1.1.1.10']],
  35. sha_ip_range_ddos)
  36. @mock.patch('ID2TLib.Utility.get_attacker_config', side_effect=Lib.get_attacker_config)
  37. def test_ddos_port_range(self, mock_get_attacker_config):
  38. self.checksum_test([['DDoSAttack', 'attackers.count=5', 'port.src=1000-2000']],
  39. sha_port_range_ddos)
  40. def test_ddos_order(self):
  41. self.order_test([['DDoSAttack', 'attackers.count=5']])