def setUp(self): super(DownloadAndExtractFileTest, self).setUp() self.req_url = "{}{}".format(self.testserver, self.url) self.file_content = b"file content" create_dir(self.temp_directory) self.download_location = join(self.temp_directory, "dummy_file")
def setUp(self): super(DownloadAndExtractZipFileTest, self).setUp() self.zip_name = "test" self.req_url = "{}/{}".format(self.testserver, self.zip_name) self.extract_location = join(self.BASE_TEMP_DIR, "test-dir") self.download_location = join(self.extract_location, "{}.zip".format(self.zip_name)) create_dir(self.extract_location) self.file_name = "test_file.txt" self.file_content = b"file_content" self.zip_file = BytesIO() with zipfile.ZipFile(self.zip_file, mode="w", compression=zipfile.ZIP_DEFLATED) as zipper: zipper.writestr(self.file_name, self.file_content)
def test_create_dir(self): create_dir(self.temp_directory) self.assertTrue(os.path.isdir(self.temp_directory)) shutil.rmtree(self.temp_directory)