sfneal / dirutility

tests/test_archivefile.py

import os
import unittest
from time import sleep

from dirutility.open.open import tardir


class TestArchive(unittest.TestCase):

This code seems to be duplicated in your project.

    def test_tardir(self):
        tempfile = '/Users/Shared/kasdkfasdf.txt'
        with open(tempfile, 'w') as wfd:
            wfd.write(tempfile)
        with open(tempfile, 'r') as rfd1, open(tempfile, 'r') as rfd2:
            paths = {
                'abc': 'abc abc abc',
                'kbs': {
                    'haha': {},
                    'yoyo': {
                        'haha': rfd1
                    }
                },
                'abcd': {
                    'kbs': rfd2
                },
            }
            with tardir('/Users/Shared/haha.tar.gz', **paths) as tarfile:
                self.assertTrue(os.path.exists(tarfile.filepath))
            self.assertFalse(os.path.exists(tarfile.filepath))
        with open(tempfile, 'r') as rfd1, open(tempfile, 'r') as rfd2:
            paths = {
                'abc': 'abc abc abc',
                'kbs': {
                    'haha': {},
                    'yoyo': {
                        'haha': rfd1
                    }
                },
                'abcd': {
                    'kbs': rfd2
                },
            }
            with tardir('/Users/Shared/haha.zip', **paths, mode='w:zip', withdir=True) as tarfile:
                self.assertTrue(os.path.exists(tarfile.filepath))
                sleep(10)
            self.assertFalse(os.path.exists(tarfile.filepath))
        os.remove(tempfile)


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