def test_isPcapFile_bad_files():
status, files = teFileHandler.getFilesFromDirectory(dir_bad_files)
for f in files:
assert teFileHandler.isPcapFile(f) == False
# import dataSetup
# dataSetup.DeleteAll()
# dataSetup.CreateDirectories()
# dataSetup.CreateAllFiles()
# test_isPcapFile_pcaps()
# dataSetup.DeleteAll()
# # Make sure the parent dir is a dir, and no files found
# def test_getFilesFromDirectory_dir_parent():
# assert teFileHandler.getFilesFromDirectory(dir_parent) == (True, [])
# # find all the files recursivly from the parent dir
# def test_getFilesFromDirectory_dir_parent_recurs():
# assert teFileHandler.getFilesFromDirectory(dir_parent, True) == (True, ['test_data\\csv_files\\csv1.csv', 'test_data\\log_files\\log1.log', 'test_data\\mixed_files\\csv1.csv', 'test_data\\mixed_files\\log1.log', 'test_data\\mixed_files\\pcap1.pcap', 'test_data\\mixed_files\\pcap2.pcap', 'test_data\\mixed_files\\txt1.txt', 'test_data\\pcap_files\\pcap1.pcap', 'test_data\\pcap_files\\pcap2.pcap', 'test_data\\txt_files\\txt1.txt'])
# # find all the pcap files in the pcap dir
# def test_getFilesFromDirectory_dir_pcap_files():
# assert teFileHandler.getFilesFromDirectory(dir_pcap_files) == (True, ['test_data\\pcap_files\\pcap1.pcap', 'test_data\\pcap_files\\pcap2.pcap'])
def test_getStats_dir_parent_recur():
status,files = teFileHandler.getFilesFromDirectory(dir_parent, True)
sortedFiles = teFileHandler.sortFilesByType(files)
assert teFileHandler.getStats(sortedFiles) == [{'domains': 1, 'ips': 8, 'filename': ['test_data\\mixed_files\\pcap1.pcap']}, {'domains': 2, 'ips': 23, 'filename': ['test_data\\mixed_files\\pcap2.pcap']}, {'domains': 1, 'ips': 8, 'filename': ['test_data\\pcap_files\\pcap1.pcap']}, {'domains': 2, 'ips': 23, 'filename': ['test_data\\pcap_files\\pcap2.pcap']}, {'domains': 1, 'ips': 3, 'md5hashes': 3, 'sha1hashes': 3, 'filename': 'test_data\\csv_files\\csv1.csv'}, {'domains': 0, 'ips': 6, 'md5hashes': 0, 'sha1hashes': 0, 'filename': 'test_data\\log_files\\log1.log'}, {'domains': 1, 'ips': 3, 'md5hashes': 3, 'sha1hashes': 3, 'filename': 'test_data\\mixed_files\\csv1.csv'}, {'domains': 0, 'ips': 6, 'md5hashes': 0, 'sha1hashes': 0, 'filename': 'test_data\\mixed_files\\log1.log'}, {'domains': 2, 'ips': 6, 'md5hashes': 6, 'sha1hashes': 5, 'filename': 'test_data\\mixed_files\\txt1.txt'}, {'domains': 2, 'ips': 6, 'md5hashes': 6, 'sha1hashes': 5, 'filename': 'test_data\\txt_files\\txt1.txt'}]
def test_sortFilesByType_mixed_files():
status, files = teFileHandler.getFilesFromDirectory(dir_mixed_files)
assert teFileHandler.sortFilesByType(files) == {'text': ['test_data\\mixed_files\\csv1.csv', 'test_data\\mixed_files\\log1.log', 'test_data\\mixed_files\\txt1.txt'], 'pcap': ['test_data\\mixed_files\\pcap1.pcap', 'test_data\\mixed_files\\pcap2.pcap']}
def test_isPcapFile_txt_files():
status, files = teFileHandler.getFilesFromDirectory(dir_txt_files)
for f in files:
assert teFileHandler.isPcapFile(f) == False
def test_isTextFile_log_files():
status, files = teFileHandler.getFilesFromDirectory(dir_log_files)
for f in files:
assert teFileHandler.isTextFile(f) == True
def test_getFilesFromDirectory_dir_pcap_files(): assert teFileHandler.getFilesFromDirectory(dir_pcap_files) == (True, ['test_data\\pcap_files\\pcap1.pcap', 'test_data\\pcap_files\\pcap2.pcap'])
def test_getFilesFromDirectory_dir_parent_recurs(): assert teFileHandler.getFilesFromDirectory(dir_parent, True) == (True, ['test_data\\csv_files\\csv1.csv', 'test_data\\log_files\\log1.log', 'test_data\\mixed_files\\csv1.csv', 'test_data\\mixed_files\\log1.log', 'test_data\\mixed_files\\pcap1.pcap', 'test_data\\mixed_files\\pcap2.pcap', 'test_data\\mixed_files\\txt1.txt', 'test_data\\pcap_files\\pcap1.pcap', 'test_data\\pcap_files\\pcap2.pcap', 'test_data\\txt_files\\txt1.txt'])
def test_getFilesFromDirectory_dir_parent(): assert teFileHandler.getFilesFromDirectory(dir_parent) == (True, [])
