Plug new coverage holes.

Author: sharpjs

PSql.Tests/Tests.Unit/ConnectedCmdletTests.cs

@@ -1,6 +1,7 @@
 // Copyright Subatomix Research Inc.
 // SPDX-License-Identifier: MIT
 
+using Markdig.Extensions.Footers;
 using PSql.Commands;
 
 namespace PSql.Tests;
@@ -26,7 +27,28 @@ public void Dispose_ConnectionNullified()
         cmdlet.Dispose(); // To test multiple disposal
     }
 
-    private class TestCmdletA : ConnectedCmdlet { }
+    [Test]
+    public void AssumeBeginProcessingInvoked_NotInvoked()
+    {
+        var cmdlet = new TestCmdletA();
+
+#if DEBUG
+        Should.Throw<InvalidOperationException>(
+            () => cmdlet.AssumeBeginProcessingInvoked()
+        );
+#else
+    // No-op in release builds
+    cmdlet.AssumeBeginProcessingInvoked();
+#endif
+    }
+
+    private class TestCmdletA : ConnectedCmdlet
+    {
+        public new void AssumeBeginProcessingInvoked()
+        {
+            base.AssumeBeginProcessingInvoked();
+        }
+    }
 
     private class TestCmdletB : ConnectedCmdlet
     {

PSql.Tests/Tests.Unit/InvokeSqlCommandTests.cs

@@ -18,8 +18,23 @@ public void ProcessRecord_NullSql()
         new TestCommand().ProcessRecord();
     }
 
+    [Test]
+    public void StopProcessing()
+    {
+        var command = new TestCommand { Sql = ["--"] };
+
+        command.BeginProcessing();
+        command.StopProcessing();
+
+        Should.Throw<OperationCanceledException>(
+            () => command.ProcessRecord()
+        );
+    }
+
     private class TestCommand : InvokeSqlCommand
     {
-        public new void ProcessRecord() => base.ProcessRecord();
+        public new void BeginProcessing() => base.BeginProcessing();
+        public new void ProcessRecord()   => base.ProcessRecord();
+        public new void StopProcessing()  => base.StopProcessing();
     }
 }