From 5dcdb5e9f6f8b56aac25f48dc0948f0d0c4b1151 Mon Sep 17 00:00:00 2001
From: "M.Amin Rayej" <m.amin.rayej@gmail.com>
Date: Sat, 18 Jan 2025 23:51:34 +0330
Subject: [PATCH] respect the required rt feature

---
 tokio/src/macros/select.rs   | 2 +-
 tokio/src/macros/support.rs  | 7 +++++++
 tokio/tests/macros_select.rs | 2 +-
 3 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/tokio/src/macros/select.rs b/tokio/src/macros/select.rs
index f100b0296a3..bfbeed3a066 100644
--- a/tokio/src/macros/select.rs
+++ b/tokio/src/macros/select.rs
@@ -526,7 +526,7 @@ doc! {macro_rules! select {
 
                 // Return `Pending` when the task budget is depleted since budget-aware futures
                 // are going to yield anyway and other futures will not cooperate.
-                ::std::task::ready!($crate::task::poll_budget_available(cx));
+                ::std::task::ready!($crate::macros::support::poll_budget_available(cx));
 
                 for i in 0..BRANCHES {
                     let branch;
diff --git a/tokio/src/macros/support.rs b/tokio/src/macros/support.rs
index 8588f75c323..43014d3eaf6 100644
--- a/tokio/src/macros/support.rs
+++ b/tokio/src/macros/support.rs
@@ -7,6 +7,13 @@ cfg_macros! {
     pub fn thread_rng_n(n: u32) -> u32 {
         crate::runtime::context::thread_rng_n(n)
     }
+
+    pub fn poll_budget_available(cx: &mut std::task::Context<'_>) -> std::task::Poll<()> {
+        #[cfg(feature = "rt")]
+        { crate::task::poll_budget_available(cx) }
+        #[cfg(not(feature = "rt"))]
+        { std::task::Poll::Ready(()) }
+    }
 }
 
 pub use std::future::{Future, IntoFuture};
diff --git a/tokio/tests/macros_select.rs b/tokio/tests/macros_select.rs
index 2cfd336436a..769fbd5fcf5 100644
--- a/tokio/tests/macros_select.rs
+++ b/tokio/tests/macros_select.rs
@@ -630,7 +630,7 @@ async fn mut_ref_patterns() {
     };
 }
 
-#[maybe_tokio_test]
+#[tokio::test]
 async fn select_is_budget_aware() {
     const BUDGET: usize = 128;