Function TestUpdateStatementPool

pkg/sql/ast/pool_test.go:68–109

Test UpdateStatement pool

(t *testing.T)

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

		// Use it
		stmt.TableName = "users"
		stmt.Assignments = []UpdateExpression{
			{
				Column: &Identifier{Name: "email"},
				Value:  &LiteralValue{Value: "new@example.com"},
			},
		}
		stmt.Where = &BinaryExpression{
			Left:     &Identifier{Name: "id"},
			Operator: "=",
			Right:    &LiteralValue{Value: "1"},
		}

		// Return to pool
		PutUpdateStatement(stmt)

		// Verify it was cleaned
		if stmt.TableName != "" {
			t.Errorf("TableName not cleared, got %v", stmt.TableName)
		}
		if len(stmt.Assignments) != 0 {
			t.Errorf("Updates not cleared, len = %d", len(stmt.Assignments))
		}
		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
		PutUpdateStatement(nil)
	})
}

// Test DeleteStatement pool
func TestDeleteStatementPool(t *testing.T) {