public async Task Migrate_AllScriptOk_NoRollback()
{
var config = ConfigProvider.GetConfig();
var connectionString = String.Format(config.ConnectionStringMask, "test_ok");
var builder = new MigratorBuilder();
builder.UseCodeMigrations().FromAssembly(Assembly.GetExecutingAssembly());
builder.UseScriptMigrations().FromDirectory(Path.Combine(Directory.GetCurrentDirectory(), "ScriptMigrations"));
builder.UsePostgreSQL(connectionString);
builder.UseUpgradeMigrationPolicy(MigrationPolicy.All);
builder.UseDowngradeMigrationPolicy(MigrationPolicy.All);
builder.SetUpTargetVersion(new DbVersion(3, 0));
var migrator = builder.Build();
await migrator.MigrateAsync();
var migrationProvider = new PostgreDbProvider(new PostgreDbProviderOptions(connectionString));
await migrationProvider.OpenConnectionAsync();
var currentDbVersion = await migrationProvider.GetDbVersionSafeAsync();
await migrationProvider.CloseConnectionAsync();
Assert.Equal(new DbVersion(3, 0), currentDbVersion);
}
public void OverrideDefaultVariablesFromProvider()
{
// arrange
var builder = new MigratorBuilder();
var providerMock = new Mock <IDbProvider>();
providerMock
.Setup(x => x.GetDefaultVariables())
.Returns(() => new Dictionary <string, string>
{
{ DefaultVariables.User, ProviderUserName },
{ DefaultVariables.DbName, ProviderDbName }
});
var factoryMock = new Mock <IDbProviderFactory>();
factoryMock
.Setup(x => x.CreateDbProvider())
.Returns(() => providerMock.Object);
IReadOnlyDictionary <string, string> scriptVariables = null;
var migrationsProviderMock = new Mock <IMigrationsProvider>();
migrationsProviderMock
.Setup(x => x.GetMigrations(
It.IsAny <IDbProvider>(),
It.IsAny <Dictionary <string, string> >(),
It.IsAny <ILogger>()))
.Callback <IDbProvider, IReadOnlyDictionary <string, string>, ILogger>((provider, variables, logger) => { scriptVariables = variables; })
.Returns <IDbProvider, IReadOnlyDictionary <string, string>, ILogger>((provider, variables, logger) => new List <IMigration>(0));
builder.UseCustomMigrationsProvider(migrationsProviderMock.Object);
builder.UserDbProviderFactory(factoryMock.Object);
builder.UseVariable(DefaultVariables.User, ManualUserName);
// act
var migrator = builder.Build();
// assert
migrator.Should().NotBeNull("because we've create migrator");
scriptVariables.Should().NotBeNull("because provider returns 2 variables");
scriptVariables.Count.Should().Be(2, "because provider returns variable with user and db names");
scriptVariables.ContainsKey(DefaultVariables.User).Should().BeTrue("because user name is default variable");
scriptVariables[DefaultVariables.User].Should().BeEquivalentTo(ManualUserName, "because we set it manually to builder");
scriptVariables.ContainsKey(DefaultVariables.DbName).Should().BeTrue("because db name is default variable");
scriptVariables[DefaultVariables.DbName].Should().BeEquivalentTo(ProviderDbName, "because we set it manually to mock");
}
public async Task Migrate_AllScriptOk_Rollback()
{
var config = ConfigProvider.GetConfig();
var connectionString = String.Format(config.ConnectionStringMask, "test_rollback");
var builder = new MigratorBuilder();
builder.UseCodeMigrations().FromAssembly(Assembly.GetExecutingAssembly());
builder.UseScriptMigrations().FromDirectory(Path.Combine(Directory.GetCurrentDirectory(), "ScriptMigrations"));
builder.UsePostgreSQL(connectionString);
builder.UseUpgradeMigrationPolicy(MigrationPolicy.Allowed);
builder.UseDowngradeMigrationPolicy(MigrationPolicy.Allowed);
builder.SetUpTargetVersion(new DbVersion(6, 0));
var migrator = builder.Build();
try
{
await migrator.MigrateAsync();
// last migration is incorrect, can not go here
Assert.False(true);
}
catch
{
// ignored
}
var migrationProvider = new PostgreDbProvider(new PostgreDbProviderOptions(connectionString));
await migrationProvider.OpenConnectionAsync();
var actualAppliedMigrations = await migrationProvider.GetAppliedMigrationVersionAsync();
await migrationProvider.CloseConnectionAsync();
var expectedAppliedMigrations = new HashSet <DbVersion>
{
new DbVersion(1, 0),
new DbVersion(2, 0),
new DbVersion(3, 0),
new DbVersion(4, 0),
new DbVersion(5, 0)
};
Assert.Equal(expectedAppliedMigrations, actualAppliedMigrations);
}
public async Task Migrate_AllScriptOk_SwitchedOffTransaction()
{
var config = ConfigProvider.GetConfig();
var connectionString = String.Format(config.ConnectionStringMask, "test_without_transactions");
var builder = new MigratorBuilder();
builder.UseCodeMigrations().FromAssembly(Assembly.GetExecutingAssembly());
builder.UseScriptMigrations().FromDirectory(Path.Combine(Directory.GetCurrentDirectory(), "ScriptMigrations"));
builder.UsePostgreSQL(connectionString);
builder.UseUpgradeMigrationPolicy(MigrationPolicy.Allowed);
builder.UseDowngradeMigrationPolicy(MigrationPolicy.Allowed);
builder.SetUpTargetVersion(new DbVersion(5, 0));
var migrator = builder.Build();
await migrator.MigrateAsync();
var migrationProvider = new PostgreDbProvider(new PostgreDbProviderOptions(connectionString));
await migrationProvider.OpenConnectionAsync();
var actualAppliedMigrations = await migrationProvider.GetAppliedMigrationVersionAsync();
await migrationProvider.CloseConnectionAsync();
var expectedAppliedMigrations = new HashSet <DbVersion>
{
new DbVersion(1, 0),
new DbVersion(2, 0),
new DbVersion(3, 0),
new DbVersion(4, 0),
new DbVersion(5, 0)
};
Assert.Equal(expectedAppliedMigrations, actualAppliedMigrations);
}
public async Task StartAsync(CancellationToken cToken)
{
_logger.LogInformation("Мигратор запущен");
var builder = new MigratorBuilder(_services);
builder.UsePostgreSQL(
_configuration.MigrationOptions.ConnectionString,
lcCollate: _configuration.MigrationOptions.LC_Collate,
lcCtype: _configuration.MigrationOptions.LC_Type,
databaseEncoding: _configuration.MigrationOptions.DataBaseEncoding,
migrationTableHistoryName: "migration_history");
builder.UseScriptMigrations()
.FromAssembly(typeof(MigrationService).Assembly, "Migrations.Scripts.");
builder.UseCodeMigrations()
.FromAssembly <IMigration>(typeof(MigrationService).Assembly);
builder.UseUpgradeMigrationPolicy(MigrationPolicy.Allowed);
builder.UserLogger(_logger);
if (_configuration.MigrationOptions.LogSql)
{
builder.UseLoggerForSql(_logger);
}
if (!String.IsNullOrWhiteSpace(_configuration.MigrationOptions.GrantUser))
{
builder.UseVariable(DefaultVariables.User, _configuration.MigrationOptions.GrantUser);
}
var migrator = builder.Build();
var result = await migrator.MigrateSafeAsync(cToken);
_logger.LogInformation(result.IsSuccessfully
? "Миграции успешно выполнены"
: $"Ошибка миграции: {result.ErrorMessage}");
}