| 1 | from core.user_mode_registry import UserModeRegistry |
| 2 | from core.user_modes.collect_mix_strategy import CollectMixUserModeStrategy |
| 3 | from core.user_modes.collect_strategy import CollectUserModeStrategy |
| 4 | from core.user_modes.like_strategy import LikeUserModeStrategy |
| 5 | from core.user_modes.mix_strategy import MixUserModeStrategy |
| 6 | from core.user_modes.music_strategy import MusicUserModeStrategy |
| 7 | from core.user_modes.post_strategy import PostUserModeStrategy |
| 8 | |
| 9 | |
| 10 | def test_user_mode_registry_contains_default_modes(): |
| 11 | registry = UserModeRegistry() |
| 12 | |
| 13 | assert registry.get("post") is PostUserModeStrategy |
| 14 | assert registry.get("like") is LikeUserModeStrategy |
| 15 | assert registry.get("mix") is MixUserModeStrategy |
| 16 | assert registry.get("music") is MusicUserModeStrategy |
| 17 | assert registry.get("collect") is CollectUserModeStrategy |
| 18 | assert registry.get("collectmix") is CollectMixUserModeStrategy |
| 19 | assert registry.get("unknown") is None |
| 20 |