FilesExpand file tree

 docs

/

test_zipstream.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

59 lines (43 loc) · 1.42 KB

 docs

/

test_zipstream.py

Top

File metadata and controls

• Code
• Blame

59 lines (43 loc) · 1.42 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

# -*- coding: utf-8 -*-

from __future__ import unicode_literals, print_function

import os

import tempfile

import unittest

import zipstream

import zipfile

class ZipInfoTestCase(unittest.TestCase):

pass

class ZipStreamTestCase(unittest.TestCase):

def setUp(self):

self.fileobjs = [

tempfile.NamedTemporaryFile(delete=False, suffix='.txt'),

tempfile.NamedTemporaryFile(delete=False, suffix='.py'),

]

def tearDown(self):

for fileobj in self.fileobjs:

fileobj.close()

os.remove(fileobj.name)

def test_init_no_args(self):

zipstream.ZipFile()

def test_init_mode(self):

try:

zipstream.ZipFile(mode='w')

except Exception as err:

self.fail(err)

for mode in ['wb', 'r', 'rb', 'a', 'ab']:

self.assertRaises(Exception, zipstream.ZipFile, mode=mode)

for mode in ['wb', 'r', 'rb', 'a', 'ab']:

self.assertRaises(Exception, zipstream.ZipFile, mode=mode + '+')

def test_write_file(self):

z = zipstream.ZipFile(mode='w')

for fileobj in self.fileobjs:

z.write(fileobj.name)

f = tempfile.NamedTemporaryFile(suffix='zip', delete=False)

for chunk in z:

f.write(chunk)

f.close()

z2 = zipfile.ZipFile(f.name, 'r')

z2.testzip()

os.remove(f.name)

if __name__ == '__main__':

unittest.main()