BaseTools: use set instead of list for a variable to be used with in

[mirror_edk2.git] / BaseTools / Source / Python / UPT / TestInstall.py

diff --git a/BaseTools/Source/Python/UPT/TestInstall.py b/BaseTools/Source/Python/UPT/TestInstall.py
index dae4415026519e5488ff8f30d18c36d598095fd2..899cae56aa8790035e75726f4ecaabb08e936d10 100644 (file)
--- a/BaseTools/Source/Python/UPT/TestInstall.py
+++ b/BaseTools/Source/Python/UPT/TestInstall.py
@@ -1,7 +1,7 @@
 # # @file\r
 # Test Install distribution package\r
 #\r
-# Copyright (c) 2016, Intel Corporation. All rights reserved.<BR>\r
+# Copyright (c) 2016 - 2017, Intel Corporation. All rights reserved.<BR>\r
 #\r
 # This program and the accompanying materials are licensed and made available\r
 # under the terms and conditions of the BSD License which accompanies this\r
@@ -90,9 +90,9 @@ def Main(Options=None):
             DistFile.Close()\r
         if ContentZipFile:\r
             ContentZipFile.Close()\r
-        if GlobalData.gUNPACK_DIR:\r
-            shutil.rmtree(GlobalData.gUNPACK_DIR)\r
-            GlobalData.gUNPACK_DIR = None\r
+        for TempDir in GlobalData.gUNPACK_DIR:\r
+            shutil.rmtree(TempDir)\r
+        GlobalData.gUNPACK_DIR = []\r
         Logger.Quiet(ST.MSG_REMOVE_TEMP_FILE_DONE)\r
     if ReturnCode == 0:\r
         Logger.Quiet(ST.MSG_FINISH)\r