Increase coverage

Author: sharpjs

…sts.Integration/InvokeSqlCommandTests.cs …tion/InvokeSqlCommandIntegrationTests.csPSql.Tests/Tests.Integration/InvokeSqlCommandTests.cs renamed to PSql.Tests/Tests.Integration/InvokeSqlCommandIntegrationTests.cs PSql.Tests/Tests.Integration/InvokeSqlCommandTests.cs renamed to PSql.Tests/Tests.Integration/InvokeSqlCommandIntegrationTests.cs

@@ -13,7 +13,7 @@ namespace PSql.Tests.Integration;
 
 [TestFixture]
 [Parallelizable(ParallelScope.All)]
-public class InvokeSqlCommandTests
+public class InvokeSqlCommandIntegrationTests
 {
     [OneTimeSetUp]
     public static void SetUpOnce()
@@ -22,6 +22,54 @@ public static void SetUpOnce()
         _ = Prelude;
     }
 
+    [Test]
+    public void Invoke_Default()
+    {
+        var (output, exception) = Execute(
+        """
+            PSql\Invoke-Sql "PRINT 'a';"
+        """);
+
+        exception.ShouldBeNull();
+
+        output.ShouldHaveSingleItem().ShouldNotBeNull()
+            .BaseObject.ShouldBe(new PSInformation("a"));
+    }
+
+    [Test]
+    public void Invoke_Context()
+    {
+        var (output, exception) = Execute(
+        """
+            PSql\Invoke-Sql -Context $Context "PRINT 'a';"
+        """);
+
+        exception.ShouldBeNull();
+
+        output.ShouldHaveSingleItem().ShouldNotBeNull()
+            .BaseObject.ShouldBe(new PSInformation("a"));
+    }
+
+    [Test]
+    public void Invoke_Connection()
+    {
+        var (output, exception) = Execute(
+        """
+            $Connection = PSql\Connect-Sql
+            try {
+                PSql\Invoke-Sql -Connection $Connection "PRINT 'a';"
+            }
+            finally {
+                PSql\Disconnect-Sql $Connection
+            }
+        """);
+
+        exception.ShouldBeNull();
+
+        output.ShouldHaveSingleItem().ShouldNotBeNull()
+            .BaseObject.ShouldBe(new PSInformation("a"));
+    }
+
     [Test]
     public void ProjectBit_ToClrBoolean()
     {

PSql.Tests/Tests.Unit/ConnectedCmdletTests.cs

@@ -0,0 +1,52 @@
+// Copyright Subatomix Research Inc.
+// SPDX-License-Identifier: MIT
+
+using PSql.Commands;
+
+namespace PSql.Tests;
+
+[TestFixture]
+public class ConnectedCmdletTests
+{
+    [Test]
+    public void Dispose_NotInvoked()
+    {
+        var cmdlet = new TestCmdletA();
+
+        cmdlet.Dispose();
+        cmdlet.Dispose(); // To test multiple disposal
+    }
+
+    [Test]
+    public void Dispose_ConnectionNullified()
+    {
+        var cmdlet = new TestCmdletB();
+
+        cmdlet.Dispose();
+        cmdlet.Dispose(); // To test multiple disposal
+    }
+
+    private class TestCmdletA : ConnectedCmdlet { }
+
+    private class TestCmdletB : ConnectedCmdlet
+    {
+        // This class does several things wrongly in order to test an edge
+        // case: where _ownsConnection is true but Connection is null.
+
+        public TestCmdletB()
+        {
+            // This here to set _ownsConnection to true
+            Context = new();
+            BeginProcessing();
+        }
+
+        public override void Dispose()
+        {
+            // This here to set Connection to null
+            Connection?.Dispose();
+            Connection = null;
+
+            base.Dispose();
+        }
+    }
+}