1 files changed, 202 insertions, 0 deletions

diff --git a/src/dtf/WixToolsetTests.Dtf.Compression/MisbehavingStreamContext.cs b/src/dtf/WixToolsetTests.Dtf.Compression/MisbehavingStreamContext.cs
new file mode 100644
index 00000000..2531f3bc
--- /dev/null
+++ b/src/dtf/WixToolsetTests.Dtf.Compression/MisbehavingStreamContext.cs
@@ -0,0 +1,202 @@
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+using System;
+using System.IO;
+using System.Collections.Generic;
+using WixToolset.Dtf.Compression;
+
+namespace WixToolset.Dtf.Test
+{
+    public class MisbehavingStreamContext : ArchiveFileStreamContext
+    {
+        public const string EXCEPTION = "Test exception.";
+
+        private bool throwEx;
+        private bool getName;
+        private bool openArchive;
+        private bool closeArchive;
+        private bool openFile;
+        private bool closeFile;
+        private int closeFileCount;
+
+        public MisbehavingStreamContext(
+            string cabinetFile,
+            string directory,
+            IDictionary<string, string> files,
+            bool throwEx,
+            bool getName,
+            bool openArchive,
+            bool closeArchive,
+            bool openFile,
+            bool closeFile)
+            : base(cabinetFile, directory, files)
+        {
+            this.throwEx = throwEx;
+            this.getName = getName;
+            this.openArchive = openArchive;
+            this.closeArchive = closeArchive;
+            this.openFile = openFile;
+            this.closeFile = closeFile;
+        }
+
+        public override string GetArchiveName(int archiveNumber)
+        {
+            if (!this.getName)
+            {
+                if (throwEx)
+                {
+                    throw new Exception(EXCEPTION);
+                }
+                else
+                {
+                    return null;
+                }
+            }
+            return base.GetArchiveName(archiveNumber);
+        }
+
+        public override Stream OpenArchiveWriteStream(
+            int archiveNumber,
+            string archiveName,
+            bool truncate,
+            CompressionEngine compressionEngine)
+        {
+            if (!this.openArchive)
+            {
+                if (throwEx)
+                {
+                    throw new Exception(EXCEPTION);
+                }
+                else
+                {
+                    return null;
+                }
+            }
+            return base.OpenArchiveWriteStream(
+                archiveNumber, archiveName, truncate, compressionEngine);
+        }
+
+        public override void CloseArchiveWriteStream(
+            int archiveNumber,
+            string archiveName,
+            Stream stream)
+        {
+            if (!this.closeArchive)
+            {
+                if (throwEx)
+                {
+                    this.closeArchive = true;
+                    throw new Exception(EXCEPTION);
+                }
+                return;
+            }
+            base.CloseArchiveWriteStream(archiveNumber, archiveName, stream);
+        }
+
+        public override Stream OpenFileReadStream(
+            string path,
+            out FileAttributes attributes,
+            out DateTime lastWriteTime)
+        {
+            if (!this.openFile)
+            {
+                if (throwEx)
+                {
+                    throw new Exception(EXCEPTION);
+                }
+                else
+                {
+                    attributes = FileAttributes.Normal;
+                    lastWriteTime = DateTime.MinValue;
+                    return null;
+                }
+            }
+            return base.OpenFileReadStream(path, out attributes, out lastWriteTime);
+        }
+
+        public override void CloseFileReadStream(string path, Stream stream)
+        {
+            if (!this.closeFile && ++closeFileCount == 2)
+            {
+                if (throwEx)
+                {
+                    throw new Exception(EXCEPTION);
+                }
+                return;
+            }
+            base.CloseFileReadStream(path, stream);
+        }
+
+        public override Stream OpenArchiveReadStream(
+            int archiveNumber,
+            string archiveName,
+            CompressionEngine compressionEngine)
+        {
+            if (!this.openArchive)
+            {
+                if (throwEx)
+                {
+                    throw new Exception(EXCEPTION);
+                }
+                else
+                {
+                    return null;
+                }
+            }
+            return base.OpenArchiveReadStream(archiveNumber, archiveName, compressionEngine);
+        }
+
+        public override void CloseArchiveReadStream(
+            int archiveNumber,
+            string archiveName,
+            Stream stream)
+        {
+            if (!this.closeArchive)
+            {
+                if (throwEx)
+                {
+                    this.closeArchive = true;
+                    throw new Exception(EXCEPTION);
+                }
+                return;
+            }
+            base.CloseArchiveReadStream(archiveNumber, archiveName, stream);
+        }
+
+        public override Stream OpenFileWriteStream(
+            string path,
+            long fileSize,
+            DateTime lastWriteTime)
+        {
+            if (!this.openFile)
+            {
+                if (throwEx)
+                {
+                    throw new Exception(EXCEPTION);
+                }
+                else
+                {
+                    return null;
+                }
+            }
+            return base.OpenFileWriteStream(path, fileSize, lastWriteTime);
+        }
+
+        public override void CloseFileWriteStream(
+            string path,
+            Stream stream,
+            FileAttributes attributes,
+            DateTime lastWriteTime)
+        {
+            if (!this.closeFile && ++closeFileCount == 2)
+            {
+                if (throwEx)
+                {
+                    throw new Exception(EXCEPTION);
+                }
+                return;
+            }
+            base.CloseFileWriteStream(path, stream, attributes, lastWriteTime);
+        }
+    }
+}