MISP
/
misp-modules
mirror of https://github.com/MISP/misp-modules
2
0
Fork 0
Code Issues Releases Wiki Activity

fix: [test] pdftotext output check 

The important part is the matching text from the PDF not any trailling
which might be different depending of the encoding.
pull/613/head v2.4.170
Alexandre Dulaunoy 2023-04-02 10:47:41 +02:00
parent 113a112001
commit c1168ac627
No known key found for this signature in database
GPG Key ID: 09E2CD4944E6CBCD
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_expansions.py
View File

@ -195,7 +195,7 @@ class TestExpansions(unittest.TestCase):
query = {"module": "dbl_spamhaus", "domain": "totalmateria.net"} query = {"module": "dbl_spamhaus", "domain": "totalmateria.net"}
response = self.misp_modules_post(query) response = self.misp_modules_post(query)
try: try:
self.assertEqual(self.get_values(response), 'totalmateria.net - spam domain') self.assertEqual(self.get_values(response), 'totalmateria.net - spam test domain')
except Exception: except Exception:
try: try:
self.assertTrue(self.get_values(response).startswith('None of DNS query names exist:')) self.assertTrue(self.get_values(response).startswith('None of DNS query names exist:'))
@ -263,7 +263,7 @@ class TestExpansions(unittest.TestCase):
self.assertEqual(to_check, 'OK (Not Found)', response) self.assertEqual(to_check, 'OK (Not Found)', response)
else: else:
self.assertEqual(self.get_errors(response), 'Have I Been Pwned authentication is incomplete (no API key)') self.assertEqual(self.get_errors(response), 'Have I Been Pwned authentication is incomplete (no API key)')
def test_hyasinsight(self): def test_hyasinsight(self):
module_name = "hyasinsight" module_name = "hyasinsight"
query = {"module": module_name, query = {"module": module_name,
@ -432,7 +432,7 @@ class TestExpansions(unittest.TestCase):
encoded = b64encode(f.read()).decode() encoded = b64encode(f.read()).decode()
query = {"module": "pdf_enrich", "attachment": filename, "data": encoded} query = {"module": "pdf_enrich", "attachment": filename, "data": encoded}
response = self.misp_modules_post(query) response = self.misp_modules_post(query)
self.assertEqual(self.get_values(response), 'Pdf test') self.assertRegex(self.get_values(response), r'^Pdf test')
def test_pptx(self): def test_pptx(self):
filename = 'test.pptx' filename = 'test.pptx'