/breezy/unstable

To get this branch, use:
bzr branch https://code.breezy-vcs.org/breezy/unstable

« back to all changes in this revision

Viewing changes to breezy/tests/test_textfile.py

  • Committer: Jelmer Vernooij
  • Date: 2017-05-24 01:39:33 UTC
  • mfrom: (3815.3776.6)
  • Revision ID: jelmer@jelmer.uk-20170524013933-ir4y4tqtrsiz2ka2
New upstream snapshot.

Show diffs side-by-side

added added

removed removed

Lines of Context:
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
16
16
 
17
 
from StringIO import StringIO
18
17
 
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
 
18
from ..errors import BinaryFile
 
19
from ..sixish import (
 
20
    BytesIO,
 
21
    )
 
22
from . import TestCase, TestCaseInTempDir
 
23
from ..textfile import text_file, check_text_lines, check_text_path
22
24
 
23
25
 
24
26
class TextFile(TestCase):
25
27
 
26
28
    def test_text_file(self):
27
 
        s = StringIO('ab' * 2048)
 
29
        s = BytesIO(b'ab' * 2048)
28
30
        self.assertEqual(text_file(s).read(), s.getvalue())
29
 
        s = StringIO('a' * 1023 + '\x00')
 
31
        s = BytesIO(b'a' * 1023 + b'\x00')
30
32
        self.assertRaises(BinaryFile, text_file, s)
31
 
        s = StringIO('a' * 1024 + '\x00')
 
33
        s = BytesIO(b'a' * 1024 + b'\x00')
32
34
        self.assertEqual(text_file(s).read(), s.getvalue())
33
35
 
34
36
    def test_check_text_lines(self):