Function TestDeleteStatementPool

pkg/sql/ast/pool_test.go:112–144

Test DeleteStatement pool

(t *testing.T)

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

		// Use it
		stmt.TableName = "users"
		stmt.Where = &BinaryExpression{
			Left:     &Identifier{Name: "id"},
			Operator: "=",
			Right:    &LiteralValue{Value: "10"},
		}

		// Return to pool
		PutDeleteStatement(stmt)

		// Verify it was cleaned
		if stmt.TableName != "" {
			t.Errorf("TableName not cleared, got %v", stmt.TableName)
		}
		if stmt.Where != nil {
			t.Errorf("Where not cleared, got %v", stmt.Where)
		}
	})

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

// Test UpdateExpression pool
func TestUpdateExpressionPool(t *testing.T) {