mirror of
https://github.com/NomaDamas/k-skill.git
synced 2026-06-24 02:04:11 +00:00
2550d19974
Move direct API calls out of the skill Python scripts and into shared proxy routes so end-users no longer need their own DATA_GO_KR_API_KEY for MFDS surfaces. Adds mfds.js helper, proxy tests, and updates all docs and setup guidance to reflect the hosted proxy workflow. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
104 lines
4.4 KiB
Python
104 lines
4.4 KiB
Python
import unittest
|
|
|
|
from scripts.mfds_food_safety import (
|
|
build_food_interview,
|
|
filter_food_items,
|
|
normalize_food_recall_row,
|
|
normalize_improper_food_item,
|
|
resolve_proxy_base_url,
|
|
search_food_safety,
|
|
)
|
|
|
|
|
|
class FoodInterviewTest(unittest.TestCase):
|
|
def test_build_food_interview_requires_symptom_followup_and_red_flags(self):
|
|
interview = build_food_interview(
|
|
question="이 김밥 먹어도 되나요?",
|
|
symptoms="복통과 설사",
|
|
)
|
|
|
|
self.assertEqual(interview["domain"], "food")
|
|
self.assertTrue(any("언제" in item for item in interview["must_ask"]))
|
|
self.assertTrue(any("얼마나" in item for item in interview["must_ask"]))
|
|
self.assertTrue(any("기저질환" in item for item in interview["must_ask"]))
|
|
self.assertTrue(any("알레르기" in item for item in interview["must_ask"]))
|
|
self.assertTrue(any("혈변" in item for item in interview["red_flags"]))
|
|
self.assertTrue(any("탈수" in item for item in interview["red_flags"]))
|
|
self.assertIn("응급실", interview["urgent_action"])
|
|
|
|
|
|
class FoodNormalizationTest(unittest.TestCase):
|
|
def test_normalize_food_recall_row_keeps_official_recall_fields(self):
|
|
item = normalize_food_recall_row(
|
|
{
|
|
"PRDLST_NM": "맛있는김밥",
|
|
"BSSH_NM": "예시식품",
|
|
"RTRVLPRVNS": "대장균 기준 규격 부적합",
|
|
"CRET_DTM": "2026-04-07 18:03:56.058442",
|
|
"DISTBTMLMT": "2027-12-18",
|
|
"PRDLST_TYPE": "가공식품",
|
|
}
|
|
)
|
|
|
|
self.assertEqual(item["source"], "foodsafetykorea_recall")
|
|
self.assertEqual(item["product_name"], "맛있는김밥")
|
|
self.assertEqual(item["company_name"], "예시식품")
|
|
self.assertIn("대장균", item["reason"])
|
|
self.assertEqual(item["category"], "가공식품")
|
|
|
|
def test_normalize_improper_food_item_keeps_official_improper_food_fields(self):
|
|
item = normalize_improper_food_item(
|
|
{
|
|
"PRDUCT": "예시 유부초밥",
|
|
"ENTRPS": "예시푸드",
|
|
"IMPROPT_ITM": "황색포도상구균",
|
|
"INSPCT_RESULT": "기준 부적합",
|
|
"FOOD_TY": "즉석조리식품",
|
|
"REGIST_DT": "2026-04-08",
|
|
}
|
|
)
|
|
|
|
self.assertEqual(item["source"], "mfds_improper_food")
|
|
self.assertEqual(item["product_name"], "예시 유부초밥")
|
|
self.assertEqual(item["company_name"], "예시푸드")
|
|
self.assertIn("황색포도상구균", item["reason"])
|
|
|
|
def test_filter_food_items_matches_product_and_company_names(self):
|
|
items = [
|
|
{"product_name": "맛있는김밥", "company_name": "예시식품"},
|
|
{"product_name": "사과주스", "company_name": "김밥나라"},
|
|
]
|
|
|
|
by_product = filter_food_items(items, "김밥")
|
|
by_company = filter_food_items(items, "나라")
|
|
|
|
self.assertEqual(len(by_product), 1)
|
|
self.assertEqual(by_product[0]["product_name"], "맛있는김밥")
|
|
self.assertEqual(len(by_company), 1)
|
|
self.assertEqual(by_company[0]["company_name"], "김밥나라")
|
|
|
|
|
|
class ProxyResolutionTest(unittest.TestCase):
|
|
def test_resolve_proxy_base_url_defaults_to_hosted_proxy(self):
|
|
self.assertEqual(resolve_proxy_base_url(None, env={}), "https://k-skill-proxy.nomadamas.org")
|
|
self.assertEqual(resolve_proxy_base_url(None, env={"KSKILL_PROXY_BASE_URL": "https://proxy.example.com/"}), "https://proxy.example.com")
|
|
with self.assertRaisesRegex(ValueError, "KSKILL_PROXY_BASE_URL"):
|
|
resolve_proxy_base_url(None, env={"KSKILL_PROXY_BASE_URL": "off"})
|
|
|
|
def test_search_food_safety_uses_proxy_route(self):
|
|
captured = {}
|
|
|
|
def fake_request_json(request):
|
|
captured["url"] = request.full_url
|
|
return {"items": [], "warnings": []}
|
|
|
|
payload = search_food_safety("김밥", limit=4, base_url="https://proxy.example.com", request_json=fake_request_json)
|
|
|
|
self.assertEqual(payload, {"items": [], "warnings": []})
|
|
self.assertIn("https://proxy.example.com/v1/mfds/food-safety/search", captured["url"])
|
|
self.assertIn("query=%EA%B9%80%EB%B0%A5", captured["url"])
|
|
self.assertIn("limit=4", captured["url"])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|