add support for capturing ScreenShot in Step, getgauge/gauge#632

Runner/IMethodExecutor.cs

@@ -29,5 +29,6 @@ public interface IMethodExecutor
         ProtoExecutionResult ExecuteHooks(string hookType, HooksStrategy strategy, IList<string> applicableTags, ExecutionContext executionContext);
         void ClearCache();
         IEnumerable<string> GetAllPendingMessages();
+        IEnumerable<byte[]> GetAllPendingScreenshots();
     }
 }

Runner/ISandbox.cs

@@ -38,6 +38,7 @@ public interface ISandbox
         List<string> GetAllStepTexts();
         void ClearObjectCache();
         IEnumerable<string> GetAllPendingMessages();
+        IEnumerable<byte[]> GetAllPendingScreenshots();
         void StartExecutionScope(string tag);
         void CloseExectionScope();
         ExecutionResult ExecuteHooks(string hookType, IHooksStrategy strategy, IList<string> applicableTags, ExecutionContext executionContext);

Runner/MethodExecutor.cs

@@ -99,6 +99,11 @@ public IEnumerable<string> GetAllPendingMessages()
             return _sandbox.GetAllPendingMessages();
         }
 
+        public IEnumerable<byte[]> GetAllPendingScreenshots()
+        {
+            return _sandbox.GetAllPendingScreenshots();
+        }
+
         private ByteString TakeScreenshot()
         {
             byte[] screenShotBytes;

Runner/Processors/HookExecutionProcessor.cs

@@ -19,8 +19,6 @@
 using System.Diagnostics;
 using System.Linq;
 using Gauge.CSharp.Core;
-using Gauge.CSharp.Lib;
-using Gauge.CSharp.Runner;
 using Gauge.CSharp.Runner.Strategy;
 using Gauge.Messages;
 

Runner/Processors/StepExecutionEndingProcessor.cs

@@ -18,6 +18,7 @@
 using System.Collections.Generic;
 using System.Linq;
 using Gauge.Messages;
+using Google.Protobuf;
 
 namespace Gauge.CSharp.Runner.Processors
 {
@@ -34,7 +35,9 @@ protected override ProtoExecutionResult ExecuteHooks(Message request)
         {
             var protoExecutionResult = base.ExecuteHooks(request);
             var allPendingMessages = MethodExecutor.GetAllPendingMessages().Where(m => m != null);
+            var allPendingScreenshots = MethodExecutor.GetAllPendingScreenshots().Select(x => ByteString.CopyFrom(x));
             protoExecutionResult.Message.AddRange(allPendingMessages);
+            protoExecutionResult.ScreenShot.AddRange(allPendingScreenshots);
             return protoExecutionResult;
         }
 

Runner/Processors/StepExecutionStartingProcessor.cs

@@ -39,6 +39,7 @@ protected override ProtoExecutionResult ExecuteHooks(Message request)
         {
             // Just need to clear the messages, but Gauge.CSharp.Lib v0.5.2 does not have MessageCollector.Clear()
             MethodExecutor.GetAllPendingMessages();
+            MethodExecutor.GetAllPendingScreenshots();
             return base.ExecuteHooks(request);
         }
 

Runner/Sandbox.cs

@@ -185,6 +185,14 @@ public IEnumerable<string> GetAllPendingMessages()
             return targetMethod.Invoke(null, null) as IEnumerable<string>;
         }
 
+        public IEnumerable<byte[]> GetAllPendingScreenshots()
+        {
+            var targetMessageCollectorType = _libAssembly.GetType("Gauge.CSharp.Lib.ScreenshotCollector");
+            var targetMethod = targetMessageCollectorType.GetMethod("GetAllPendingScreenshots",
+                BindingFlags.Static | BindingFlags.Public);
+            return targetMethod.Invoke(null, null) as IEnumerable<byte[]>;
+        }
+
         public void StartExecutionScope(string tag)
         {
             _classInstanceManager.StartScope(tag);

appveyor.yml

@@ -1,4 +1,5 @@
 version: 1.0.{build}
+os: Visual Studio 2017
 environment:
   GAUGE_PARALLEL: false
   GAUGE_TELEMETRY_ENABLED: false