def test_getnames(self):
d = 'hello-bfg/'
names = [
d, d + 'build.bfg', d + 'include/', d + 'include/hello.hpp',
d + 'src/', d + 'src/hello.cpp'
]
path = os.path.join(test_data_dir, 'hello-bfg.tar.gz')
with open(path, 'rb') as f, archive.open(f, 'r:gz') as arc:
self.assertEqual(arc.getnames(), names)
path = os.path.join(test_data_dir, 'hello-bfg.zip')
with open(path, 'rb') as f, archive.open(f, 'r:zip') as arc:
self.assertEqual(arc.getnames(), names)
def test_extractall(self):
f = mock.MagicMock()
with mock.patch('tarfile.open') as mtar, \
mock.patch('tarfile.TarFile', type(mtar())): # noqa
with archive.open(f, 'r:tar') as arc:
arc.extractall()
arc.extractall('path')
arc.extractall(members=['dir/', 'file.txt'])
mtar().extractall.assert_has_calls([
mock.call('.', None),
mock.call('path', None),
mock.call('.', ['dir', 'file.txt'])
])
with mock.patch('zipfile.ZipFile') as mzip:
with archive.open(f, 'r:zip') as arc:
arc.extractall()
arc.extractall('path')
arc.extractall(members=['dir/', 'file.txt'])
mzip().extractall.assert_has_calls([
mock.call('.', None),
mock.call('path', None),
mock.call('.', ['dir/', 'file.txt'])
])
def test_extract(self):
f = mock.MagicMock()
with mock.patch('tarfile.open') as mtar, \
mock.patch('tarfile.TarFile', type(mtar())): # noqa
with archive.open(f, 'r:tar') as arc:
arc.extract('file.txt')
arc.extract('file2.txt', 'path')
arc.extract('dir/')
mtar().extract.assert_has_calls([
mock.call('file.txt', '.'),
mock.call('file2.txt', 'path'),
mock.call('dir', '.'),
])
with mock.patch('zipfile.ZipFile') as mzip:
with archive.open(f, 'r:zip') as arc:
arc.extract('file.txt')
arc.extract('file2.txt', 'path')
arc.extract('dir/')
mzip().extract.assert_has_calls([
mock.call('file.txt', '.'),
mock.call('file2.txt', 'path'),
mock.call('dir/', '.'),
])
def test_open_tar(self):
f = mock.MagicMock()
with mock.patch('tarfile.open') as mtar:
archive.open(f, 'r:tar')
mtar.assert_called_once_with(mode='r:tar', fileobj=f)
with mock.patch('tarfile.open') as mtar, \
mock.patch('zipfile.is_zipfile', return_value=False): # noqa
archive.open(f, 'r:*')
mtar.assert_called_once_with(mode='r:*', fileobj=f)
with mock.patch('tarfile.open') as mtar, \
mock.patch('zipfile.is_zipfile', return_value=False): # noqa
archive.open(f, 'r')
mtar.assert_called_once_with(mode='r:*', fileobj=f)
with mock.patch('tarfile.open') as mtar, \
mock.patch('zipfile.is_zipfile', return_value=False): # noqa
archive.open(f)
mtar.assert_called_once_with(mode='r:*', fileobj=f)
def test_open_zip(self):
f = mock.MagicMock()
with mock.patch('zipfile.ZipFile') as mzip:
archive.open(f, 'r:zip')
mzip.assert_called_once_with(f, 'r')
with mock.patch('zipfile.ZipFile') as mzip, \
mock.patch('zipfile.is_zipfile', return_value=True): # noqa
archive.open(f, 'r:*')
mzip.assert_called_once_with(f, 'r')
with mock.patch('zipfile.ZipFile') as mzip, \
mock.patch('zipfile.is_zipfile', return_value=True): # noqa
archive.open(f, 'r')
mzip.assert_called_once_with(f, 'r')
with mock.patch('zipfile.ZipFile') as mzip, \
mock.patch('zipfile.is_zipfile', return_value=True): # noqa
archive.open(f)
mzip.assert_called_once_with(f, 'r')
def test_with(self):
f = mock.MagicMock()
with mock.patch('tarfile.open'):
with archive.open(f, 'r:tar'):
pass