From a617b0eda2ef0578055ed455dbc9c73368a0e0e4 Mon Sep 17 00:00:00 2001
From: Maximellerbach <maxime.ellerbach@huggingface.co>
Date: Wed, 3 Jun 2026 16:52:44 +0200
Subject: [PATCH] adding legacy to existing tests

---
 tests/test_rollout.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tests/test_rollout.py b/tests/test_rollout.py
index 5a1ec4703..0edcb0a33 100644
--- a/tests/test_rollout.py
+++ b/tests/test_rollout.py
@@ -60,6 +60,7 @@ def test_strategy_config_types():
         BaseStrategyConfig,
         DAggerStrategyConfig,
         HighlightStrategyConfig,
+        LegacyStrategyConfig,
         SentryStrategyConfig,
     )
 
@@ -67,6 +68,7 @@ def test_strategy_config_types():
     assert SentryStrategyConfig().type == "sentry"
     assert HighlightStrategyConfig().type == "highlight"
     assert DAggerStrategyConfig().type == "dagger"
+    assert LegacyStrategyConfig().type == "legacy"
 
 
 def test_dagger_config_invalid_input_device():
@@ -203,6 +205,8 @@ def test_create_strategy_dispatches():
         BaseStrategyConfig,
         DAggerStrategy,
         DAggerStrategyConfig,
+        LegacyStrategy,
+        LegacyStrategyConfig,
         SentryStrategy,
         SentryStrategyConfig,
         create_strategy,
@@ -211,6 +215,7 @@ def test_create_strategy_dispatches():
     assert isinstance(create_strategy(BaseStrategyConfig()), BaseStrategy)
     assert isinstance(create_strategy(SentryStrategyConfig()), SentryStrategy)
     assert isinstance(create_strategy(DAggerStrategyConfig()), DAggerStrategy)
+    assert isinstance(create_strategy(LegacyStrategyConfig()), LegacyStrategy)
 
 
 def test_create_strategy_unknown_raises():