Revision a037fd61
| b/kamaki/clients/__init__.py | ||
|---|---|---|
| 231 | 231 |
self.http_client.reset_params() |
| 232 | 232 |
|
| 233 | 233 |
if success is not None: |
| 234 |
# Success can either be an in or a collection |
|
| 234 |
# Success can either be an int or a collection
|
|
| 235 | 235 |
success = (success,) if isinstance(success, int) else success |
| 236 | 236 |
if r.status_code not in success: |
| 237 | 237 |
r.release() |
| b/kamaki/clients/test/__init__.py | ||
|---|---|---|
| 32 | 32 |
# or implied, of GRNET S.A. |
| 33 | 33 |
|
| 34 | 34 |
from unittest import TestCase, TestSuite, makeSuite, TextTestRunner |
| 35 |
#from mock import Mock, patch |
|
| 35 |
from mock import Mock, patch |
|
| 36 |
|
|
| 37 |
|
|
| 38 |
class _fakeResponse(object): |
|
| 39 |
sample = 'sample string' |
|
| 40 |
getheaders = Mock(return_value={})
|
|
| 41 |
content = Mock(return_value=sample) |
|
| 42 |
read = Mock(return_value=sample) |
|
| 43 |
status = Mock(return_value=None) |
|
| 44 |
status_code = 200 |
|
| 45 |
reason = Mock(return_value=None) |
|
| 46 |
release = Mock() |
|
| 47 |
headers = {}
|
|
| 48 |
|
|
| 49 |
|
|
| 50 |
class Client(TestCase): |
|
| 51 |
|
|
| 52 |
def setUp(self): |
|
| 53 |
from kamaki.clients import Client |
|
| 54 |
from kamaki.clients.connection.kamakicon import ( |
|
| 55 |
KamakiHTTPConnection) |
|
| 56 |
self.token = 'F@k3T0k3n' |
|
| 57 |
self.base_url = 'http://www.example.com' |
|
| 58 |
self.KC = KamakiHTTPConnection |
|
| 59 |
self.KR = _fakeResponse |
|
| 60 |
self.c = Client(self.base_url, self.token, self.KC()) |
|
| 61 |
|
|
| 62 |
def test_request(self): |
|
| 63 |
req = self.c.request |
|
| 64 |
method = 'GET' |
|
| 65 |
path = '/online/path' |
|
| 66 |
with patch.object(self.KC, 'perform_request', return_value=self.KR()): |
|
| 67 |
r = req(method, path) |
|
| 68 |
self.assertTrue(isinstance(r, self.KR)) |
|
| 36 | 69 |
|
| 37 | 70 |
|
| 38 | 71 |
def get_test_classes(module=__import__(__name__), name=''): |
Also available in: Unified diff