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)