from __future__ import annotations

from search_yaoex import YaoexAuth, build_yaoex_search_request
from search_ybm import build_ybm_search_payload


def test_yaoex_keyword_is_the_query_key():
    auth = YaoexAuth(token="t", user_id="u", buyer_code="b")
    _, payload = build_yaoex_search_request(auth, "06901234567890")
    assert payload["keyword"] == "06901234567890"


def test_ybm_query_word_is_the_query_key():
    payload = build_ybm_search_payload("06901234567890")
    assert payload["queryWord"] == "06901234567890"