nomadamas__k-skill/scripts/test_daangn_realty.py

import importlib.util
import json
import pathlib
import unittest
from unittest import mock


MODULE_PATH = pathlib.Path(__file__).resolve().parents[1] / "daangn-realty-search" / "scripts" / "daangn_realty.py"
MODULE_SPEC = importlib.util.spec_from_file_location("daangn_realty", MODULE_PATH)
if MODULE_SPEC is None or MODULE_SPEC.loader is None:
    raise RuntimeError(f"Unable to load daangn_realty module from {MODULE_PATH}")
daangn_realty = importlib.util.module_from_spec(MODULE_SPEC)
MODULE_SPEC.loader.exec_module(daangn_realty)


def relay_html(store: dict) -> str:
    encoded = json.dumps(json.dumps(store, ensure_ascii=False), ensure_ascii=False)
    return f"<html><script>window.RELAY_STORE = {encoded};</script></html>"


class RelayStoreParsingTest(unittest.TestCase):
    def test_extract_relay_store_returns_none_for_malformed_payload(self):
        parsed = daangn_realty.extract_relay_store(
            '<html><script>window.RELAY_STORE = "{bad json";</script></html>'
        )

        self.assertIsNone(parsed)

    def test_extract_articles_parses_trade_prices_and_numeric_area(self):
        store = {
            "card:1": {
                "__typename": "ArticleFeedCard",
                "article": {"__ref": "article:1"},
            },
            "article:1": {
                "__typename": "Article",
                "originalId": "123",
                "area": "33.05785",
                "salesTypeV3": {"__ref": "sales:1"},
                "trades": {"__refs": ["trade:month", "trade:buy", "trade:borrow"]},
            },
            "sales:1": {"__typename": "ArticleSalesTypeV2", "type": "OFFICETEL"},
            "trade:month": {
                "__typename": "MonthTrade",
                "deposit": "1000",
                "monthlyPay": "50",
            },
            "trade:buy": {"__typename": "BuyTrade", "price": "20000"},
            "trade:borrow": {"__typename": "BorrowTrade", "deposit": "15000"},
        }

        parsed = daangn_realty.extract_relay_store(relay_html(store))
        items = daangn_realty.extract_articles(parsed, max_items=5)

        self.assertEqual(len(items), 1)
        self.assertEqual(items[0]["article_id"], "123")
        self.assertEqual(items[0]["salesType"], "OFFICETEL")
        self.assertEqual(items[0]["area_pyeong"], 10.0)
        self.assertEqual(
            [(trade["type"], trade["per_pyeong_manwon"]) for trade in items[0]["trades"]],
            [("MONTH", 5.0), ("BUY", 2000.0), ("BORROW", 1500.0)],
        )


class SearchFilteringTest(unittest.TestCase):
    def test_collect_for_region_filters_keyword_against_enriched_item_text(self):
        store = {
            "card:1": {"__typename": "ArticleFeedCard", "article": {"__ref": "article:1"}},
            "article:1": {
                "__typename": "Article",
                "originalId": "123",
                "area": "33.05785",
                "salesTypeV3": {"__ref": "sales:1"},
                "trades": {"__refs": ["trade:month"]},
            },
            "card:2": {"__typename": "ArticleFeedCard", "article": {"__ref": "article:2"}},
            "article:2": {
                "__typename": "Article",
                "originalId": "456",
                "area": "20",
                "salesTypeV3": {"__ref": "sales:1"},
                "trades": {"__refs": ["trade:month"]},
            },
            "sales:1": {"__typename": "ArticleSalesTypeV2", "type": "STORE"},
            "trade:month": {"__typename": "MonthTrade", "deposit": "1000", "monthlyPay": "50"},
        }

        with mock.patch.object(daangn_realty, "fetch_text", return_value=relay_html(store)):
            _, items, error = daangn_realty.collect_for_region(
                {"name1": "경기도", "name2": "수원시", "name3": "매교동"},
                sales_type_filter=None,
                trade_type_filter=None,
                limit=10,
                keyword="456",
            )

        self.assertIsNone(error)
        self.assertEqual([item["article_id"] for item in items], ["456"])


class DetailParsingTest(unittest.TestCase):
    def test_parse_detail_handles_missing_graph_and_extracts_floor(self):
        html = """
        <script type="application/ld+json">{"@context":"https://schema.org","@graph":null}</script>
        <script type="application/ld+json">
        {"@graph":[
          {"@type":"Product","name":"상가 월세","additionalProperty":[
            {"name":"floor","value":"8.0"},
            {"name":"topFloor","value":"10"},
            {"name":"nearbySubwayStation","value":"매교역"}
          ]},
          {"@type":"Place","name":"경기도 수원시 팔달구 매교동"}
        ]}
        </script>
        """

        with mock.patch.object(daangn_realty, "fetch_text", return_value=html):
            detail = daangn_realty.parse_detail("https://realty.daangn.com/articles/123")

        self.assertEqual(detail["title"], "상가 월세")
        self.assertEqual(detail["address"], "경기도 수원시 팔달구 매교동")
        self.assertEqual(detail["floor_label"], "8층/10층")
        self.assertEqual(detail["nearby_subway"], "매교역")


class CliCompatibilityTest(unittest.TestCase):
    def test_search_parser_keeps_keyword_and_only_verified_flags(self):
        parser = daangn_realty.build_parser()

        args = parser.parse_args(
            [
                "search",
                "--region",
                "매교동",
                "--keyword",
                "상가",
                "--only-verified",
                "--limit",
                "1",
                "--titles",
                "0",
            ]
        )

        self.assertEqual(args.keyword, "상가")
        self.assertTrue(args.only_verified)


if __name__ == "__main__":
    unittest.main()