Added NpgsqlDataSource to Postgresql IntegrationContext

JasperFx · Dec 5, 2023 · 7219abe · 7219abe
1 parent 8c3efac
commit 7219abe
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/Weasel.Postgresql.Tests/IntegrationContext.cs b/src/Weasel.Postgresql.Tests/IntegrationContext.cs
@@ -7,7 +7,13 @@ namespace Weasel.Postgresql.Tests;
 
 public abstract class IntegrationContext: IDisposable, IAsyncLifetime
 {
-    protected readonly NpgsqlConnection theConnection = new NpgsqlConnection(ConnectionSource.ConnectionString);
+    protected readonly NpgsqlDataSource theDataSource = NpgsqlDataSource.Create(ConnectionSource.ConnectionString);
+    private NpgsqlConnection? connection;
+
+    protected NpgsqlConnection theConnection
+    {
+        get => (connection ??= theDataSource.CreateConnection());
+    }
 
     protected IntegrationContext(string schemaName)
     {
@@ -24,7 +30,8 @@ protected IntegrationContext(string schemaName)
 
     public void Dispose()
     {
-        theConnection?.Dispose();
+        connection?.Dispose();
+        theDataSource.Dispose();
     }
 
     protected async Task ResetSchema()