class TestHTTPClient(testtools.TestCase): def setUp(self): super(TestHTTPClient, self).setUp() self.mox = mox.Mox() self.mox.StubOutWithMock(HTTPClient, 'request') self.addCleanup(self.mox.UnsetStubs) self.http = HTTPClient(token=AUTH_TOKEN, endpoint_url=END_URL) def test_request_error(self): HTTPClient.request( URL, METHOD, headers=mox.IgnoreArg() ).AndRaise(Exception('error msg')) self.mox.ReplayAll() self.assertRaises( exceptions.ConnectionFailed, self.http._cs_request, URL, METHOD ) self.mox.VerifyAll() def test_request_success(self): rv_should_be = MyResp(200), 'test content' HTTPClient.request( URL, METHOD, headers=mox.IgnoreArg() ).AndReturn(rv_should_be) self.mox.ReplayAll() self.assertEqual(rv_should_be, self.http._cs_request(URL, METHOD)) self.mox.VerifyAll() def test_request_unauthorized(self): rv_should_be = MyResp(401), 'unauthorized message' HTTPClient.request( URL, METHOD, headers=mox.IgnoreArg() ).AndReturn(rv_should_be) self.mox.ReplayAll() e = self.assertRaises(exceptions.Unauthorized, self.http._cs_request, URL, METHOD) self.assertEqual('unauthorized message', e.message) self.mox.VerifyAll() def test_request_forbidden_is_returned_to_caller(self): rv_should_be = MyResp(403), 'forbidden message' HTTPClient.request( URL, METHOD, headers=mox.IgnoreArg() ).AndReturn(rv_should_be) self.mox.ReplayAll() self.assertEqual(rv_should_be, self.http._cs_request(URL, METHOD)) self.mox.VerifyAll()
class TestHTTPClient(testtools.TestCase): def setUp(self): super(TestHTTPClient, self).setUp() self.mox = mox.Mox() self.mox.StubOutWithMock(HTTPClient, 'request') self.addCleanup(self.mox.UnsetStubs) self.http = HTTPClient(token=AUTH_TOKEN, endpoint_url=END_URL) def test_request_error(self): HTTPClient.request(URL, METHOD, headers=mox.IgnoreArg()).AndRaise( Exception('error msg')) self.mox.ReplayAll() self.assertRaises(exceptions.ConnectionFailed, self.http._cs_request, URL, METHOD) self.mox.VerifyAll() def test_request_success(self): rv_should_be = MyResp(200), 'test content' HTTPClient.request(URL, METHOD, headers=mox.IgnoreArg()).AndReturn(rv_should_be) self.mox.ReplayAll() self.assertEqual(rv_should_be, self.http._cs_request(URL, METHOD)) self.mox.VerifyAll() def test_request_unauthorized(self): rv_should_be = MyResp(401), 'unauthorized message' HTTPClient.request(URL, METHOD, headers=mox.IgnoreArg()).AndReturn(rv_should_be) self.mox.ReplayAll() e = self.assertRaises(exceptions.Unauthorized, self.http._cs_request, URL, METHOD) self.assertEqual('unauthorized message', e.message) self.mox.VerifyAll() def test_request_forbidden_is_returned_to_caller(self): rv_should_be = MyResp(403), 'forbidden message' HTTPClient.request(URL, METHOD, headers=mox.IgnoreArg()).AndReturn(rv_should_be) self.mox.ReplayAll() self.assertEqual(rv_should_be, self.http._cs_request(URL, METHOD)) self.mox.VerifyAll()
class TestHTTPClient(testtools.TestCase): def setUp(self): super(TestHTTPClient, self).setUp() self.mox = mox.Mox() self.mox.StubOutWithMock(httplib2.Http, 'request') self.addCleanup(self.mox.UnsetStubs) self.http = HTTPClient(token=AUTH_TOKEN, endpoint_url=END_URL) def test_request_error(self): httplib2.Http.request(URL, METHOD, headers=mox.IgnoreArg()).AndRaise( Exception('error msg')) self.mox.ReplayAll() self.assertRaises(exceptions.ConnectionFailed, self.http._cs_request, URL, METHOD) self.mox.VerifyAll() def test_request_success(self): rv_should_be = MyResp(200), 'test content' httplib2.Http.request(URL, METHOD, headers=mox.IgnoreArg()).AndReturn(rv_should_be) self.mox.ReplayAll() self.assertEqual(rv_should_be, self.http._cs_request(URL, METHOD)) self.mox.VerifyAll()
class TestHTTPClient(testtools.TestCase): def setUp(self): super(TestHTTPClient, self).setUp() self.mox = mox.Mox() self.mox.StubOutWithMock(httplib2.Http, 'request') self.addCleanup(self.mox.UnsetStubs) self.http = HTTPClient(token=AUTH_TOKEN, endpoint_url=END_URL) def test_request_error(self): httplib2.Http.request( URL, METHOD, headers=mox.IgnoreArg() ).AndRaise(Exception('error msg')) self.mox.ReplayAll() self.assertRaises( exceptions.ConnectionFailed, self.http._cs_request, URL, METHOD ) self.mox.VerifyAll() def test_request_success(self): rv_should_be = MyResp(200), 'test content' httplib2.Http.request( URL, METHOD, headers=mox.IgnoreArg() ).AndReturn(rv_should_be) self.mox.ReplayAll() self.assertEqual(rv_should_be, self.http._cs_request(URL, METHOD)) self.mox.VerifyAll()