Function TestAbort_AfterRowCopy

go/logic/migrator_test.go:1242–1281

(t *testing.T)

}

func TestAbort_AfterRowCopy(t *testing.T) {
	migrationContext := base.NewMigrationContext()
	migrator := NewMigrator(migrationContext, "1.0.0")

	// Start listenOnPanicAbort
	go migrator.listenOnPanicAbort()

	// Give listenOnPanicAbort time to start
	time.Sleep(20 * time.Millisecond)

	// Simulate row copy error by sending to rowCopyComplete in a goroutine
	// (unbuffered channel, so send must be async)
	testErr := errors.New("row copy failed")
	go func() {
		migrator.rowCopyComplete <- testErr
	}()

	// Consume the error (simulating what Migrate() does)
	// This is a blocking call that waits for the error
	migrator.consumeRowCopyComplete()

	// Wait for the error to be processed by listenOnPanicAbort
	time.Sleep(50 * time.Millisecond)

	// Check that error was stored
	if got := migrationContext.GetAbortError(); got == nil {
		t.Fatal("Expected abort error to be stored after row copy error")
	} else if got.Error() != "row copy failed" {
		t.Errorf("Expected 'row copy failed', got %v", got)
	}

	// Verify context was cancelled
	ctx := migrationContext.GetContext()
	select {
	case <-ctx.Done():
		// Success
	case <-time.After(1 * time.Second):
		t.Error("Expected context to be cancelled after row copy error")
	}
}

func TestAbort_DuringInspection(t *testing.T) {
	migrationContext := base.NewMigrationContext()