test(success): magicmock

sladg · Apr 25, 2023 · f5dc6b1 · f5dc6b1
1 parent b5b8c96
commit f5dc6b1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/test_success.py b/test/test_success.py
@@ -3,6 +3,7 @@
 import pytest
 
 from imaginex_lambda.handler import handler, S3_BUCKET_NAME, download_and_optimize
+from unittest.mock import MagicMock
 
 
 def test_handler_success():
@@ -16,10 +17,9 @@ def test_handler_success():
     """
 
     context = {'queryStringParameters': {'url': 'abc.png', 'q': '50', 'w': '100'}}
-    fake_optimization_return = (b'abcdef', 'application/someimage', 0.3)
+    fake_optimization_return = MagicMock(return_value=((b'abcdef', 'application/someimage', 0.3)))
 
-    with patch('imaginex_lambda.handler.download_and_optimize') as p:
-        p.return_value = fake_optimization_return
+    with patch('imaginex_lambda.handler.download_and_optimize', fake_optimization_return) as p:
         r = handler(context, None)
 
     assert r['statusCode'] == 200