| 12345678910111213141516171819202122232425262728293031323334353637 |
- import os
- import unittest
- from mock import call, MagicMock, patch
- from yarg import get, HTTPError
- class GoodResponse(object):
- status_code = 200
- package = os.path.join(os.path.dirname(__file__),
- 'package.json')
- content = open(package).read()
- class BadResponse(object):
- status_code = 300
- reason = "Mocked"
- class TestClient(unittest.TestCase):
- @patch('requests.get', return_value=BadResponse)
- def test_get(self, get_mock):
- # Python 2.6....
- try:
- get("test")
- except HTTPError as e:
- self.assertEqual(300, e.status_code)
- self.assertEqual(e.status_code, e.errno)
- self.assertEqual(e.reason, e.message)
- @patch('requests.get', return_value=GoodResponse)
- def test_end_slash(self, get_mock):
- get("test", pypi_server="https://mock.test.mock/test")
- self.assertEqual(call('https://mock.test.mock/test/test/json'),
- get_mock.call_args)
|
