def test_assertion_types_and_extension(tmpdir): """ Given: - both types and extensions arguments When: - Running script Then: - Validate AssertionError is raises as you shouldn't supply them both. """ with raises(AssertionError): upload_files('', tmpdir, types='1,2,3', extensions='1,2,3')
def test_decryption_rsa(tmpdir): """ Given: - A PCAP file with ssl encryption - A Key file for the pcap When: - Running PcapFileExtractor with TLS protected file and key Then: - Validate results output that the files are exported and returned to CortexSOAR """ file_path = './TestData/rsa.cap' key_path = './TestData/rsa.key' results = upload_files(file_path, tmpdir, rsa_path=key_path) assert 5 == len(results.outputs)
def test_decryption_wpa_pwd(tmpdir): """ Given: - A PCAP file protected with WPA-PWD - A password to the file When: - Running PcapFileExtractor with WPA-PWD protected file Then: - Validate results output that the files are exported and returned to CortexSOAR """ file_path = './TestData/wpa-Induction.pcap' password = '******' results = upload_files(file_path, tmpdir, wpa_pwd=password) assert 5 == len(results.outputs)
def test_extract_files(tmpdir): """ Given - Valid Pcap File. When - Extracting the file from the pcap. Then - ensure file is being extracted correctly. - ensure outputs are correct. """ file_path = './TestData/tftp_rrq.pcap' results = upload_files(file_path, tmpdir) assert 'Pcap Extracted Files' in results.readable_output assert OUTPUTS == results.outputs assert os.path.isfile(os.path.join(tmpdir, 'rfc1350.txt'))