def test_deferred_close_by_archive(self):
""" Test archive deferred close without a stream. """
f = file(ZIPPATH, mode='w')
z = ZipFile(f, 'w')
o = z.writestream(FILENAMES[0])
z.close()
self.assertIsNotNone(z._a)
self.assertIsNotNone(z._stream)
o.write('testdata')
o.close()
self.assertIsNone(z._a)
self.assertIsNone(z._stream)
z.close()
def test_deferred_close_by_archive(self):
""" Test archive deferred close without a stream. """
f = file(ZIPPATH, mode='w')
z = ZipFile(f, 'w')
o = z.writestream(FILENAMES[0])
z.close()
self.assertIsNotNone(z._a)
self.assertIsNotNone(z._stream)
o.write('testdata')
o.close()
self.assertIsNone(z._a)
self.assertIsNone(z._stream)
z.close()
def test_deferred_close_by_archive(self):
""" Test archive deferred close without a stream. """
z = ZipFile(self.f, 'w')
o = z.writestream(FILENAMES[0])
z.close()
self.assertIsNotNone(z._a)
self.assertIsNotNone(z._stream)
if PY3:
o.write('testdata')
else:
o.write(unicode('testdata'))
o.close()
self.assertIsNone(z._a)
self.assertIsNone(z._stream)
z.close()
def test_writestream_unbuffered(self):
f = file(ZIPPATH, mode='w')
z = ZipFile(f, 'w')
for fname in FILENAMES:
full_path = os.path.join(TMPDIR, fname)
i = file(full_path)
o = z.writestream(fname, os.path.getsize(full_path))
while True:
data = i.read(1)
if not data:
break
o.write(data)
o.close()
i.close()
z.close()
def test_writestream_unbuffered(self):
f = file(ZIPPATH, mode='w')
z = ZipFile(f, 'w')
for fname in FILENAMES:
full_path = os.path.join(TMPDIR, fname)
i = file(full_path)
o = z.writestream(fname, os.path.getsize(full_path))
while True:
data = i.read(1)
if not data:
break
o.write(data)
o.close()
i.close()
z.close()
def test_writestream_unbuffered(self):
z = ZipFile(self.f, 'w')
for fname in FILENAMES:
full_path = os.path.join(TMPDIR, fname)
i = open(full_path)
o = z.writestream(fname, os.path.getsize(full_path))
while True:
data = i.read(1)
if not data:
break
if PY3:
o.write(data)
else:
o.write(unicode(data))
o.close()
i.close()
z.close()
