diff options
Diffstat (limited to 'share/newton_base/tests')
| -rw-r--r-- | share/newton_base/tests/test_base.py | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/share/newton_base/tests/test_base.py b/share/newton_base/tests/test_base.py index 15160d06..b9923d1c 100644 --- a/share/newton_base/tests/test_base.py +++ b/share/newton_base/tests/test_base.py @@ -12,11 +12,17 @@ # See the License for the specific language governing permissions and # limitations under the License. +import sys import json -import mock +# import mock from rest_framework import status import unittest +if sys.version_info < (3, 0): + import mock +else: + from unittest import mock + from abc import ABCMeta from django.conf import settings from django.test import Client @@ -35,9 +41,15 @@ class MockResponse(object): pass +def get_mock_response(return_value=None): + mock_response = mock.Mock(spec=MockResponse) + mock_response.status_code = status.HTTP_200_OK + mock_response.json.return_value = return_value + return mock_response + def get_mock_session(http_actions, response_dict={}): mock_session = mock.Mock( - name='mock_session',spec=http_actions) + name='mock_session', spec=http_actions) for action in http_actions: side_effect = response_dict.get("side_effect") if side_effect and isinstance(side_effect, list): |