Function TestDropStatementPool

pkg/sql/ast/pool_ddl_test.go:519–566

============================================================ DropStatement pool tests ============================================================

(t *testing.T)

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

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

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

	t.Run("Fields zeroed after Put", func(t *testing.T) {
		stmt := GetDropStatement()
		stmt.ObjectType = "TABLE"
		stmt.IfExists = true
		stmt.Names = append(stmt.Names, "users", "orders")
		stmt.CascadeType = "CASCADE"

		PutDropStatement(stmt)

		if stmt.ObjectType != "" {
			t.Errorf("ObjectType not cleared, got %q", stmt.ObjectType)
		}
		if stmt.IfExists {
			t.Error("IfExists not cleared")
		}
		if len(stmt.Names) != 0 {
			t.Errorf("Names not cleared, len=%d", len(stmt.Names))
		}
		if stmt.CascadeType != "" {
			t.Errorf("CascadeType not cleared, got %q", stmt.CascadeType)
		}
	})

	t.Run("Pool roundtrip reuse", func(t *testing.T) {
		stmt1 := GetDropStatement()
		stmt1.ObjectType = "VIEW"
		PutDropStatement(stmt1)

		stmt2 := GetDropStatement()
		if stmt2.ObjectType != "" {
			t.Errorf("Reused statement not clean, ObjectType=%q", stmt2.ObjectType)
		}
		PutDropStatement(stmt2)
	})
}

// ============================================================
// TruncateStatement pool tests