Function TestIdentityColumnHandling

backend/plugin/parser/tsql/backup_test.go:163–190

TestIdentityColumnHandling validates that our implementation correctly handles IDENTITY columns This test verifies: 1. The backup uses simple SELECT INTO (copying IDENTITY properties naturally) 2. The restore.go handles IDENTITY_INSERT during rollback

(t *testing.T)

// 1. The backup uses simple SELECT INTO (copying IDENTITY properties naturally)
// 2. The restore.go handles IDENTITY_INSERT during rollback
func TestIdentityColumnHandling(t *testing.T) {
	a := require.New(t)

	// Test case: DELETE from a table that typically has IDENTITY columns
	input := `DELETE FROM positions WHERE position_id = 1;`

	result, err := TransformDMLToSelect(context.Background(), base.TransformContext{},
		input, "db", "backupDB", "rollback")
	a.NoError(err)
	a.Len(result, 1)

	// Verify the generated SQL uses simple SELECT INTO
	stmt := result[0].Statement

	// Key assertions about the generated SQL:
	// 1. Uses simple SELECT INTO
	a.Contains(stmt, "SELECT * INTO [backupDB].[dbo].[rollback_positions_db]")

	// 2. Selects from the original table with proper bracketing
	a.Contains(stmt, "SELECT [db].[dbo].[positions].* FROM")

	// 3. Includes the WHERE clause
	a.Contains(stmt, "WHERE position_id = 1")

	// The approach:
	// Backup: Simple SELECT * INTO backup_table (copies IDENTITY property naturally)
	// Restore: restore.go handles IDENTITY_INSERT ON/OFF for rollback
}

// TestBackupStatementStructure validates the structure of backup statements
func TestBackupStatementStructure(t *testing.T) {