Fixed: https://github.com/BeehiveInnovations/zen-mcp-server/issues/123

Fixed restriction checks for OpenRouter

tests/test_parse_model_option.py

@@ -0,0 +1,79 @@
+"""Tests for parse_model_option function."""
+
+from server import parse_model_option
+
+
+class TestParseModelOption:
+    """Test cases for model option parsing."""
+
+    def test_openrouter_free_suffix_preserved(self):
+        """Test that OpenRouter :free suffix is preserved as part of model name."""
+        model, option = parse_model_option("openai/gpt-3.5-turbo:free")
+        assert model == "openai/gpt-3.5-turbo:free"
+        assert option is None
+
+    def test_openrouter_beta_suffix_preserved(self):
+        """Test that OpenRouter :beta suffix is preserved as part of model name."""
+        model, option = parse_model_option("anthropic/claude-3-opus:beta")
+        assert model == "anthropic/claude-3-opus:beta"
+        assert option is None
+
+    def test_openrouter_preview_suffix_preserved(self):
+        """Test that OpenRouter :preview suffix is preserved as part of model name."""
+        model, option = parse_model_option("google/gemini-pro:preview")
+        assert model == "google/gemini-pro:preview"
+        assert option is None
+
+    def test_ollama_tag_parsed_as_option(self):
+        """Test that Ollama tags are parsed as options."""
+        model, option = parse_model_option("llama3.2:latest")
+        assert model == "llama3.2"
+        assert option == "latest"
+
+    def test_consensus_stance_parsed_as_option(self):
+        """Test that consensus stances are parsed as options."""
+        model, option = parse_model_option("o3:for")
+        assert model == "o3"
+        assert option == "for"
+
+        model, option = parse_model_option("gemini-2.5-pro:against")
+        assert model == "gemini-2.5-pro"
+        assert option == "against"
+
+    def test_openrouter_unknown_suffix_parsed_as_option(self):
+        """Test that unknown suffixes on OpenRouter models are parsed as options."""
+        model, option = parse_model_option("openai/gpt-4:custom-tag")
+        assert model == "openai/gpt-4"
+        assert option == "custom-tag"
+
+    def test_plain_model_name(self):
+        """Test plain model names without colons."""
+        model, option = parse_model_option("gpt-4")
+        assert model == "gpt-4"
+        assert option is None
+
+    def test_url_not_parsed(self):
+        """Test that URLs are not parsed for options."""
+        model, option = parse_model_option("http://localhost:8080")
+        assert model == "http://localhost:8080"
+        assert option is None
+
+    def test_whitespace_handling(self):
+        """Test that whitespace is properly stripped."""
+        model, option = parse_model_option("  openai/gpt-3.5-turbo:free  ")
+        assert model == "openai/gpt-3.5-turbo:free"
+        assert option is None
+
+        model, option = parse_model_option("  llama3.2 : latest  ")
+        assert model == "llama3.2"
+        assert option == "latest"
+
+    def test_case_insensitive_suffix_matching(self):
+        """Test that OpenRouter suffix matching is case-insensitive."""
+        model, option = parse_model_option("openai/gpt-3.5-turbo:FREE")
+        assert model == "openai/gpt-3.5-turbo:FREE"  # Original case preserved
+        assert option is None
+
+        model, option = parse_model_option("openai/gpt-3.5-turbo:Free")
+        assert model == "openai/gpt-3.5-turbo:Free"  # Original case preserved
+        assert option is None