Function TestBackupWithQuotedStrings

backend/plugin/parser/tsql/backup_test.go:216–245

TestBackupWithQuotedStrings validates that single quotes in WHERE clauses are properly handled

(t *testing.T)

// TestBackupWithQuotedStrings validates that single quotes in WHERE clauses are properly handled
func TestBackupWithQuotedStrings(t *testing.T) {
	a := require.New(t)

	// Test case with single quotes
	input := `DELETE FROM AdminPosition WHERE positionName = 'BPM Admin';`

	result, err := TransformDMLToSelect(context.Background(), base.TransformContext{},
		input, "TestIdentityDB", "bbdataarchive", "backup")
	a.NoError(err)
	a.Len(result, 1)

	stmt := result[0].Statement

	// With simple SELECT INTO, quotes are handled naturally by SQL Server
	// The WHERE clause should appear as-is
	a.Contains(stmt, "WHERE positionName = 'BPM Admin'")

	// Also test with apostrophes in the string
	input2 := `UPDATE positions SET title = 'O''Reilly''s Manager' WHERE id = 1;`
	result2, err := TransformDMLToSelect(context.Background(), base.TransformContext{},
		input2, "db", "backup", "rollback")
	a.NoError(err)
	a.Len(result2, 1)

	// For UPDATE, we only backup the rows that will be changed (WHERE clause)
	// The SET clause is not part of the backup, only the WHERE clause matters
	stmt2 := result2[0].Statement
	// The WHERE clause should be present
	a.Contains(stmt2, "WHERE id = 1")
}

// TestBackupSkipsTempTableTargets validates BYT-9359: prior backup must skip
// any UPDATE/DELETE whose target table is a temp table (#name local, ##name