From e49aa9f0c416727d2a5a542a8e86f70c68cf9584 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Thu, 29 Oct 2026 11:11:43 -0700
Subject: [PATCH] test(proxy): give the batch tests' fake router the alias and
 routed-group lookups

---
 .../batches_endpoints/test_litellm_executed_batches.py      | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/tests/unit/proxy/batches_endpoints/test_litellm_executed_batches.py b/tests/unit/proxy/batches_endpoints/test_litellm_executed_batches.py
index dd590ac1bd3..334d636ec91 100644
--- a/tests/unit/proxy/batches_endpoints/test_litellm_executed_batches.py
+++ b/tests/unit/proxy/batches_endpoints/test_litellm_executed_batches.py
@@ -255,6 +255,12 @@ class FakeRouter:
     def get_model_list(self, model_name: str | None = None) -> None:
         return None
 
+    def get_model_from_alias(self, model: str) -> None:
+        return None
+
+    def get_model_list_of_routed_group(self, model_group: str) -> list[object]:
+        return []
+
 
 class FakeStorageBackend:
     def __init__(self, contents: Mapping[str, bytes]) -> None: