Update TestTempFile.cs

nissl-lab · Jan 3, 2024 · 4e11700 · 4e11700
1 parent f03e78d
commit 4e11700
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/testcases/main/Util/TestTempFile.cs b/testcases/main/Util/TestTempFile.cs
@@ -20,14 +20,18 @@ public void TestCreateTempFile()
 
             string tempDirPath = Path.GetDirectoryName(fileInfo.FullName);
 
-            if (Directory.Exists(tempDirPath))
+            if(Directory.Exists(tempDirPath))
                 Directory.Delete(tempDirPath, true);
-            
+
             Assert.IsFalse(File.Exists(fileInfo.FullName));
             Assert.IsFalse(Directory.Exists(tempDirPath));
 
-            Assert.DoesNotThrow(() => TempFile.CreateTempFile("test2", ".xls"));
+            FileInfo file = null;
+            Assert.DoesNotThrow(() => file = TempFile.CreateTempFile("test2", ".xls"));
             Assert.IsTrue(Directory.Exists(tempDirPath));
+
+            if(file !=null && file.Exists)
+                file.Delete();
         }
 
         [Test]
@@ -40,13 +44,17 @@ public void TestGetTempFilePath()
 
             string tempDirPath = Path.GetDirectoryName(path);
 
-            if (Directory.Exists(tempDirPath))
+            if(Directory.Exists(tempDirPath))
                 Directory.Delete(tempDirPath, true);
 
             Assert.IsFalse(Directory.Exists(tempDirPath));
 
-            Assert.DoesNotThrow(() => TempFile.GetTempFilePath("test", ".xls"));
+            string fileName = "";
+            Assert.DoesNotThrow(() => fileName =  TempFile.GetTempFilePath("test", ".xls"));
             Assert.IsTrue(Directory.Exists(tempDirPath));
+
+            if(File.Exists(fileName))
+                File.Delete(fileName);
         }
     }
 }