From f05eb742ddfae25e39508c05e875867e206d8640 Mon Sep 17 00:00:00 2001 From: Wes Date: Fri, 15 Dec 2023 14:26:33 +0000 Subject: [PATCH] Fix patch --- .../files/analyzers/malwarebazaar/malwarebazaar_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/salt/sensoroni/files/analyzers/malwarebazaar/malwarebazaar_test.py b/salt/sensoroni/files/analyzers/malwarebazaar/malwarebazaar_test.py index 44c454358..0a4f496a8 100644 --- a/salt/sensoroni/files/analyzers/malwarebazaar/malwarebazaar_test.py +++ b/salt/sensoroni/files/analyzers/malwarebazaar/malwarebazaar_test.py @@ -15,7 +15,7 @@ class TestMalwarebazaarMethods(unittest.TestCase): def test_main_success(self): with patch('sys.stdout', new=StringIO()) as mock_cmd: - with patch('malwarebazaar.analyze', new=MagicMock(return_value={'test': 'val'})) as mock: + with patch('malwarebazaar.malwarebazaar.analyze', new=MagicMock(return_value={'test': 'val'})) as mock: sys.argv = ["cmd", "input"] malwarebazaar.main() expected = '{"test": "val"}\n' @@ -32,8 +32,8 @@ class TestMalwarebazaarMethods(unittest.TestCase): prepareResultOutput = {'response': '', 'summary': 'no result', 'status': 'info'} - with patch('malwarebazaar.sendReq', new=MagicMock(return_value=sendReqOutput)) as mock: - with patch('malwarebazaar.prepareResults', new=MagicMock(return_value=prepareResultOutput)) as mock2: + with patch('malwarebazaar.malwarebazaar.sendReq', new=MagicMock(return_value=sendReqOutput)) as mock: + with patch('malwarebazaar.malwarebazaar.prepareResults', new=MagicMock(return_value=prepareResultOutput)) as mock2: results = malwarebazaar.analyze(input) results2 = malwarebazaar.analyze(input2) results3 = malwarebazaar.analyze(input3)