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():