Function TestAlterTableStatementPool

pkg/sql/ast/pool_ddl_test.go:124–181

============================================================ AlterTableStatement pool tests ============================================================

(t *testing.T)

// ============================================================

func TestAlterTableStatementPool(t *testing.T) {
	t.Run("Get returns non-nil", func(t *testing.T) {
		stmt := GetAlterTableStatement()
		if stmt == nil {
			t.Fatal("GetAlterTableStatement() returned nil")
		}
		PutAlterTableStatement(stmt)
	})

	t.Run("Put nil is safe", func(t *testing.T) {
		PutAlterTableStatement(nil)
	})

	t.Run("Fields zeroed after Put", func(t *testing.T) {
		stmt := GetAlterTableStatement()
		stmt.Table = "users"
		stmt.Actions = append(stmt.Actions, AlterTableAction{
			Type:       "ADD COLUMN",
			ColumnName: "email",
			ColumnDef: &ColumnDef{
				Name: "email",
				Type: "VARCHAR(255)",
				Constraints: []ColumnConstraint{
					{
						Type:    "DEFAULT",
						Default: &LiteralValue{Value: "''"},
						Check:   &BinaryExpression{Left: &Identifier{Name: "email"}, Operator: "IS NOT", Right: &LiteralValue{Value: "NULL"}},
					},
				},
			},
			Constraint: &TableConstraint{
				Type:  "UNIQUE",
				Check: &BinaryExpression{Left: &Identifier{Name: "email"}, Operator: "!=", Right: &LiteralValue{Value: "''"}},
			},
		})

		PutAlterTableStatement(stmt)

		if stmt.Table != "" {
			t.Errorf("Table not cleared, got %q", stmt.Table)
		}
		if len(stmt.Actions) != 0 {
			t.Errorf("Actions not cleared, len=%d", len(stmt.Actions))
		}
	})

	t.Run("Pool roundtrip reuse", func(t *testing.T) {
		stmt1 := GetAlterTableStatement()
		stmt1.Table = "products"
		PutAlterTableStatement(stmt1)

		stmt2 := GetAlterTableStatement()
		if stmt2.Table != "" {
			t.Errorf("Reused statement not clean, Table=%q", stmt2.Table)
		}
		PutAlterTableStatement(stmt2)
	})
}