Implement support for IExtensionCommandLine and IPreprocessorExtension

1 files changed, 9 insertions, 2 deletions

diff --git a/src/test/Example.Extension/ExampleExtensionFactory.cs b/src/test/Example.Extension/ExampleExtensionFactory.cs
index 9539ee85..b91d06e9 100644
--- a/src/test/Example.Extension/ExampleExtensionFactory.cs
+++ b/src/test/Example.Extension/ExampleExtensionFactory.cs
@@ -7,11 +7,18 @@ namespace Example.Extension
 
     public class ExampleExtensionFactory : IExtensionFactory
     {
+        private ExamplePreprocessorExtensionAndCommandLine preprocessorExtension;
+
         public bool TryCreateExtension(Type extensionType, out object extension)
         {
-            if (extensionType == typeof(IPreprocessorExtension))
+            if (extensionType == typeof(IExtensionCommandLine) || extensionType == typeof(IPreprocessorExtension))
             {
-                extension = new ExamplePreprocessorExtension();
+                if (preprocessorExtension == null)
+                {
+                    preprocessorExtension = new ExamplePreprocessorExtensionAndCommandLine();
+                }
+
+                extension = preprocessorExtension;
             }
             else if (extensionType == typeof(ICompilerExtension))
             {