Restore test suite compatability with Python 3.x.
This commit is contained in:
@@ -9,6 +9,7 @@ from testtools import TestCase
|
||||
from testtools.compat import (
|
||||
_b,
|
||||
_u,
|
||||
BytesIO,
|
||||
StringIO,
|
||||
)
|
||||
from testtools.content import (
|
||||
@@ -127,24 +128,23 @@ class TestContent(TestCase):
|
||||
|
||||
def test_from_file_with_simple_seek(self):
|
||||
f = tempfile.NamedTemporaryFile()
|
||||
f.write('some data')
|
||||
f.write(_b('some data'))
|
||||
f.flush()
|
||||
self.addCleanup(f.close)
|
||||
content = content_from_file(
|
||||
f.name, UTF8_TEXT, chunk_size=50, seek_offset=5)
|
||||
self.assertThat(
|
||||
list(content.iter_bytes()), Equals(['data']))
|
||||
list(content.iter_bytes()), Equals([_b('data')]))
|
||||
|
||||
def test_from_file_with_whence_seek(self):
|
||||
f = tempfile.NamedTemporaryFile()
|
||||
f.write('some data')
|
||||
f.write(_b('some data'))
|
||||
f.flush()
|
||||
self.addCleanup(f.close)
|
||||
content = content_from_file(
|
||||
f.name, UTF8_TEXT, chunk_size=50, seek_offset=-4, seek_whence=2)
|
||||
self.assertThat(
|
||||
list(content.iter_bytes()), Equals(['data']))
|
||||
|
||||
list(content.iter_bytes()), Equals([_b('data')]))
|
||||
|
||||
def test_from_stream(self):
|
||||
data = StringIO('some data')
|
||||
@@ -170,18 +170,18 @@ class TestContent(TestCase):
|
||||
''.join(content.iter_text()), Equals('some data'))
|
||||
|
||||
def test_from_stream_with_simple_seek(self):
|
||||
data = StringIO('some data')
|
||||
data = BytesIO(_b('some data'))
|
||||
content = content_from_stream(
|
||||
data, UTF8_TEXT, chunk_size=50, seek_offset=5)
|
||||
self.assertThat(
|
||||
list(content.iter_bytes()), Equals(['data']))
|
||||
list(content.iter_bytes()), Equals([_b('data')]))
|
||||
|
||||
def test_from_stream_with_whence_seek(self):
|
||||
data = StringIO('some data')
|
||||
data = BytesIO(_b('some data'))
|
||||
content = content_from_stream(
|
||||
data, UTF8_TEXT, chunk_size=50, seek_offset=-4, seek_whence=2)
|
||||
self.assertThat(
|
||||
list(content.iter_bytes()), Equals(['data']))
|
||||
list(content.iter_bytes()), Equals([_b('data')]))
|
||||
|
||||
def test_from_text(self):
|
||||
data = _u("some data")
|
||||
|
||||
Reference in New Issue
Block a user