Improved test

pgvector · Nov 28, 2023 · a839187 · a839187
1 parent 4c84d5d
commit a839187
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/Pgvector.Tests/EntityFrameworkCoreTests.cs b/tests/Pgvector.Tests/EntityFrameworkCoreTests.cs
@@ -1,4 +1,6 @@
 using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.Infrastructure;
+using Microsoft.EntityFrameworkCore.Storage;
 using Pgvector.EntityFrameworkCore;
 using System.ComponentModel.DataAnnotations.Schema;
 
@@ -42,8 +44,10 @@ public class EntityFrameworkCoreTests
     public async Task Main()
     {
         await using var ctx = new ItemContext();
-        await ctx.Database.EnsureDeletedAsync();
-        await ctx.Database.EnsureCreatedAsync();
+
+        ctx.Database.ExecuteSql($"DROP TABLE IF EXISTS efcore_items");
+        var databaseCreator = ctx.GetService<IRelationalDatabaseCreator>();
+        databaseCreator.CreateTables();
 
         ctx.Items.Add(new Item { Embedding = new Vector(new float[] { 1, 1, 1 }) });
         ctx.Items.Add(new Item { Embedding = new Vector(new float[] { 2, 2, 2 }) });