from unittest import TestCase
from ID2TLib.Utility import *
from Test.Lib import test_resource_dir
class TestUtility(TestCase):
def test_update_timestamp_no_delay(self):
self.assertTrue(100+10/5 >= update_timestamp(100, 5) >= 100+1/5)
def test_update_timestamp_with_delay(self):
self.assertTrue(100+1/5+10*100 >= update_timestamp(100, 5, 10) >= 100+1/5+10)
def test_update_timestamp_comparison(self):
self.assertTrue(update_timestamp(100, 5) <= update_timestamp(100, 5, 10))
def test_get_interval_pps_below_max(self):
cipps = [(5, 1), (10, 2), (15, 3)]
self.assertEqual(get_interval_pps(cipps, 3), 1)
self.assertEqual(get_interval_pps(cipps, 7), 2)
self.assertEqual(get_interval_pps(cipps, 12), 3)
def test_get_interval_pps_above_max(self):
cipps = [(5, 1), (10, 2), (15, 3)]
self.assertEqual(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]
results = [("A", 1), ("B", 2), ("C", 3)]
self.assertIn(get_nth_random_element(letters, numbers), results)
def test_get_nth_random_element_unequal_no(self):
letters = ["A", "B", "C"]
numbers = [1, 2]
results = [("A", 1), ("B", 2)]
self.assertIn(get_nth_random_element(letters, numbers), results)
#def test_get_nth_random_element_single_list(self):
#letters = ["A", "B", "C"]
#self.assertIn(get_nth_random_element(letters), letters)
def test_index_increment_not_max(self):
self.assertEqual(index_increment(5, 10), 6)
def test_index_increment_max(self):
self.assertEqual(index_increment(10, 10), 0)
# Correct?
def test_index_increment_max2(self):
self.assertEqual(index_increment(9, 10), 0)
def test_get_rnd_os(self):
self.assertIn(get_rnd_os(), platforms)
def test_check_platform_valid(self):
check_platform("linux")
def test_check_platform_invalid(self):
with self.assertRaises(SystemExit):
check_platform("abc")
def test_get_ip_range_forwards(self):
start = "192.168.178.254"
end = "192.168.179.1"
result = ["192.168.178.254", "192.168.178.255", "192.168.179.0", "192.168.179.1"]
self.assertEqual(get_ip_range(start, end), result)
def test_get_ip_range_backwards(self):
end = "192.168.178.254"
start = "192.168.179.1"
result = ["192.168.179.1", "192.168.179.0", "192.168.178.255", "192.168.178.254"]
self.assertEqual(get_ip_range(start, end), result)
def test_get_ip_range_equal(self):
end = "192.168.178.254"
start = "192.168.178.254"
result = ["192.168.178.254"]
self.assertEqual(get_ip_range(start, end), result)
def test_generate_source_port_from_platform_invalid(self):
with self.assertRaises(SystemExit):
generate_source_port_from_platform("abc")
def test_generate_source_port_from_platform_oldwin_firstport(self):
self.assertTrue(1024 <= generate_source_port_from_platform("winxp") <= 5000)
def test_generate_source_port_from_platform_oldwin_nextport(self):
self.assertEqual(generate_source_port_from_platform("winxp", 2000), 2001)
def test_generate_source_port_from_platform_oldwin_maxport(self):
self.assertTrue(1024 <= generate_source_port_from_platform("winxp", 5000) <= 5000)
def test_generate_source_port_from_platform_linux(self):
self.assertTrue(32768 <= generate_source_port_from_platform("linux") <= 61000)
def test_generate_source_port_from_platform_newwinmac_firstport(self):
self.assertTrue(49152 <= generate_source_port_from_platform("win7") <= 65535)
def test_generate_source_port_from_platform_newwinmac_nextport(self):
self.assertEqual(generate_source_port_from_platform("win7", 50000), 50001)
def test_generate_source_port_from_platform_newwinmac_maxport(self):
self.assertTrue(49152 <= generate_source_port_from_platform("win7", 65535) <= 65535)
# Test get_filetime_format????
def test_get_rnd_boot_time_invalid(self):
with self.assertRaises(SystemExit):
get_rnd_boot_time(10, "abc")
def test_get_rnd_boot_time_linux(self):
self.assertTrue(get_rnd_boot_time(100, "linux") < 100)
def test_get_rnd_boot_time_macos(self):
self.assertTrue(get_rnd_boot_time(100, "macos") < 100)
def test_get_rnd_boot_time_win(self):
self.assertTrue(get_rnd_boot_time(100, "win7") < 100)
def test_get_rnd_x86_nop_len(self):
result = get_rnd_x86_nop(1000)
self.assertEqual(len(result), 1000)
def test_get_rnd_x86_nop_with_sideeffects(self):
result = get_rnd_x86_nop(1000, False)
correct = True
for byte in result:
if byte.to_bytes(1, "little") not in x86_nops and byte.to_bytes(1, "little") not in x86_pseudo_nops:
correct = False
self.assertTrue(correct)
def test_get_rnd_x86_nop_without_sideeffects(self):
result = get_rnd_x86_nop(1000, True)
correct = True
for byte in result:
if byte.to_bytes(1, "little") in x86_pseudo_nops:
correct = False
self.assertTrue(correct)
def test_get_rnd_x86_nop_filter(self):
result = get_rnd_x86_nop(1000, False, x86_nops.copy())
correct = True
for byte in result:
if byte.to_bytes(1, "little") in x86_nops:
correct = False
self.assertTrue(correct)
def test_get_rnd_x86_nop_single_filter(self):
result = get_rnd_x86_nop(1000, False, b'\x20')
correct = True
for byte in result:
if byte.to_bytes(1, "little") == b'\x20':
correct = False
self.assertTrue(correct)
def test_get_rnd_bytes_number(self):
result = get_rnd_bytes(1000)
self.assertEqual(len(result), 1000)
def test_get_rnd_bytes_filter(self):
result = get_rnd_bytes(1000, x86_pseudo_nops.copy())
correct = True
for byte in result:
if byte.to_bytes(1, "little") in x86_pseudo_nops:
correct = False
self.assertTrue(correct)
def test_get_bytes_from_file_invalid_path(self):
with self.assertRaises(SystemExit):
get_bytes_from_file(test_resource_dir+"/NonExistingFile.txt")
def test_get_bytes_from_file_invalid_header(self):
with self.assertRaises(SystemExit):
get_bytes_from_file(test_resource_dir+"/InvalidHeader.txt")
def test_get_bytes_from_file_invalid_hexfile(self):
with self.assertRaises(SystemExit):
get_bytes_from_file(test_resource_dir+"/InvalidHexFile.txt")
def test_get_bytes_from_file_invalid_strfile(self):
with self.assertRaises(SystemExit):
get_bytes_from_file(test_resource_dir+"/InvalidStringFile.txt")
def test_get_bytes_from_file_str(self):
result = get_bytes_from_file(test_resource_dir+"/StringTestFile.txt")
self.assertEqual(result, b'This is a string-test')
def test_get_bytes_from_file_hex(self):
result = get_bytes_from_file(test_resource_dir+"/HexTestFile.txt")
self.assertEqual(result, b'\xab\xcd\xef\xff\x10\xff\xaa\xab')