nit fix; incorrect zipfile name being passed to the tests.

Azure · Sep 26, 2019 · 2334ce3 · 2334ce3
1 parent 27f519d
commit 2334ce3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/ga/test_exthandlers_download_extension.py b/tests/ga/test_exthandlers_download_extension.py
@@ -86,7 +86,7 @@ def _create_invalid_zip_file(filename):
             file.write("An invalid ZIP file\n")
 
     def _get_extension_package_file(self):
-        return os.path.join(self.agent_dir, os.path.basename(self.pkg.uris[0].uri) + ".zip")
+        return os.path.join(self.agent_dir, self.ext_handler_instance.get_extension_package_zipfile_name())
 
     def _get_extension_command_file(self):
         return os.path.join(self.extension_dir, DownloadExtensionTestCase._extension_command)