Merge pull request #46 from mpekurny/master

Adding method to allow screenshots from the plugin with the datastores setup
getgauge · Sep 23, 2024 · 6d9f3a9 · 6d9f3a9
2 parents 1006acb + 86f97f7
commit 6d9f3a9
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 3 deletions.
diff --git a/Gauge.CSharp.Lib/Gauge.CSharp.Lib.csproj b/Gauge.CSharp.Lib/Gauge.CSharp.Lib.csproj
@@ -4,9 +4,9 @@
 		<TargetFrameworks>net6.0;net7.0;net8.0</TargetFrameworks>
 		<Description>CSharp bindings for Gauge. Write CSharp step implementation for Gauge specs. https://gauge.org</Description>
 		<ImplicitUsings>enable</ImplicitUsings>
-		<Version>0.11.1</Version>
-		<AssemblyVersion>0.11.1.0</AssemblyVersion>
-		<FileVersion>0.11.1.0</FileVersion>
+		<Version>0.11.2</Version>
+		<AssemblyVersion>0.11.2.0</AssemblyVersion>
+		<FileVersion>0.11.2.0</FileVersion>
 		<Authors>getgauge</Authors>
 		<Company>ThoughtWorks Inc.</Company>
 		<Copyright>Copyright © ThoughtWorks Inc. 2018</Copyright>

diff --git a/Gauge.CSharp.Lib/GaugeScreenshots.cs b/Gauge.CSharp.Lib/GaugeScreenshots.cs
@@ -20,4 +20,27 @@ public static void Capture()
     {
         ScreenshotFiles.Add(screenshotWriter.TakeScreenShot());
     }
+
+    public static void CaptureByStream(int streamId)
+    {
+        SetDataStores(streamId);
+        ScreenshotFiles.Add(screenshotWriter.TakeScreenShot());
+    }
+
+    private static void SetDataStores(int streamId)
+    {
+        var dataStore = DataStoreFactory.GetDataStoresByStream(streamId);
+        lock (SuiteDataStore.Store)
+        {
+            SuiteDataStore.Store.Value = DataStoreFactory.SuiteDataStore;
+        }
+        lock (SpecDataStore.Store)
+        {
+            SpecDataStore.Store.Value = dataStore.GetValueOrDefault(DataStoreType.Spec, null);
+        }
+        lock (ScenarioDataStore.Store)
+        {
+            ScenarioDataStore.Store.Value = dataStore.GetValueOrDefault(DataStoreType.Scenario, null);
+        }
+    }
 }