Function TestMergeStatementPool

pkg/sql/ast/pool_ddl_test.go:256–329

============================================================ MergeStatement pool tests ============================================================

(t *testing.T)

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

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

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

	t.Run("Fields zeroed after Put", func(t *testing.T) {
		stmt := GetMergeStatement()
		stmt.TargetTable = TableReference{Name: "target"}
		stmt.TargetAlias = "t"
		stmt.SourceTable = TableReference{Name: "source"}
		stmt.SourceAlias = "s"
		stmt.OnCondition = &BinaryExpression{
			Left:     &Identifier{Name: "t.id"},
			Operator: "=",
			Right:    &Identifier{Name: "s.id"},
		}
		stmt.WhenClauses = append(stmt.WhenClauses, &MergeWhenClause{
			Type:      "MATCHED",
			Condition: &BinaryExpression{Left: &Identifier{Name: "x"}, Operator: "=", Right: &LiteralValue{Value: "1"}},
			Action: &MergeAction{
				ActionType: "UPDATE",
				SetClauses: []SetClause{
					{Column: "name", Value: &LiteralValue{Value: "new_name"}},
				},
				Values: []Expression{&LiteralValue{Value: "v1"}},
			},
		})
		stmt.Output = append(stmt.Output, &Identifier{Name: "inserted.id"})

		PutMergeStatement(stmt)

		if stmt.TargetAlias != "" {
			t.Errorf("TargetAlias not cleared, got %q", stmt.TargetAlias)
		}
		if stmt.SourceAlias != "" {
			t.Errorf("SourceAlias not cleared, got %q", stmt.SourceAlias)
		}
		if stmt.OnCondition != nil {
			t.Error("OnCondition not cleared")
		}
		if len(stmt.WhenClauses) != 0 {
			t.Errorf("WhenClauses not cleared, len=%d", len(stmt.WhenClauses))
		}
		if len(stmt.Output) != 0 {
			t.Errorf("Output not cleared, len=%d", len(stmt.Output))
		}
		if stmt.TargetTable.Name != "" {
			t.Errorf("TargetTable not cleared, Name=%q", stmt.TargetTable.Name)
		}
		if stmt.SourceTable.Name != "" {