Fix more tests

ast/alter_database_set_statement.go

@@ -44,6 +44,17 @@ type DelayedDurabilityDatabaseOption struct {
 func (d *DelayedDurabilityDatabaseOption) node()           {}
 func (d *DelayedDurabilityDatabaseOption) databaseOption() {}
 
+// AutoCreateStatisticsDatabaseOption represents AUTO_CREATE_STATISTICS option with optional INCREMENTAL
+type AutoCreateStatisticsDatabaseOption struct {
+	OptionKind       string // "AutoCreateStatistics"
+	OptionState      string // "On" or "Off"
+	HasIncremental   bool   // Whether INCREMENTAL is specified
+	IncrementalState string // "On" or "Off"
+}
+
+func (a *AutoCreateStatisticsDatabaseOption) node()           {}
+func (a *AutoCreateStatisticsDatabaseOption) databaseOption() {}
+
 // IdentifierDatabaseOption represents a database option with an identifier value
 type IdentifierDatabaseOption struct {
 	OptionKind string      `json:"OptionKind,omitempty"` // "CatalogCollation"
@@ -64,6 +75,14 @@ func (o *OnOffDatabaseOption) createDatabaseOption()            {}
 func (i *IdentifierDatabaseOption) createDatabaseOption()       {}
 func (d *DelayedDurabilityDatabaseOption) createDatabaseOption() {}
 
+// SimpleDatabaseOption represents a simple database option with just OptionKind (e.g., ENABLE_BROKER)
+type SimpleDatabaseOption struct {
+	OptionKind string `json:"OptionKind,omitempty"`
+}
+
+func (d *SimpleDatabaseOption) node()                 {}
+func (d *SimpleDatabaseOption) createDatabaseOption() {}
+
 // MaxSizeDatabaseOption represents a MAXSIZE option.
 type MaxSizeDatabaseOption struct {
 	OptionKind string           `json:"OptionKind,omitempty"`
@@ -173,3 +192,79 @@ type DatabaseConfigurationClearOption struct {
 }
 
 func (d *DatabaseConfigurationClearOption) node() {}
+
+// RemoteDataArchiveDatabaseOption represents REMOTE_DATA_ARCHIVE database option
+type RemoteDataArchiveDatabaseOption struct {
+	OptionKind  string                       // "RemoteDataArchive"
+	OptionState string                       // "On", "Off", "NotSet"
+	Settings    []RemoteDataArchiveDbSetting // Settings like SERVER, CREDENTIAL, FEDERATED_SERVICE_ACCOUNT
+}
+
+func (r *RemoteDataArchiveDatabaseOption) node()           {}
+func (r *RemoteDataArchiveDatabaseOption) databaseOption() {}
+
+// RemoteDataArchiveDbSetting is an interface for Remote Data Archive settings
+type RemoteDataArchiveDbSetting interface {
+	Node
+	remoteDataArchiveDbSetting()
+}
+
+// RemoteDataArchiveDbServerSetting represents the SERVER setting
+type RemoteDataArchiveDbServerSetting struct {
+	SettingKind string           // "Server"
+	Server      ScalarExpression // The server string literal
+}
+
+func (r *RemoteDataArchiveDbServerSetting) node()                      {}
+func (r *RemoteDataArchiveDbServerSetting) remoteDataArchiveDbSetting() {}
+
+// RemoteDataArchiveDbCredentialSetting represents the CREDENTIAL setting
+type RemoteDataArchiveDbCredentialSetting struct {
+	SettingKind string      // "Credential"
+	Credential  *Identifier // The credential name
+}
+
+func (r *RemoteDataArchiveDbCredentialSetting) node()                      {}
+func (r *RemoteDataArchiveDbCredentialSetting) remoteDataArchiveDbSetting() {}
+
+// RemoteDataArchiveDbFederatedServiceAccountSetting represents the FEDERATED_SERVICE_ACCOUNT setting
+type RemoteDataArchiveDbFederatedServiceAccountSetting struct {
+	SettingKind string // "FederatedServiceAccount"
+	IsOn        bool   // true for ON, false for OFF
+}
+
+func (r *RemoteDataArchiveDbFederatedServiceAccountSetting) node()                      {}
+func (r *RemoteDataArchiveDbFederatedServiceAccountSetting) remoteDataArchiveDbSetting() {}
+
+// ChangeTrackingDatabaseOption represents the CHANGE_TRACKING database option
+type ChangeTrackingDatabaseOption struct {
+	OptionKind  string                            // "ChangeTracking"
+	OptionState string                            // "On", "Off", "NotSet"
+	Details     []ChangeTrackingOptionDetail      // AUTO_CLEANUP, CHANGE_RETENTION
+}
+
+func (c *ChangeTrackingDatabaseOption) node()           {}
+func (c *ChangeTrackingDatabaseOption) databaseOption() {}
+
+// ChangeTrackingOptionDetail is an interface for change tracking option details
+type ChangeTrackingOptionDetail interface {
+	Node
+	changeTrackingOptionDetail()
+}
+
+// AutoCleanupChangeTrackingOptionDetail represents AUTO_CLEANUP option
+type AutoCleanupChangeTrackingOptionDetail struct {
+	IsOn bool
+}
+
+func (a *AutoCleanupChangeTrackingOptionDetail) node()                        {}
+func (a *AutoCleanupChangeTrackingOptionDetail) changeTrackingOptionDetail() {}
+
+// ChangeRetentionChangeTrackingOptionDetail represents CHANGE_RETENTION option
+type ChangeRetentionChangeTrackingOptionDetail struct {
+	RetentionPeriod ScalarExpression
+	Unit            string // "Days", "Hours", "Minutes"
+}
+
+func (c *ChangeRetentionChangeTrackingOptionDetail) node()                        {}
+func (c *ChangeRetentionChangeTrackingOptionDetail) changeTrackingOptionDetail() {}

ast/alter_index_statement.go

@@ -5,14 +5,42 @@ type AlterIndexStatement struct {
 	Name           *Identifier
 	All            bool
 	OnName         *SchemaObjectName
-	AlterIndexType string // "Rebuild", "Reorganize", "Disable", "Set", etc.
+	AlterIndexType string // "Rebuild", "Reorganize", "Disable", "Set", "UpdateSelectiveXmlPaths", etc.
 	Partition      *PartitionSpecifier
 	IndexOptions   []IndexOption
+	PromotedPaths  []*SelectiveXmlIndexPromotedPath
+	XmlNamespaces  *XmlNamespaces
 }
 
 func (s *AlterIndexStatement) statement() {}
 func (s *AlterIndexStatement) node()      {}
 
+// SelectiveXmlIndexPromotedPath represents a path in a selective XML index
+type SelectiveXmlIndexPromotedPath struct {
+	Name           *Identifier
+	Path           *StringLiteral
+	XQueryDataType *StringLiteral
+	MaxLength      *IntegerLiteral
+	IsSingleton    bool
+}
+
+func (s *SelectiveXmlIndexPromotedPath) node() {}
+
+// XmlNamespaces represents a WITH XMLNAMESPACES clause
+type XmlNamespaces struct {
+	XmlNamespacesElements []*XmlNamespacesAliasElement
+}
+
+func (x *XmlNamespaces) node() {}
+
+// XmlNamespacesAliasElement represents an alias element in XMLNAMESPACES
+type XmlNamespacesAliasElement struct {
+	Identifier *Identifier
+	String     *StringLiteral
+}
+
+func (x *XmlNamespacesAliasElement) node() {}
+
 // PartitionSpecifier represents a partition specifier
 type PartitionSpecifier struct {
 	All     bool

ast/alter_simple_statements.go

@@ -257,12 +257,54 @@ type OnOffFullTextCatalogOption struct {
 
 // AlterFulltextIndexStatement represents an ALTER FULLTEXT INDEX statement.
 type AlterFulltextIndexStatement struct {
-	OnName *SchemaObjectName `json:"OnName,omitempty"`
+	OnName *SchemaObjectName              `json:"OnName,omitempty"`
+	Action AlterFullTextIndexActionOption `json:"Action,omitempty"`
 }
 
 func (s *AlterFulltextIndexStatement) node()      {}
 func (s *AlterFulltextIndexStatement) statement() {}
 
+// AlterFullTextIndexActionOption is an interface for fulltext index actions
+type AlterFullTextIndexActionOption interface {
+	alterFullTextIndexAction()
+}
+
+// SimpleAlterFullTextIndexAction represents simple actions like ENABLE, DISABLE, etc.
+type SimpleAlterFullTextIndexAction struct {
+	ActionKind string `json:"ActionKind,omitempty"`
+}
+
+func (*SimpleAlterFullTextIndexAction) node()                     {}
+func (*SimpleAlterFullTextIndexAction) alterFullTextIndexAction() {}
+
+// AddAlterFullTextIndexAction represents an ADD action for fulltext index
+type AddAlterFullTextIndexAction struct {
+	Columns          []*FullTextIndexColumn `json:"Columns,omitempty"`
+	WithNoPopulation bool                   `json:"WithNoPopulation"`
+}
+
+func (*AddAlterFullTextIndexAction) node()                     {}
+func (*AddAlterFullTextIndexAction) alterFullTextIndexAction() {}
+
+// DropAlterFullTextIndexAction represents a DROP action for fulltext index
+type DropAlterFullTextIndexAction struct {
+	Columns          []*Identifier `json:"Columns,omitempty"`
+	WithNoPopulation bool          `json:"WithNoPopulation"`
+}
+
+func (*DropAlterFullTextIndexAction) node()                     {}
+func (*DropAlterFullTextIndexAction) alterFullTextIndexAction() {}
+
+// FullTextIndexColumn represents a column in a fulltext index
+type FullTextIndexColumn struct {
+	Name                 *Identifier              `json:"Name,omitempty"`
+	TypeColumn           *Identifier              `json:"TypeColumn,omitempty"`
+	LanguageTerm         *IdentifierOrValueExpression `json:"LanguageTerm,omitempty"`
+	StatisticalSemantics bool                     `json:"StatisticalSemantics"`
+}
+
+func (*FullTextIndexColumn) node() {}
+
 // AlterSymmetricKeyStatement represents an ALTER SYMMETRIC KEY statement.
 type AlterSymmetricKeyStatement struct {
 	Name                 *Identifier        `json:"Name,omitempty"`

ast/alter_table_alter_index_statement.go

@@ -53,3 +53,13 @@ type OrderIndexOption struct {
 
 func (o *OrderIndexOption) indexOption() {}
 func (o *OrderIndexOption) node()        {}
+
+// MaxDurationOption represents MAX_DURATION option for resumable index operations
+type MaxDurationOption struct {
+	MaxDuration ScalarExpression
+	Unit        string // "", "Minutes"
+	OptionKind  string // "MaxDuration"
+}
+
+func (m *MaxDurationOption) indexOption() {}
+func (m *MaxDurationOption) node()        {}

ast/alter_table_filetable_namespace_statement.go

@@ -0,0 +1,10 @@
+package ast
+
+// AlterTableFileTableNamespaceStatement represents ALTER TABLE ... ENABLE/DISABLE FILETABLE_NAMESPACE
+type AlterTableFileTableNamespaceStatement struct {
+	SchemaObjectName *SchemaObjectName `json:"SchemaObjectName,omitempty"`
+	IsEnable         bool              `json:"IsEnable,omitempty"`
+}
+
+func (s *AlterTableFileTableNamespaceStatement) node()      {}
+func (s *AlterTableFileTableNamespaceStatement) statement() {}

ast/alter_table_switch_statement.go

@@ -28,7 +28,16 @@ type TruncateTargetTableSwitchOption struct {
 func (o *TruncateTargetTableSwitchOption) tableSwitchOption() {}
 func (o *TruncateTargetTableSwitchOption) node()              {}
 
-// LowPriorityLockWait represents LOW_PRIORITY_LOCK_WAIT option
+// LowPriorityLockWaitTableSwitchOption represents WAIT_AT_LOW_PRIORITY option
+type LowPriorityLockWaitTableSwitchOption struct {
+	OptionKind string
+	Options    []LowPriorityLockWaitOption
+}
+
+func (o *LowPriorityLockWaitTableSwitchOption) tableSwitchOption() {}
+func (o *LowPriorityLockWaitTableSwitchOption) node()              {}
+
+// LowPriorityLockWait represents LOW_PRIORITY_LOCK_WAIT option (legacy)
 type LowPriorityLockWait struct {
 	MaxDuration       ScalarExpression
 	MaxDurationUnit   string // "MINUTES", "SECONDS"

ast/at_time_zone.go

@@ -0,0 +1,10 @@
+package ast
+
+// AtTimeZoneCall represents an AT TIME ZONE expression
+type AtTimeZoneCall struct {
+	DateValue ScalarExpression
+	TimeZone  ScalarExpression
+}
+
+func (*AtTimeZoneCall) node()             {}
+func (*AtTimeZoneCall) scalarExpression() {}

ast/boolean_expression.go

@@ -5,3 +5,13 @@ type BooleanExpression interface {
 	Node
 	booleanExpression()
 }
+
+// BooleanScalarPlaceholder is a temporary marker used during parsing when we
+// encounter a scalar expression in a boolean context without a comparison operator.
+// This allows the caller to detect and handle cases like (XACT_STATE()) = -1.
+type BooleanScalarPlaceholder struct {
+	Scalar ScalarExpression
+}
+
+func (b *BooleanScalarPlaceholder) booleanExpression() {}
+func (b *BooleanScalarPlaceholder) node()              {}

ast/create_simple_statements.go

@@ -74,10 +74,38 @@ type QueueOptionSimple struct {
 func (o *QueueOptionSimple) node()        {}
 func (o *QueueOptionSimple) queueOption() {}
 
+// QueueProcedureOption represents a PROCEDURE_NAME option.
+type QueueProcedureOption struct {
+	OptionValue *SchemaObjectName `json:"OptionValue,omitempty"`
+	OptionKind  string            `json:"OptionKind,omitempty"` // "ActivationProcedureName"
+}
+
+func (o *QueueProcedureOption) node()        {}
+func (o *QueueProcedureOption) queueOption() {}
+
+// QueueValueOption represents an option with an integer value.
+type QueueValueOption struct {
+	OptionValue ScalarExpression `json:"OptionValue,omitempty"`
+	OptionKind  string           `json:"OptionKind,omitempty"` // "ActivationMaxQueueReaders"
+}
+
+func (o *QueueValueOption) node()        {}
+func (o *QueueValueOption) queueOption() {}
+
+// QueueExecuteAsOption represents an EXECUTE AS option.
+type QueueExecuteAsOption struct {
+	OptionValue *ExecuteAsClause `json:"OptionValue,omitempty"`
+	OptionKind  string           `json:"OptionKind,omitempty"` // "ActivationExecuteAs"
+}
+
+func (o *QueueExecuteAsOption) node()        {}
+func (o *QueueExecuteAsOption) queueOption() {}
+
 // CreateQueueStatement represents a CREATE QUEUE statement.
 type CreateQueueStatement struct {
-	Name         *SchemaObjectName `json:"Name,omitempty"`
-	QueueOptions []QueueOption     `json:"QueueOptions,omitempty"`
+	Name         *SchemaObjectName            `json:"Name,omitempty"`
+	OnFileGroup  *IdentifierOrValueExpression `json:"OnFileGroup,omitempty"`
+	QueueOptions []QueueOption                `json:"QueueOptions,omitempty"`
 }
 
 func (s *CreateQueueStatement) node()      {}
@@ -408,8 +436,13 @@ func (s *CreateTypeTableStatement) statement() {}
 
 // CreateXmlIndexStatement represents a CREATE XML INDEX statement.
 type CreateXmlIndexStatement struct {
-	Name   *Identifier       `json:"Name,omitempty"`
-	OnName *SchemaObjectName `json:"OnName,omitempty"`
+	Primary               bool          `json:"Primary,omitempty"`
+	XmlColumn             *Identifier   `json:"XmlColumn,omitempty"`
+	SecondaryXmlIndexName *Identifier   `json:"SecondaryXmlIndexName,omitempty"`
+	SecondaryXmlIndexType string        `json:"SecondaryXmlIndexType,omitempty"` // "NotSpecified", "Value", "Path", "Property"
+	Name                  *Identifier   `json:"Name,omitempty"`
+	OnName                *SchemaObjectName `json:"OnName,omitempty"`
+	IndexOptions          []IndexOption `json:"IndexOptions,omitempty"`
 }
 
 func (s *CreateXmlIndexStatement) node()      {}

ast/create_table_statement.go

@@ -11,8 +11,17 @@ type CreateTableStatement struct {
 	TextImageOn                  *IdentifierOrValueExpression
 	FileStreamOn                 *IdentifierOrValueExpression
 	Options                      []TableOption
+	FederationScheme             *FederationScheme
 }
 
+// FederationScheme represents a FEDERATED ON clause
+type FederationScheme struct {
+	DistributionName *Identifier
+	ColumnName       *Identifier
+}
+
+func (*FederationScheme) node() {}
+
 // TableDataCompressionOption represents a DATA_COMPRESSION option
 type TableDataCompressionOption struct {
 	DataCompressionOption *DataCompressionOption

ast/drop_statements.go

@@ -123,7 +123,7 @@ type LowPriorityLockWaitOption interface {
 
 // LowPriorityLockWaitMaxDurationOption represents MAX_DURATION option
 type LowPriorityLockWaitMaxDurationOption struct {
-	MaxDuration *IntegerLiteral
+	MaxDuration ScalarExpression
 	Unit        string // Minutes or Seconds
 	OptionKind  string // MaxDuration
 }