test: 增加单元测试

Author: ArgoZhang

test/UnitTest/Services/ZipArchiveServiceTest.cs

@@ -3,6 +3,8 @@
 // See the LICENSE file in the project root for more information.
 // Maintainer: Argo Zhang(argo@live.ca) Website: https://www.blazor.zone
 
+using System.IO.Compression;
+
 namespace UnitTest.Services;
 
 public class ZipArchiveServiceTest : BootstrapBlazorTestBase
@@ -22,10 +24,15 @@ public async Task Archive_Ok()
             using var fs = File.OpenWrite(f);
             fs.WriteByte(65);
         });
-        var stream = await archService.ArchiveAsync(files);
+        var items = files.Select(i => new ArchiveEntry()
+        {
+            SourceFileName = i,
+            EntryName = Path.GetFileName(i)
+        });
+        var stream = await archService.ArchiveAsync(items);
         Assert.NotNull(stream);
 
-        stream = await archService.ArchiveAsync(files, new ArchiveOptions()
+        stream = await archService.ArchiveAsync(items, new ArchiveOptions()
         {
             CompressionLevel = System.IO.Compression.CompressionLevel.Optimal,
             Encoding = System.Text.Encoding.UTF8,
@@ -35,7 +42,7 @@ public async Task Archive_Ok()
         Assert.NotNull(stream);
 
         var archiveFile = Path.Combine(root, "test.zip");
-        await archService.ArchiveAsync(archiveFile, files);
+        await archService.ArchiveAsync(archiveFile, items);
         Assert.True(File.Exists(archiveFile));
 
         // GetEntry
@@ -48,7 +55,7 @@ public async Task Archive_Ok()
         {
             Directory.Delete(destFolder, true);
         }
-        archService.ExtractToDirectory(archiveFile, destFolder);
+        await archService.ExtractToDirectoryAsync(archiveFile, destFolder);
         Assert.True(Directory.Exists(destFolder));
 
         // 删除文件夹
@@ -70,32 +77,83 @@ public async Task Archive_Ok()
         {
             File.Delete(destFile);
         }
-        await archService.ArchiveDirectory(destFile, destFolder, includeBaseDirectory: true);
+        await archService.ArchiveDirectoryAsync(destFile, destFolder, includeBaseDirectory: true);
         Assert.True(File.Exists(destFile));
         File.Delete(destFile);
 
-        await Assert.ThrowsAsync<ArgumentNullException>(() => archService.ArchiveDirectory(null!, destFolder, includeBaseDirectory: true));
+        await Assert.ThrowsAsync<ArgumentNullException>(() => archService.ArchiveDirectoryAsync(null!, destFolder, includeBaseDirectory: true));
+    }
 
-        // 测试压缩多个文件夹
-        var entries = new List<string>()
+    [Fact]
+    public async Task ZipArchive_Ok()
+    {
+        var fileName = Path.Combine(AppContext.BaseDirectory, "test", "3.zip");
+        if (File.Exists(fileName))
         {
-            destFolder,
-            tempFolder,
-        };
-        entries.AddRange(files);
+            File.Delete(fileName);
+        }
 
-        destFile = Path.Combine(root, "folder.zip");
-        if (File.Exists(destFile))
+        using var fs = File.OpenWrite(fileName);
+        using var zip = new ZipArchive(fs, ZipArchiveMode.Create);
+
+        var item = Path.Combine(AppContext.BaseDirectory, "test", "1.txt");
+        zip.CreateEntry("text/");
+        await zip.CreateEntryFromFileAsync(item, "text/1.txt");
+    }
+
+    [Fact]
+    public async Task ArchiveAsync_Ok()
+    {
+        var fileName = Path.Combine(AppContext.BaseDirectory, "archive_test", "test.zip");
+        if (File.Exists(fileName))
         {
-            File.Delete(destFile);
+            File.Delete(fileName);
         }
-        Assert.False(File.Exists(destFile));
-        var subFolder = Path.Combine(tempFolder, "sub");
-        if (!Directory.Exists(subFolder))
+
+        var root = AppContext.BaseDirectory;
+        var files = new string[]
         {
-            Directory.CreateDirectory(subFolder);
-        }
-        await archService.ArchiveDirectory(destFile, entries);
-        Assert.True(File.Exists(destFile));
+            Path.Combine(root, "archive_test", "test1", "1.txt"),
+            Path.Combine(root, "archive_test", "test2", "2.txt")
+        };
+        files.ToList().ForEach(f =>
+        {
+            var folder = Path.GetDirectoryName(f);
+            if (!string.IsNullOrEmpty(folder) && !Directory.Exists(folder))
+            {
+                Directory.CreateDirectory(folder);
+            }
+            using var fs = File.OpenWrite(f);
+            fs.WriteByte(65);
+        });
+
+        var archService = Context.Services.GetRequiredService<IZipArchiveService>();
+        await archService.ArchiveAsync(fileName, new List<ArchiveEntry>()
+        {
+            new ArchiveEntry()
+            {
+                SourceFileName = files[0],
+                EntryName = "test1/test.log"
+            },
+            new ArchiveEntry()
+            {
+                SourceFileName = files[1],
+                EntryName = "test2/test.log",
+                CompressionLevel = CompressionLevel.Optimal
+            },
+            new ArchiveEntry()
+            {
+                SourceFileName = Path.Combine(AppContext.BaseDirectory, "archive_test", "test1"),
+                EntryName = "test1",
+            },
+            new ArchiveEntry()
+            {
+                SourceFileName = Path.Combine(AppContext.BaseDirectory, "archive_test", "test1"),
+                EntryName = "test2",
+                CompressionLevel = CompressionLevel.Optimal
+            }
+        });
+
+        Assert.True(File.Exists(fileName));
     }
 }