CSEC_PUBLIC/securityonion
1
0
Fork 0
mirror of https://github.com/Security-Onion-Solutions/securityonion.git synced 2025-12-06 17:22:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update tests to account for change in 'file_path' value verification

Browse Source
This commit is contained in:
Wes
2022-12-13 16:29:18 +00:00
parent eae05e83e6
commit 3ab8a0be60
1 changed files with 1 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
salt/sensoroni/files/analyzers/localfile/localfile_test.py
View File

@@ -28,22 +28,6 @@ class TestLocalfileMethods(unittest.TestCase):
mock.assert_called_once() mock.assert_called_once()
lcmock.assert_called_once() lcmock.assert_called_once()
def test_checkConfigRequirements_present(self):
conf = {"file_path": "['intel.csv']"}
self.assertTrue(localfile.checkConfigRequirements(conf))
def test_checkConfigRequirements_not_present(self):
conf = {"not_a_file_path": "blahblah"}
with self.assertRaises(SystemExit) as cm:
localfile.checkConfigRequirements(conf)
self.assertEqual(cm.exception.code, 126)
def test_checkConfigRequirements_empty(self):
conf = {"file_path": ""}
with self.assertRaises(SystemExit) as cm:
localfile.checkConfigRequirements(conf)
self.assertEqual(cm.exception.code, 126)
def test_searchFile_multiple_found(self): def test_searchFile_multiple_found(self):
artifact = "abcd1234" artifact = "abcd1234"
results = localfile.searchFile(artifact, ["localfile_test.csv"]) results = localfile.searchFile(artifact, ["localfile_test.csv"])
@@ -115,7 +99,7 @@ class TestLocalfileMethods(unittest.TestCase):
} }
] ]
artifactInput = '{"value":"foo","artifactType":"url"}' artifactInput = '{"value":"foo","artifactType":"url"}'
conf = {"file_path": "/home/intel.csv"} conf = {"file_path": ['/home/intel.csv']}
with patch('localfile.localfile.searchFile', new=MagicMock(return_value=output)) as mock: with patch('localfile.localfile.searchFile', new=MagicMock(return_value=output)) as mock:
results = localfile.analyze(conf, artifactInput) results = localfile.analyze(conf, artifactInput)
self.assertEqual(results["summary"], "suspicious") self.assertEqual(results["summary"], "suspicious")