CSEC_PUBLIC/securityonion
1
0
Fork 0
mirror of https://github.com/Security-Onion-Solutions/securityonion.git synced 2026-05-04 18:37:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

Improve unit test stability

Browse Source
This commit is contained in:
Jason Ertel
2022-04-05 07:36:25 -04:00
parent d2fa80e48a
commit d8defdd7b0
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
salt/sensoroni/files/analyzers/urlhaus/urlhaus_test.py
+4 -4
View File
@@ -13,7 +13,7 @@ class TestUrlhausMethods(unittest.TestCase):
urlhaus.main()
expected = '{"foo": "bar"}\n'
self.assertEqual(mock_stdout.getvalue(), expected)
mock.assert_called_once_with('.')
mock.assert_called_once()
def test_buildReq(self):
result = urlhaus.buildReq("test")
@@ -31,14 +31,14 @@ class TestUrlhausMethods(unittest.TestCase):
results = urlhaus.prepareResults(raw)
self.assertEqual(results["response"], raw)
self.assertEqual(results["summary"], "no_results")
self.assertEqual(results["status"], "info")
self.assertEqual(results["status"], "ok")
def test_prepareResults_invalidUrl(self):
raw = {"query_status": "invalid_url"}
results = urlhaus.prepareResults(raw)
self.assertEqual(results["response"], raw)
self.assertEqual(results["summary"], "invalid_url")
self.assertEqual(results["status"], "error")
self.assertEqual(results["status"], "caution")
def test_prepareResults_threat(self):
raw = {"query_status": "invalid_url"} # This is overrided in this scenario
@@ -46,7 +46,7 @@ class TestUrlhausMethods(unittest.TestCase):
results = urlhaus.prepareResults(raw)
self.assertEqual(results["response"], raw)
self.assertEqual(results["summary"], "bad_actor")
self.assertEqual(results["status"], "danger")
self.assertEqual(results["status"], "threat")
def test_analyze(self):
output = {"threat": "malware_download"}