diff options
Diffstat (limited to 'oslo_middleware/tests/test_sizelimit.py')
| -rw-r--r-- | oslo_middleware/tests/test_sizelimit.py | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/oslo_middleware/tests/test_sizelimit.py b/oslo_middleware/tests/test_sizelimit.py index 3d57a61..5d547ac 100644 --- a/oslo_middleware/tests/test_sizelimit.py +++ b/oslo_middleware/tests/test_sizelimit.py @@ -12,9 +12,10 @@ # License for the specific language governing permissions and limitations # under the License. +import io + from oslo_config import fixture as config from oslotest import base as test_base -import six import webob from oslo_middleware import sizelimit @@ -25,14 +26,14 @@ class TestLimitingReader(test_base.BaseTestCase): def test_limiting_reader(self): BYTES = 1024 bytes_read = 0 - data = six.StringIO("*" * BYTES) + data = io.StringIO("*" * BYTES) for chunk in sizelimit.LimitingReader(data, BYTES): bytes_read += len(chunk) self.assertEqual(BYTES, bytes_read) bytes_read = 0 - data = six.StringIO("*" * BYTES) + data = io.StringIO("*" * BYTES) reader = sizelimit.LimitingReader(data, BYTES) byte = reader.read(1) while len(byte) != 0: @@ -44,7 +45,7 @@ class TestLimitingReader(test_base.BaseTestCase): def test_read_default_value(self): BYTES = 1024 data_str = "*" * BYTES - data = six.StringIO(data_str) + data = io.StringIO(data_str) reader = sizelimit.LimitingReader(data, BYTES) res = reader.read() self.assertEqual(data_str, res) @@ -54,7 +55,7 @@ class TestLimitingReader(test_base.BaseTestCase): def _consume_all_iter(): bytes_read = 0 - data = six.StringIO("*" * BYTES) + data = io.StringIO("*" * BYTES) for chunk in sizelimit.LimitingReader(data, BYTES - 1): bytes_read += len(chunk) @@ -63,7 +64,7 @@ class TestLimitingReader(test_base.BaseTestCase): def _consume_all_read(): bytes_read = 0 - data = six.StringIO("*" * BYTES) + data = io.StringIO("*" * BYTES) reader = sizelimit.LimitingReader(data, BYTES - 1) byte = reader.read(1) while len(byte) != 0: |