def test_validate_archive_dirs_fails(self):
work_dir = os.path.join(self.base_path, 'download_test_data')
with ChangedWorkingDirectory(work_dir):
symlink = 'link'
os.symlink(os.path.join(self.base_path, 'subfolder'), symlink)
self.assertFalse(upload_to_google_storage.validate_archive_dirs([symlink]))
self.assertFalse(upload_to_google_storage.validate_archive_dirs(['foobar']))
def test_create_archive(self):
work_dir = os.path.join(self.base_path, 'download_test_data')
with ChangedWorkingDirectory(work_dir):
dirname = 'subfolder'
dirs = [dirname]
tar_gz_file = '%s.tar.gz' % dirname
self.assertTrue(upload_to_google_storage.validate_archive_dirs(dirs))
upload_to_google_storage.create_archives(dirs)
self.assertTrue(os.path.exists(tar_gz_file))
with tarfile.open(tar_gz_file, 'r:gz') as tar:
content = map(lambda x: x.name, tar.getmembers())
self.assertTrue(dirname in content)
self.assertTrue(os.path.join(dirname, 'subfolder_text.txt') in content)
self.assertTrue(
os.path.join(dirname, 'subfolder_text.txt.sha1') in content)
