88 lines
2.9 KiB
Python
88 lines
2.9 KiB
Python
|
import unittest2
|
||
|
|
|
||
|
|
|
||
|
|
class Test_HttpError(unittest2.TestCase):
|
||
|
|
|
||
|
|
def _getTargetClass(self):
|
||
|
|
from gcloud.streaming.exceptions import HttpError
|
||
|
|
return HttpError
|
||
|
|
|
||
|
|
def _makeOne(self, *args, **kw):
|
||
|
|
return self._getTargetClass()(*args, **kw)
|
||
|
|
|
||
|
|
def test_ctor(self):
|
||
|
|
RESPONSE = {'status': '404'}
|
||
|
|
CONTENT = b'CONTENT'
|
||
|
|
URL = 'http://www.example.com'
|
||
|
|
exception = self._makeOne(RESPONSE, CONTENT, URL)
|
||
|
|
self.assertEqual(exception.response, RESPONSE)
|
||
|
|
self.assertEqual(exception.content, CONTENT)
|
||
|
|
self.assertEqual(exception.url, URL)
|
||
|
|
self.assertEqual(exception.status_code, 404)
|
||
|
|
self.assertEqual(
|
||
|
|
str(exception),
|
||
|
|
"HttpError accessing <http://www.example.com>: "
|
||
|
|
"response: <{'status': '404'}>, content <CONTENT>")
|
||
|
|
|
||
|
|
def test_from_response(self):
|
||
|
|
RESPONSE = {'status': '404'}
|
||
|
|
CONTENT = b'CONTENT'
|
||
|
|
URL = 'http://www.example.com'
|
||
|
|
|
||
|
|
class _Response(object):
|
||
|
|
info = RESPONSE
|
||
|
|
content = CONTENT
|
||
|
|
request_url = URL
|
||
|
|
|
||
|
|
klass = self._getTargetClass()
|
||
|
|
exception = klass.from_response(_Response())
|
||
|
|
self.assertTrue(isinstance(exception, klass))
|
||
|
|
self.assertEqual(exception.response, RESPONSE)
|
||
|
|
self.assertEqual(exception.content, CONTENT)
|
||
|
|
self.assertEqual(exception.url, URL)
|
||
|
|
|
||
|
|
|
||
|
|
class Test_RetryAfterError(unittest2.TestCase):
|
||
|
|
|
||
|
|
def _getTargetClass(self):
|
||
|
|
from gcloud.streaming.exceptions import RetryAfterError
|
||
|
|
return RetryAfterError
|
||
|
|
|
||
|
|
def _makeOne(self, *args, **kw):
|
||
|
|
return self._getTargetClass()(*args, **kw)
|
||
|
|
|
||
|
|
def test_ctor(self):
|
||
|
|
RESPONSE = {'status': '404'}
|
||
|
|
CONTENT = b'CONTENT'
|
||
|
|
URL = 'http://www.example.com'
|
||
|
|
RETRY_AFTER = 60
|
||
|
|
exception = self._makeOne(RESPONSE, CONTENT, URL, RETRY_AFTER)
|
||
|
|
self.assertEqual(exception.response, RESPONSE)
|
||
|
|
self.assertEqual(exception.content, CONTENT)
|
||
|
|
self.assertEqual(exception.url, URL)
|
||
|
|
self.assertEqual(exception.retry_after, RETRY_AFTER)
|
||
|
|
self.assertEqual(
|
||
|
|
str(exception),
|
||
|
|
"HttpError accessing <http://www.example.com>: "
|
||
|
|
"response: <{'status': '404'}>, content <CONTENT>")
|
||
|
|
|
||
|
|
def test_from_response(self):
|
||
|
|
RESPONSE = {'status': '404'}
|
||
|
|
CONTENT = b'CONTENT'
|
||
|
|
URL = 'http://www.example.com'
|
||
|
|
RETRY_AFTER = 60
|
||
|
|
|
||
|
|
class _Response(object):
|
||
|
|
info = RESPONSE
|
||
|
|
content = CONTENT
|
||
|
|
request_url = URL
|
||
|
|
retry_after = RETRY_AFTER
|
||
|
|
|
||
|
|
klass = self._getTargetClass()
|
||
|
|
exception = klass.from_response(_Response())
|
||
|
|
self.assertTrue(isinstance(exception, klass))
|
||
|
|
self.assertEqual(exception.response, RESPONSE)
|
||
|
|
self.assertEqual(exception.content, CONTENT)
|
||
|
|
self.assertEqual(exception.url, URL)
|
||
|
|
self.assertEqual(exception.retry_after, RETRY_AFTER)
|