1 files changed, 42 insertions, 0 deletions

diff --git a/src/dtf/WixToolsetTests.Dtf.Compression/OptionStreamContext.cs b/src/dtf/WixToolsetTests.Dtf.Compression/OptionStreamContext.cs
new file mode 100644
index 00000000..98354d97
--- /dev/null
+++ b/src/dtf/WixToolsetTests.Dtf.Compression/OptionStreamContext.cs
@@ -0,0 +1,42 @@
+// 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.Collections.Generic;
+using WixToolset.Dtf.Compression;
+
+namespace WixToolset.Dtf.Test
+{
+    public class OptionStreamContext : ArchiveFileStreamContext
+    {
+        private PackOptionHandler packOptionHandler;
+
+        public OptionStreamContext(IList<string> archiveFiles, string directory, IDictionary<string, string> files)
+            : base(archiveFiles, directory, files)
+        {
+        }
+
+        public delegate object PackOptionHandler(string optionName, object[] parameters);
+
+        public PackOptionHandler OptionHandler
+        {
+            get
+            {
+                return this.packOptionHandler;
+            }
+            set
+            {
+                this.packOptionHandler = value;
+            }
+        }
+
+        public override object GetOption(string optionName, object[] parameters)
+        {
+            if (this.OptionHandler == null)
+            {
+                return null;
+            }
+
+            return this.OptionHandler(optionName, parameters);
+        }
+    }
+}