diff options
Diffstat (limited to '')
| -rw-r--r-- | pygithub3/tests/utils/core.py | 5 | ||||
| -rw-r--r-- | pygithub3/tests/utils/resources.py | 7 |
2 files changed, 12 insertions, 0 deletions
diff --git a/pygithub3/tests/utils/core.py b/pygithub3/tests/utils/core.py index 8d18732..32170e4 100644 --- a/pygithub3/tests/utils/core.py +++ b/pygithub3/tests/utils/core.py @@ -1,6 +1,11 @@ #!/usr/bin/env python # -*- encoding: utf-8 -*- +try: + from unittest2 import TestCase # Python 2.6 +except ImportError: + from unittest import TestCase # Python >2.7 + from .base import Mock, DummyRequest request = DummyRequest() diff --git a/pygithub3/tests/utils/resources.py b/pygithub3/tests/utils/resources.py index d9e4975..be9245b 100644 --- a/pygithub3/tests/utils/resources.py +++ b/pygithub3/tests/utils/resources.py @@ -1,7 +1,14 @@ #!/usr/bin/env python # -*- encoding: utf-8 -*- +from mock import Mock + from .base import Resource +from pygithub3.resources.base import json +from pygithub3.tests.utils.base import mock_json + +json.dumps = Mock(side_effect=mock_json) +json.loads = Mock(side_effect=mock_json) class Simple(Resource): |