Fix tests

DevTeam · Jul 19, 2024 · 9830523 · 9830523
1 parent 912f1af
commit 9830523
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/CSharpInteractive.Tests/UsageScenarios/BaseScenario.cs b/CSharpInteractive.Tests/UsageScenarios/BaseScenario.cs
@@ -15,13 +15,16 @@ public class BaseScenario : IHost, IDisposable
 
     public BaseScenario()
     {
+        Composition.Shared.Root.TestEnvironment.IsTesting = true;
         Composition.Shared.Root.TestEnvironment.ExitCode = default;
         _tempDir = Path.Combine(Path.GetTempPath(), Guid.NewGuid().ToString()[..4]);
         Directory.CreateDirectory(_tempDir);
         _prevCurDir = Environment.CurrentDirectory;
         Environment.CurrentDirectory = _tempDir;
     }
 
+    public int ExpectedExitCode { get; set; }
+
     // ReSharper disable once MemberCanBeProtected.Global
     public IHost Host => this;
 
@@ -82,6 +85,9 @@ void IDisposable.Dispose()
         }
 
         Environment.CurrentDirectory = _prevCurDir;
-        Composition.Shared.Root.TestEnvironment.ExitCode.ShouldBe(0);
+        if (Composition.Shared.Root.TestEnvironment.ExitCode is { } exitCode)
+        {
+            exitCode.ShouldBe(ExpectedExitCode);
+        }
     }
 }
diff --git a/CSharpInteractive.Tests/UsageScenarios/CommandLineWithTimeoutScenario.cs b/CSharpInteractive.Tests/UsageScenarios/CommandLineWithTimeoutScenario.cs
@@ -14,6 +14,7 @@ public void Run()
     {
         Skip.IfNot(Environment.OSVersion.Platform == PlatformID.Win32NT);
         Skip.IfNot(string.IsNullOrWhiteSpace(Environment.GetEnvironmentVariable("TEAMCITY_VERSION")));
+        ExpectedExitCode = 1;
 
         // $visible=true
         // $tag=10 Command Line API