ZipTest.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using Ionic.Zip;
using Microsoft.Build.Utilities;
using NUnit.Framework;

namespace MSBuild.Community.Tasks.Tests
{
    /// <summary>
    /// Summary description for ZipTest
    /// </summary>
    [TestFixture]
    public class ZipTest
    {
        public const string ZIP_FILE_NAME = @"MSBuild.Community.Tasks.zip";
        public const string ZIP_WITH_FOLDERS_FILE_NAME = @"MSBuild.Community.Tasks.WithFolders.zip";
        public const string ZIP_WITH_RELATIVE_FILE_NAME = @"MSBuild.Community.Tasks.WithReltive.zip";
	    public const string ZIP_128KB_FILE_NAME = @"MSBuild.Community.Tasks.128KB.zip";

        [Test(Description = "Zip files into a zip archive")]
        public void ZipExecute()
        {
            var task = new Zip();
            task.BuildEngine = new MockBuild();

            string testDir = TaskUtility.TestDirectory;
            string prjRootPath = TaskUtility.GetProjectRootDirectory(true);

            string workingDir = Path.Combine(prjRootPath, @"Source\MSBuild.Community.Tasks.Tests");
            string[] files = Directory.GetFiles(workingDir, "*.*", SearchOption.TopDirectoryOnly);

            TaskItem[] items = TaskUtility.StringArrayToItemArray(files);

            task.Files = items;
            task.WorkingDirectory = workingDir;
            task.ZipFileName = Path.Combine(testDir, ZIP_FILE_NAME);

            if (File.Exists(task.ZipFileName))
                File.Delete(task.ZipFileName);

            Assert.IsTrue(task.Execute(), "Execute Failed");
            Assert.IsTrue(File.Exists(task.ZipFileName), "Zip file not found");
        }

        [Test(Description = "Zip files and empty folders into a zip archive")]
        public void ZipExecuteWithEmptyFolders()
        {
            var task = new Zip();
            task.BuildEngine = new MockBuild();

            string testDir = TaskUtility.TestDirectory;
            string prjRootPath = TaskUtility.GetProjectRootDirectory(true);

            string workingDir = Path.Combine(prjRootPath, @"Source\MSBuild.Community.Tasks.Tests");
            string[] files = Directory.GetFiles(workingDir, "*.*", SearchOption.TopDirectoryOnly);
            string[] directories = Directory.GetDirectories(workingDir, "*.*", SearchOption.TopDirectoryOnly);
            string[] filesFromXmlDirectory = new string[0]; //Directory.GetFiles(workingDir + "\\" + "Xml", "*.*", SearchOption.TopDirectoryOnly);

            var filesAndDirectories = new string[files.Length + directories.Length + filesFromXmlDirectory.Length];


            files.CopyTo(filesAndDirectories, 0);
            directories.CopyTo(filesAndDirectories, files.Length + filesFromXmlDirectory.Length);
            filesFromXmlDirectory.CopyTo(filesAndDirectories, files.Length);

            TaskItem[] items = TaskUtility.StringArrayToItemArray(filesAndDirectories);

            task.Files = items;
            task.WorkingDirectory = workingDir;
            task.ZipFileName = Path.Combine(testDir, ZIP_WITH_FOLDERS_FILE_NAME);

            if (File.Exists(task.ZipFileName))
                File.Delete(task.ZipFileName);

            Assert.IsTrue(task.Execute(), "Execute Failed");
            Assert.IsTrue(File.Exists(task.ZipFileName), "Zip file not found");
        }

        [Test]
        public void ZipNoRoot()
        {
            var task = new Zip();
            task.BuildEngine = new MockBuild();

            string testDir = TaskUtility.TestDirectory;
            string prjRootPath = TaskUtility.GetProjectRootDirectory(true);

            string workingDir = Path.Combine(prjRootPath, @"Source\MSBuild.Community.Tasks.Tests");
            string[] files = Directory.GetFiles(workingDir, "*.*", SearchOption.AllDirectories);

            TaskItem[] items = TaskUtility.StringArrayToItemArray(files);

            task.Files = items;
            task.ZipFileName = Path.Combine(testDir, ZIP_FILE_NAME);

            if (File.Exists(task.ZipFileName))
                File.Delete(task.ZipFileName);

            Assert.IsTrue(task.Execute(), "Execute Failed");
            Assert.IsTrue(File.Exists(task.ZipFileName), "Zip file not found");
        }

        [Test(Description = "Zip files with relative path")]
        public void ZipRelativeExecute()
        {
            var task = new Zip();
            task.BuildEngine = new MockBuild();

            string testDir = TaskUtility.TestDirectory;
            string prjRootPath = TaskUtility.GetProjectRootDirectory(true);

            string workingDir = Path.Combine(prjRootPath, @"Source\MSBuild.Community.Tasks.Tests");
            List<string> files = Directory.GetFiles(workingDir, "*.*", SearchOption.TopDirectoryOnly).ToList();
            files.Add(Path.Combine(workingDir, "..\\..\\readme.md"));

            TaskItem[] items = TaskUtility.StringArrayToItemArray(files.ToArray());

            task.Files = items;
            task.WorkingDirectory = workingDir;
            task.ZipFileName = Path.Combine(testDir, ZIP_WITH_RELATIVE_FILE_NAME);

            if (File.Exists(task.ZipFileName))
                File.Delete(task.ZipFileName);
            
            Assert.IsTrue(task.Execute(), "Execute Failed");                        
            Assert.IsTrue(ZipFile.Read(task.ZipFileName).ContainsEntry("readme.md"), "The zip doesnt contains the readme.md file");            
            Assert.IsTrue(File.Exists(task.ZipFileName), "Zip file not found");            
        }

        [Test(Description = "Zip 9 * 128kb with parallel compression")]
        public void ZipWithParallelCompression()
        {
            var task = new Zip();
            task.BuildEngine = new MockBuild();

            string testDir = TaskUtility.TestDirectory;

            string testFile = Path.Combine(testDir, "zip-128kb-test-1.dat");
            const int numberOfBytes = 1024*128*9;
            if (File.Exists(testFile)) File.Delete(testFile);
            CreateTestFile(numberOfBytes, testFile);

            task.Files = new[] {new TaskItem(testFile)};
            task.WorkingDirectory = testDir;
            task.ParallelCompression = true;
            task.ZipFileName = Path.Combine(testDir, ZIP_128KB_FILE_NAME);

            if (File.Exists(task.ZipFileName)) File.Delete(task.ZipFileName);

            // First zip up the file
            Assert.IsTrue(task.Execute(), "Execute Failed");
            Assert.IsTrue(File.Exists(task.ZipFileName), "Zip file not found");

            // Then delete the original, and try to unzip the file
            File.Delete(testFile);
            Unzip unzipTask = new Unzip();
            unzipTask.BuildEngine = new MockBuild();
            unzipTask.ZipFileName = task.ZipFileName;
            unzipTask.TargetDirectory = testDir;
            Assert.IsTrue(unzipTask.Execute());
        }


        [Test(Description = "Zip file update instead of overwrite")]
        public void ZipFileUpdate()
        {
            var task = new Zip();
            task.BuildEngine = new MockBuild();

            string testDir = TaskUtility.TestDirectory;
            const string fileName = "zip-128kb-test-1.dat";
            string testFile = Path.Combine(testDir, fileName);
            const int numberOfBytes = 1024 * 128 * 9;
            if (File.Exists(testFile))
                File.Delete(testFile);
            CreateTestFile(numberOfBytes, testFile);

            task.Files = new[] { new TaskItem(testFile) };
            task.WorkingDirectory = testDir;
            task.ZipFileName = Path.Combine(testDir, ZIP_FILE_NAME);
            task.Overwrite = false;

            if (File.Exists(task.ZipFileName))
                File.Delete(task.ZipFileName);

            // First zip up the file
            Assert.IsTrue(task.Execute(), "Execute Failed");
            Assert.IsTrue(File.Exists(task.ZipFileName), "Zip file not found");

            // Then delete the original, and try to unzip the file
            File.Delete(testFile);
            Unzip unzipTask = new Unzip();
            unzipTask.BuildEngine = new MockBuild();
            unzipTask.ZipFileName = task.ZipFileName;
            unzipTask.TargetDirectory = testDir;
            Assert.IsTrue(unzipTask.Execute());

            long length = new FileInfo(Path.Combine(testDir, fileName)).Length;
            Assert.AreEqual(numberOfBytes, length, "File length does not match");

            if (File.Exists(testFile))
                File.Delete(testFile);
            CreateTestFile(numberOfBytes * 2, testFile);

            // Update the zip file
            Assert.IsTrue(task.Execute(), "Execute Failed");
            Assert.IsTrue(File.Exists(task.ZipFileName), "Zip file not found");

            // Then delete the original, and try to unzip the file
            File.Delete(testFile);
            unzipTask = new Unzip();
            unzipTask.BuildEngine = new MockBuild();
            unzipTask.ZipFileName = task.ZipFileName;
            unzipTask.TargetDirectory = testDir;
            Assert.IsTrue(unzipTask.Execute());

            length = new FileInfo(Path.Combine(testDir, fileName)).Length;
            Assert.AreEqual(numberOfBytes * 2, length, "File length does not match");

            if (File.Exists(testFile))
                File.Delete(testFile);
            CreateTestFile(numberOfBytes, testFile);

            // Update the zip file again
            Assert.IsTrue(task.Execute(), "Execute Failed");
            Assert.IsTrue(File.Exists(task.ZipFileName), "Zip file not found");

            // Then delete the original, and try to unzip the file
            File.Delete(testFile);
            unzipTask = new Unzip();
            unzipTask.BuildEngine = new MockBuild();
            unzipTask.ZipFileName = task.ZipFileName;
            unzipTask.TargetDirectory = testDir;
            Assert.IsTrue(unzipTask.Execute());

            length = new FileInfo(Path.Combine(testDir, fileName)).Length;
            Assert.AreEqual(numberOfBytes, length, "File length does not match");
        }

        public void CreateTestFile(int fileSize, string filePath)
        {
            byte[] testFileBytes = new byte[fileSize];
            new Random().NextBytes(testFileBytes);
            using (FileStream filestream = File.Create(filePath))
            {
                filestream.Write(testFileBytes, 0, fileSize);
            }
        }
    }
}