'application/json')
self.assertEqual(retval.body, '{}')
- def test_proxy_request_network_404(self):
+ def _test_proxy_request_network_4xx(self, status, method, expected):
self.handler.network_id = 'network_id'
- resp = mock.Mock(status=404)
+ resp = mock.Mock(status=status)
with mock.patch('httplib2.Http') as mock_http:
mock_http.return_value.request.return_value = (resp, '')
retval = self.handler._proxy_request('192.168.1.1',
- 'GET',
+ method,
'/latest/meta-data',
'',
'')
mock_http.assert_has_calls([
mock.call().request(
'http://169.254.169.254/latest/meta-data',
- method='GET',
+ method=method,
headers={
'X-Forwarded-For': '192.168.1.1',
'X-Neutron-Network-ID': 'network_id'
)]
)
- self.assertIsInstance(retval, webob.exc.HTTPNotFound)
+ self.assertIsInstance(retval, expected)
- def test_proxy_request_network_409(self):
- self.handler.network_id = 'network_id'
-
- resp = mock.Mock(status=409)
- with mock.patch('httplib2.Http') as mock_http:
- mock_http.return_value.request.return_value = (resp, '')
-
- retval = self.handler._proxy_request('192.168.1.1',
- 'POST',
- '/latest/meta-data',
- '',
- '')
+ def test_proxy_request_network_400(self):
+ self._test_proxy_request_network_4xx(
+ 400, 'GET', webob.exc.HTTPBadRequest)
- mock_http.assert_has_calls([
- mock.call().request(
- 'http://169.254.169.254/latest/meta-data',
- method='POST',
- headers={
- 'X-Forwarded-For': '192.168.1.1',
- 'X-Neutron-Network-ID': 'network_id'
- },
- connection_type=ns_proxy.UnixDomainHTTPConnection,
- body=''
- )]
- )
+ def test_proxy_request_network_404(self):
+ self._test_proxy_request_network_4xx(
+ 404, 'GET', webob.exc.HTTPNotFound)
- self.assertIsInstance(retval, webob.exc.HTTPConflict)
+ def test_proxy_request_network_409(self):
+ self._test_proxy_request_network_4xx(
+ 409, 'POST', webob.exc.HTTPConflict)
def test_proxy_request_network_500(self):
self.handler.network_id = 'network_id'