ea8b9e7a16
The Oslo libraries have moved all of their code out of the 'oslo' namespace package into per-library packages. The namespace package was retained during kilo for backwards compatibility, but will be removed by the liberty-2 milestone. This change removes the use of the namespace package, replacing it with the new package names. The patches in the libraries will be put on hold until application patches have landed, or L2, whichever comes first. At that point, new versions of the libraries without namespace packages will be released as a major version update. Please merge this patch, or an equivalent, before L2 to avoid problems with those library releases. Blueprint: remove-namespace-packages https://blueprints.launchpad.net/oslo-incubator/+spec/remove-namespace-packages Change-Id: I7344c794b73aca1e0dcad681a4c43e3a66dd2eb3
109 lines
3.8 KiB
Python
109 lines
3.8 KiB
Python
# Copyright (c) 2012 Red Hat, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
from oslotest import base as test_base
|
|
import six
|
|
import webob
|
|
|
|
from oslo.middleware import sizelimit
|
|
from oslo_config import fixture as config
|
|
|
|
|
|
class TestLimitingReader(test_base.BaseTestCase):
|
|
|
|
def test_limiting_reader(self):
|
|
BYTES = 1024
|
|
bytes_read = 0
|
|
data = six.StringIO("*" * BYTES)
|
|
for chunk in sizelimit.LimitingReader(data, BYTES):
|
|
bytes_read += len(chunk)
|
|
|
|
self.assertEqual(bytes_read, BYTES)
|
|
|
|
bytes_read = 0
|
|
data = six.StringIO("*" * BYTES)
|
|
reader = sizelimit.LimitingReader(data, BYTES)
|
|
byte = reader.read(1)
|
|
while len(byte) != 0:
|
|
bytes_read += 1
|
|
byte = reader.read(1)
|
|
|
|
self.assertEqual(bytes_read, BYTES)
|
|
|
|
def test_read_default_value(self):
|
|
BYTES = 1024
|
|
data_str = "*" * BYTES
|
|
data = six.StringIO(data_str)
|
|
reader = sizelimit.LimitingReader(data, BYTES)
|
|
res = reader.read()
|
|
self.assertEqual(data_str, res)
|
|
|
|
def test_limiting_reader_fails(self):
|
|
BYTES = 1024
|
|
|
|
def _consume_all_iter():
|
|
bytes_read = 0
|
|
data = six.StringIO("*" * BYTES)
|
|
for chunk in sizelimit.LimitingReader(data, BYTES - 1):
|
|
bytes_read += len(chunk)
|
|
|
|
self.assertRaises(webob.exc.HTTPRequestEntityTooLarge,
|
|
_consume_all_iter)
|
|
|
|
def _consume_all_read():
|
|
bytes_read = 0
|
|
data = six.StringIO("*" * BYTES)
|
|
reader = sizelimit.LimitingReader(data, BYTES - 1)
|
|
byte = reader.read(1)
|
|
while len(byte) != 0:
|
|
bytes_read += 1
|
|
byte = reader.read(1)
|
|
|
|
self.assertRaises(webob.exc.HTTPRequestEntityTooLarge,
|
|
_consume_all_read)
|
|
|
|
|
|
class TestRequestBodySizeLimiter(test_base.BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestRequestBodySizeLimiter, self).setUp()
|
|
fixture = self.useFixture(config.Config(sizelimit.CONF))
|
|
self.MAX_REQUEST_BODY_SIZE = \
|
|
fixture.conf.oslo_middleware.max_request_body_size
|
|
|
|
@webob.dec.wsgify()
|
|
def fake_app(req):
|
|
return webob.Response(req.body)
|
|
|
|
self.middleware = sizelimit.RequestBodySizeLimiter(fake_app)
|
|
self.request = webob.Request.blank('/', method='POST')
|
|
|
|
def test_content_length_acceptable(self):
|
|
self.request.headers['Content-Length'] = self.MAX_REQUEST_BODY_SIZE
|
|
self.request.body = b"0" * self.MAX_REQUEST_BODY_SIZE
|
|
response = self.request.get_response(self.middleware)
|
|
self.assertEqual(response.status_int, 200)
|
|
|
|
def test_content_length_too_large(self):
|
|
self.request.headers['Content-Length'] = self.MAX_REQUEST_BODY_SIZE + 1
|
|
self.request.body = b"0" * (self.MAX_REQUEST_BODY_SIZE + 1)
|
|
response = self.request.get_response(self.middleware)
|
|
self.assertEqual(response.status_int, 413)
|
|
|
|
def test_request_too_large_no_content_length(self):
|
|
self.request.body = b"0" * (self.MAX_REQUEST_BODY_SIZE + 1)
|
|
self.request.headers['Content-Length'] = None
|
|
response = self.request.get_response(self.middleware)
|
|
self.assertEqual(response.status_int, 413)
|