Updated migrator

malovicg · Apr 26, 2023 · fe106b8 · fe106b8
1 parent dcddd24
commit fe106b8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Tactical.DDD.EventSourcing.Integration.Tests/IntegrationTest.cs b/Tactical.DDD.EventSourcing.Integration.Tests/IntegrationTest.cs
@@ -16,7 +16,7 @@ protected IntegrationTest()
         {
             var conn = NpgsqlDataSource.Create(CreateDb());
 
-            new EventStoreMigrator(conn.CreateConnection()).EnsureEventStoreCreated();
+            new EventStoreMigrator(conn).EnsureEventStoreCreated();
 
             EventStore = new EventStore(conn);
             OffsetTracker = new PostgresOffsetTracker(conn);

diff --git a/Tactical.DDD.EventSourcing.Postgres/EventStoreMigrator.cs b/Tactical.DDD.EventSourcing.Postgres/EventStoreMigrator.cs
@@ -7,15 +7,15 @@ namespace Tactical.DDD.EventSourcing.Postgres
 {
     public sealed class EventStoreMigrator
     {
-        private readonly NpgsqlConnection _conn;
+        private readonly NpgsqlDataSource _conn;
 
-        public EventStoreMigrator(NpgsqlConnection conn)
+        public EventStoreMigrator(NpgsqlDataSource conn)
         {
             _conn = conn;
         }
 
         public void EnsureEventStoreCreated() =>
-            _conn.Execute(Script());
+            _conn.CreateConnection().Execute(Script());
 
         private string Script()
         {