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, [])