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)