from config import FunctionScoreConfig, IndexConfig, QueryConfig, RerankConfig, SPUConfig, SearchConfig
  from query.query_parser import QueryParser
  
  
  class _DummyTranslator:
      def translate(self, text, target_lang, source_lang, scene, model_name):
          return f"{text}-{target_lang}"
  
  

  def _tokenizer(text):
      return str(text).split()
  
  

  def _build_config() -> SearchConfig:
      return SearchConfig(
          es_index_name="test_products",
          field_boosts={"title.en": 3.0, "title.zh": 3.0},
          indexes=[IndexConfig(name="default", label="default", fields=["title.en", "title.zh"])],
          query_config=QueryConfig(
              enable_text_embedding=False,
              enable_query_rewrite=False,
              supported_languages=["en", "zh"],
              default_language="zh",
          ),
          function_score=FunctionScoreConfig(),
          rerank=RerankConfig(),
          spu_config=SPUConfig(enabled=False),
      )
  
  

  def test_parse_mixed_zh_query_translates_to_en(monkeypatch):

      parser = QueryParser(_build_config(), translator=_DummyTranslator(), tokenizer=_tokenizer)

      monkeypatch.setattr(parser.language_detector, "detect", lambda text: "zh")

      result = parser.parse(
          "法式 dress 连衣裙",
          tenant_id="162",
          generate_vector=False,
          target_languages=["zh", "en"],
      )

  
      assert result.detected_language == "zh"

      assert result.translations == {"en": "法式 dress 连衣裙-en"}
      assert result.query_tokens == ["法式", "dress", "连衣裙"]
      assert not hasattr(result, "query_text_by_lang")
      assert not hasattr(result, "search_langs")

  def test_parse_mixed_en_query_translates_to_zh(monkeypatch):

      parser = QueryParser(_build_config(), translator=_DummyTranslator(), tokenizer=_tokenizer)

      monkeypatch.setattr(parser.language_detector, "detect", lambda text: "en")

      result = parser.parse(
          "red 连衣裙",
          tenant_id="0",
          generate_vector=False,
          target_languages=["en", "zh"],
      )

  
      assert result.detected_language == "en"

      assert result.translations == {"zh": "red 连衣裙-zh"}
      assert result.query_tokens == ["red", "连衣裙"]

  
  
  def test_parse_waits_for_translation_when_source_in_index_languages(monkeypatch):
      """en 在 index_languages 内时仍应等待并采纳 en->zh 翻译结果（与向量共用预算）。"""

      parser = QueryParser(_build_config(), translator=_DummyTranslator(), tokenizer=_tokenizer)

      monkeypatch.setattr(parser.language_detector, "detect", lambda text: "en")

      result = parser.parse(
          "off shoulder top",
          tenant_id="0",
          generate_vector=False,
          target_languages=["en", "zh"],
      )

  
      assert result.detected_language == "en"

      assert result.translations.get("zh") == "off shoulder top-zh"

      assert not hasattr(result, "source_in_index_languages")