DD.Persistence.Database.Postgres.Test/UnitTestCheckHyperTables.cs

@@ -1,5 +1,6 @@
 using DD.Persistence.Database.Entity;
 using Mapster;
+using Microsoft.EntityFrameworkCore;
 using Microsoft.Extensions.DependencyInjection;
 using Npgsql;
 
@@ -19,6 +20,8 @@ public class UnitTestCheckHyperTables : IClassFixture<DbFixture>
     [Fact]
     public void CreateHyperTable_For_ParameterData_Return_Success()
     {
+        var chunksCount = 0;
+
         var entity = new ParameterData()
         {
             DiscriminatorId = Guid.NewGuid(),
@@ -45,24 +48,11 @@ public class UnitTestCheckHyperTables : IClassFixture<DbFixture>
             context.ParameterData.Add(entity4);
 
             context.SaveChanges();
-        }
 
-        var chunksCount = 0;
-        using (var connection = new NpgsqlConnection(_connectionString))
-        {
-            connection.Open();
             string sql = "select count(*) from (select show_chunks('parameter_data'));";
+            var queryRow = context.Database.SqlQueryRaw<int>(sql);
 
-            using (var command = new NpgsqlCommand(sql, connection))
-            {
-                using (var reader = command.ExecuteReader())
-                {
-                    while (reader.Read())
-                    {
-                        chunksCount += reader.GetInt32(0);
-                    }
-                }
-            }
+            chunksCount = queryRow.AsEnumerable().FirstOrDefault();
         }
 
         Assert.Equal(2, chunksCount);