Gecko [ gtgates.com ]

Name	Size	Permission	Date
__init__.py	107 B	-rw-r--r--	2016-06-19 12:55
__init__.pyc	323 B	-rw-r--r--	2021-10-02 02:45
filechunkio.py	2.25 KB	-rw-r--r--	2016-06-19 12:36
filechunkio.pyc	3.17 KB	-rw-r--r--	2021-10-02 02:45
tests.py	3.36 KB	-rw-r--r--	2016-06-19 12:36
tests.pyc	5.88 KB	-rw-r--r--	2021-10-02 02:45

Rename

import io
import os
import tempfile
import unittest

from filechunkio import FileChunkIO
from filechunkio import SEEK_CUR
from filechunkio import SEEK_END

class FileChunkIOTest(unittest.TestCase):

def setUp(self):
        self.tf = tempfile.mkstemp()[1]

with open(self.tf, 'w') as f:
            f.write('123456789%s' % os.linesep)
            f.write('234567891%s' % os.linesep)
            f.write('345678912%s' % os.linesep)
            f.write('456789123%s' % os.linesep)
            f.write('56789')

def tearDown(self):
        if os.path.exists(self.tf):
            os.remove(self.tf)

def test_open_write_raises_valueerror(self):
        self.assertRaises(ValueError, FileChunkIO, self.tf, 'w')

def test_init_autosets_bytes(self):
        with FileChunkIO(self.tf) as c:
            self.assertEqual(c.bytes, 45)

def test_init_autosets_bytes_and_respects_offset(self):
        with FileChunkIO(self.tf, offset=1) as c:
            self.assertEqual(c.bytes, 44)

def test_init_seeks_to_offset(self):
        with FileChunkIO(self.tf, offset=1) as c:
            self.assertEqual(c.tell(), 0)
            self.assertEqual(c.read(1), b'2')

def test_seek_respects_offset(self):
        with FileChunkIO(self.tf, offset=1) as c:
            c.seek(1)
            self.assertEqual(c.read(1), b'3')

def test_seek_cur(self):
        with FileChunkIO(self.tf, offset=20, bytes=10) as c:
            c.seek(5)
            c.seek(-3, SEEK_CUR)
            self.assertEqual(c.tell(), 2)

def test_seek_end(self):
        with FileChunkIO(self.tf, offset=10, bytes=10) as c:
            c.seek(-5, SEEK_END)
            self.assertEqual(c.read(3), b'789')

def test_tell_respects_offset(self):
        with FileChunkIO(self.tf, offset=1) as c:
            self.assertEqual(c.tell(), 0)
            self.assertEqual(c.read(1), b'2')

def test_read_with_minus_one_calls_readall(self):
        with FileChunkIO(self.tf) as c:
            def mocked_readall(*args, **kwargs):
                self._readall_called = True
            c.readall = mocked_readall

c.read(-1)
            self.assertTrue(self._readall_called)

def test_read_respects_offset_and_bytes(self):
        with FileChunkIO(self.tf, offset=1, bytes=3) as c:
            self.assertEqual(c.read(), b'234')

def test_readinto(self):
        with FileChunkIO(self.tf, offset=1, bytes=2) as c:
            n = 3
            b = bytearray(n.__index__())
            c.readinto(b)
            self.assertEqual(b, b'23\x00')

def test_readline(self):
        with FileChunkIO(self.tf, offset=1, bytes=20) as c:
            lines = []
            while True:
                line = c.readline()
                if not line:
                    break
                lines.append(line)
            self.assertEqual(lines, [b'23456789\n', b'234567891\n', b'3'])

def test_readlines(self):
        with FileChunkIO(self.tf, offset=1, bytes=15) as c:
            self.assertEqual(c.readlines(), [b'23456789\n', b'234567'])

def test_next(self):
        with FileChunkIO(self.tf, offset=1, bytes=20) as c:
            lines = []
            while True:
                try:
                    lines.append(next(c))
                except StopIteration:
                    break
            self.assertEqual(lines, [b'23456789\n', b'234567891\n', b'3'])

if __name__ == '__main__':
    unittest.main()