14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
from StringIO import StringIO
18
from ..errors import BinaryFile
19
from ..sixish import (
22
from . import TestCase, TestCaseInTempDir
23
from ..textfile import text_file, check_text_lines, check_text_path
19
from bzrlib.errors import BinaryFile
20
from bzrlib.tests import TestCase, TestCaseInTempDir
21
from bzrlib.textfile import text_file, check_text_lines, check_text_path
26
24
class TextFile(TestCase):
28
26
def test_text_file(self):
29
s = BytesIO(b'ab' * 2048)
27
s = StringIO('ab' * 2048)
30
28
self.assertEqual(text_file(s).read(), s.getvalue())
31
s = BytesIO(b'a' * 1023 + b'\x00')
29
s = StringIO('a' * 1023 + '\x00')
32
30
self.assertRaises(BinaryFile, text_file, s)
33
s = BytesIO(b'a' * 1024 + b'\x00')
31
s = StringIO('a' * 1024 + '\x00')
34
32
self.assertEqual(text_file(s).read(), s.getvalue())
36
34
def test_check_text_lines(self):