/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/tests/test_textfile.py

  • Committer: Breezy landing bot
  • Author(s): Jelmer Vernooij
  • Date: 2020-07-28 02:47:10 UTC
  • mfrom: (7519.1.1 merge-3.1)
  • Revision ID: breezy.the.bot@gmail.com-20200728024710-a2ylds219f1lsl62
Merge lp:brz/3.1.

Merged from https://code.launchpad.net/~jelmer/brz/merge-3.1/+merge/388173

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/tests/test_textfile.py
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
 
17
from io import BytesIO
18
18
 
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
 
19
from ..errors import BinaryFile
 
20
from . import TestCase, TestCaseInTempDir
 
21
from ..textfile import text_file, check_text_lines, check_text_path
22
22
 
23
23
 
24
24
class TextFile(TestCase):
25
25
 
26
26
    def test_text_file(self):
27
 
        s = StringIO('ab' * 2048)
 
27
        s = BytesIO(b'ab' * 2048)
28
28
        self.assertEqual(text_file(s).read(), s.getvalue())
29
 
        s = StringIO('a' * 1023 + '\x00')
 
29
        s = BytesIO(b'a' * 1023 + b'\x00')
30
30
        self.assertRaises(BinaryFile, text_file, s)
31
 
        s = StringIO('a' * 1024 + '\x00')
 
31
        s = BytesIO(b'a' * 1024 + b'\x00')
32
32
        self.assertEqual(text_file(s).read(), s.getvalue())
33
33
 
34
34
    def test_check_text_lines(self):
35
 
        lines = ['ab' * 2048]
 
35
        lines = [b'ab' * 2048]
36
36
        check_text_lines(lines)
37
 
        lines = ['a' * 1023 + '\x00']
 
37
        lines = [b'a' * 1023 + b'\x00']
38
38
        self.assertRaises(BinaryFile, check_text_lines, lines)
39
39
 
40
40
 
41
41
class TextPath(TestCaseInTempDir):
42
42
 
43
43
    def test_text_file(self):
44
 
        file('boo', 'wb').write('ab' * 2048)
 
44
        with open('boo', 'wb') as f:
 
45
            f.write(b'ab' * 2048)
45
46
        check_text_path('boo')
46
 
        file('boo', 'wb').write('a' * 1023 + '\x00')
 
47
        with open('boo', 'wb') as f:
 
48
            f.write(b'a' * 1023 + b'\x00')
47
49
        self.assertRaises(BinaryFile, check_text_path, 'boo')