Method TestRevertEmpty

go/logic/migrator_test.go:1042–1091

()

}

func (suite *MigratorTestSuite) TestRevertEmpty() {
	ctx := context.Background()

	_, err := suite.db.ExecContext(ctx, fmt.Sprintf("CREATE TABLE %s (id INT PRIMARY KEY, s CHAR(32))", getTestTableName()))
	suite.Require().NoError(err)

	var oldTableName string

	// perform original migration
	connectionConfig, err := getTestConnectionConfig(ctx, suite.mysqlContainer)
	suite.Require().NoError(err)
	{
		migrationContext := newTestMigrationContext()
		migrationContext.ApplierConnectionConfig = connectionConfig
		migrationContext.InspectorConnectionConfig = connectionConfig
		migrationContext.SetConnectionConfig("innodb")
		migrationContext.AlterStatement = "ADD COLUMN newcol CHAR(32)"
		migrationContext.Checkpoint = true
		migrationContext.CheckpointIntervalSeconds = 10
		migrationContext.DropServeSocket = true
		migrationContext.InitiallyDropOldTable = true
		migrationContext.UseGTIDs = true

		migrator := NewMigrator(migrationContext, "0.0.0")

		err = migrator.Migrate()
		oldTableName = migrationContext.GetOldTableName()
		suite.Require().NoError(err)
		suite.Require().Less(migrationContext.TimeSinceLastHeartbeatOnChangelog(), 24*time.Hour)
	}

	// revert the original migration
	{
		migrationContext := newTestMigrationContext()
		migrationContext.ApplierConnectionConfig = connectionConfig
		migrationContext.InspectorConnectionConfig = connectionConfig
		migrationContext.SetConnectionConfig("innodb")
		migrationContext.DropServeSocket = true
		migrationContext.UseGTIDs = true
		migrationContext.Revert = true
		migrationContext.OkToDropTable = true
		migrationContext.OldTableName = oldTableName

		migrator := NewMigrator(migrationContext, "0.0.0")

		err = migrator.Revert()
		suite.Require().NoError(err)
		suite.Require().Less(migrationContext.TimeSinceLastHeartbeatOnChangelog(), 24*time.Hour)
	}
}

func (suite *MigratorTestSuite) TestRevert() {
	ctx := context.Background()