Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/agent/test_gemini_fast_fallback.py

@@ -0,0 +1,36 @@
+"""Regression tests for #11314 — credential-pool rotation vs. fallback.
+
+_pool_may_recover_from_rate_limit() is the hinge between credential-pool
+rotation and fallback-provider activation.  Rotation is only worth waiting on
+when the pool exists, has an available entry, and has more than one entry to
+rotate to; otherwise we should fall back to the configured fallback provider
+immediately.
+"""
+import inspect
+from unittest.mock import MagicMock
+
+from agent import conversation_loop
+from run_agent import _pool_may_recover_from_rate_limit
+
+
+def _pool(entries: int = 2):
+    p = MagicMock()
+    p.has_available.return_value = True
+    p.entries.return_value = list(range(entries))
+    return p
+
+
+def test_multi_entry_pool_recovers():
+    assert _pool_may_recover_from_rate_limit(_pool(entries=3)) is True
+
+
+
+
+def test_exhausted_pool_skips_rotation():
+    p = MagicMock()
+    p.has_available.return_value = False
+    assert _pool_may_recover_from_rate_limit(p) is False
+
+
+
+