Exemple #1
0
        public static async Task <TestDbContext> GetDbContextAsync(DbProvider provider, IEnumerable <object> seedData = null, TestConfiguration testConfiguration = TestConfiguration.Default)
        {
            var optionsBuilder = new DbContextOptionsBuilder();

            TestDbContext context = null;

            if (provider == DbProvider.Sqlite)
            {
                var sqlConnection = new SqliteConnection("Data Source=:memory:;");
                await sqlConnection.OpenAsync();

                optionsBuilder.UseSqlite(sqlConnection).EnableSensitiveDataLogging(true);

                context = new TestDbContext(optionsBuilder.Options);
            }
            else
            {
                string sqlUser     = Environment.GetEnvironmentVariable("INTEGRATION_TEST_SQL_SERVER_DB_USER");
                string sqlPassword = Environment.GetEnvironmentVariable("INTEGRATION_TEST_SQL_SERVER_DB_PASSWORD");
                string sqlServer   = Environment.GetEnvironmentVariable("INTEGRATION_TEST_SQL_SERVER");
                string sqldb       = Environment.GetEnvironmentVariable("INTEGRATION_TEST_SQL_SERVER_DB");

                var connectionStringBuilder = new SqlConnectionStringBuilder
                {
                    DataSource     = string.IsNullOrWhiteSpace(sqlServer) ? @"localhost\SQLEXPRESS" : sqlServer,
                    InitialCatalog = string.IsNullOrWhiteSpace(sqldb) ? @"entityframeworkcore-manipulation-extensions-integration-testing" : sqldb,
                };

                if (string.IsNullOrWhiteSpace(sqlUser) || string.IsNullOrWhiteSpace(sqlUser))
                {
                    connectionStringBuilder.IntegratedSecurity = true;
                }
                else
                {
                    connectionStringBuilder.UserID   = sqlUser;
                    connectionStringBuilder.Password = sqlPassword;
                }

                var sqlConnection = new SqlConnection(connectionStringBuilder.ConnectionString);
                await sqlConnection.OpenAsync();

                optionsBuilder.UseSqlServer(sqlConnection).EnableSensitiveDataLogging(true);

                context = new TestDbContext(optionsBuilder.Options);

                await context.Database.MigrateAsync();

                // Clear the DB
                SqlCommand command = sqlConnection.CreateCommand();
                command.CommandText = "DELETE FROM TestEntities; DELETE FROM TestEntitiesWithCompositeKey; DELETE FROM TestInterceptorEntities;";
                await command.ExecuteNonQueryAsync();
            }

            if (seedData != null)
            {
                using var seedContext = new TestDbContext(optionsBuilder.Options);
                seedContext.AddRange(seedData);
                await seedContext.SaveChangesAsync();
            }

            if (testConfiguration == TestConfiguration.SqlServerRegularTableTypes)
            {
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.UseMemoryOptimizedTableTypes = false;
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.DetaultUseTableValuedParametersParameterCountTreshold = 0;
            }
            else if (testConfiguration == TestConfiguration.SqlServerRegularTableTypesWithClusteredIndex)
            {
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.DetaultUseTableValuedParametersParameterCountTreshold = 0;
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.AddEntityConifugration <TestEntity>(new EntityConifugration {
                    UseMemoryOptimizedTableTypes = false, TableTypeIndex = SqlServerTableTypeIndex.ClusteredIndex
                });
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.AddEntityConifugration <TestEntityCompositeKey>(new EntityConifugration {
                    UseMemoryOptimizedTableTypes = false, TableTypeIndex = SqlServerTableTypeIndex.ClusteredIndex
                });
            }
            else if (testConfiguration == TestConfiguration.SqlServerRegularTableTypesWithNonclusteredIndex)
            {
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.DetaultUseTableValuedParametersParameterCountTreshold = 0;
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.AddEntityConifugration <TestEntity>(new EntityConifugration {
                    UseMemoryOptimizedTableTypes = false, TableTypeIndex = SqlServerTableTypeIndex.NonClusteredIndex
                });
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.AddEntityConifugration <TestEntityCompositeKey>(new EntityConifugration {
                    UseMemoryOptimizedTableTypes = false, TableTypeIndex = SqlServerTableTypeIndex.NonClusteredIndex
                });
            }
            else if (testConfiguration == TestConfiguration.SqlServerMemoryOptimizedTableTypes)
            {
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.UseMemoryOptimizedTableTypes = true;
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.DetaultUseTableValuedParametersParameterCountTreshold = 0;
            }
            else if (testConfiguration == TestConfiguration.SqlServerMemoryOptimizedTableTypesWithNonclusteredIndex)
            {
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.DetaultUseTableValuedParametersParameterCountTreshold = 0;
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.AddEntityConifugration <TestEntity>(new EntityConifugration {
                    UseMemoryOptimizedTableTypes = true, TableTypeIndex = SqlServerTableTypeIndex.NonClusteredIndex
                });
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.AddEntityConifugration <TestEntityCompositeKey>(new EntityConifugration {
                    UseMemoryOptimizedTableTypes = true, TableTypeIndex = SqlServerTableTypeIndex.NonClusteredIndex
                });
            }
            else if (testConfiguration == TestConfiguration.SqlServerOutputInto)
            {
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.AddEntityConifugration <TestEntity>(new EntityConifugration {
                    HasTrigger = true
                });
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.AddEntityConifugration <TestEntityCompositeKey>(new EntityConifugration {
                    HasTrigger = true
                });
            }
            else if (testConfiguration == TestConfiguration.SqlServerMergeSync)
            {
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.UseMerge = true;
            }
            else if (testConfiguration == TestConfiguration.SqlServerSimpleStatementsSync)
            {
                context.ManipulationExtensionsConfiguration.SqlServerConfiguration.UseMerge = false;
            }

            return(context);
        }