Function TestInsertStatementPool

pkg/sql/ast/pool_test.go:24–65

Test InsertStatement pool

(t *testing.T)

// Test InsertStatement pool
func TestInsertStatementPool(t *testing.T) {
	t.Run("Get and Put", func(t *testing.T) {
		// Get from pool
		stmt := GetInsertStatement()
		if stmt == nil {
			t.Fatal("GetInsertStatement() returned nil")
		}

		// Use it
		stmt.TableName = "users"
		stmt.Columns = []Expression{
			&Identifier{Name: "name"},
			&Identifier{Name: "email"},
		}
		// Values is now [][]Expression for multi-row support
		stmt.Values = [][]Expression{
			{
				&LiteralValue{Value: "John"},
				&LiteralValue{Value: "john@example.com"},
			},
		}

		// Return to pool
		PutInsertStatement(stmt)

		// Verify it was cleaned
		if stmt.TableName != "" {
			t.Errorf("TableName not cleared, got %v", stmt.TableName)
		}
		if len(stmt.Columns) != 0 {
			t.Errorf("Columns not cleared, len = %d", len(stmt.Columns))
		}
		if len(stmt.Values) != 0 {
			t.Errorf("Values not cleared, len = %d", len(stmt.Values))
		}
	})

	t.Run("Put nil statement", func(t *testing.T) {
		// Should not panic
		PutInsertStatement(nil)
	})
}

// Test UpdateStatement pool
func TestUpdateStatementPool(t *testing.T) {