Function TestCreateTableStatementPool

pkg/sql/ast/pool_ddl_test.go:25–118

============================================================ CreateTableStatement pool tests ============================================================

(t *testing.T)

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

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

	t.Run("Put nil is safe", func(t *testing.T) {
		PutCreateTableStatement(nil) // must not panic
	})

	t.Run("Fields zeroed after Put", func(t *testing.T) {
		stmt := GetCreateTableStatement()
		stmt.Name = "users"
		stmt.IfNotExists = true
		stmt.Temporary = true
		stmt.Columns = append(stmt.Columns, ColumnDef{
			Name: "id",
			Type: "INT",
			Constraints: []ColumnConstraint{
				{
					Type:    "NOT NULL",
					Default: &LiteralValue{Value: "0"},
					Check:   &BinaryExpression{Left: &Identifier{Name: "id"}, Operator: ">", Right: &LiteralValue{Value: "0"}},
				},
			},
		})
		stmt.Constraints = append(stmt.Constraints, TableConstraint{
			Name:  "pk",
			Type:  "PRIMARY KEY",
			Check: &BinaryExpression{Left: &Identifier{Name: "x"}, Operator: "=", Right: &LiteralValue{Value: "1"}},
		})
		stmt.PartitionBy = &PartitionBy{
			Type:    "RANGE",
			Columns: []string{"created_at"},
			Boundary: []Expression{
				&LiteralValue{Value: "2024-01-01"},
			},
		}
		stmt.Partitions = append(stmt.Partitions, PartitionDefinition{
			Name:     "p0",
			LessThan: &LiteralValue{Value: "2024-01-01"},
			From:     &LiteralValue{Value: "2024-01-01"},
			To:       &LiteralValue{Value: "2025-01-01"},
			InValues: []Expression{&LiteralValue{Value: "a"}},
		})
		stmt.Inherits = append(stmt.Inherits, "parent_table")
		stmt.Options = append(stmt.Options, TableOption{Name: "ENGINE", Value: "InnoDB"})

		PutCreateTableStatement(stmt)

		if stmt.Name != "" {
			t.Errorf("Name not cleared, got %q", stmt.Name)
		}
		if stmt.IfNotExists {
			t.Error("IfNotExists not cleared")