import unittest
from real_estate_agent.scraper import build_queries, extract_price, parse_bing_rss_results, parse_duckduckgo_results, score_result
class ScraperTests(unittest.TestCase):
def test_build_queries_includes_montpellier_ring_towns(self):
queries = build_queries(center="Montpellier", radius_km=20, max_price=400000, property_type="maison")
self.assertGreaterEqual(len(queries), 1)
joined = "\n".join(queries)
self.assertIn("Montpellier", joined)
self.assertIn("Castelnau-le-Lez", joined)
self.assertIn("Lattes", joined)
self.assertIn("maison à vendre", joined)
def test_parse_duckduckgo_results_extracts_title_link_and_snippet(self):
html = """
Maison Montpellier 340000
Belle maison 4 pièces, jardin, Montpellier.
Villa proche Lattes
400 000 € max, 15 km de Montpellier.
"""
results = parse_duckduckgo_results(html)
self.assertEqual(len(results), 2)
self.assertEqual(results[0]["title"], "Maison Montpellier 340000")
self.assertEqual(results[0]["url"], "https://example.com/listing1")
self.assertIn("jardin", results[0]["snippet"])
self.assertEqual(results[1]["title"], "Villa proche Lattes")
self.assertEqual(results[1]["url"], "https://example.com/listing2")
def test_parse_bing_rss_results_extracts_entries(self):
xml = """
-
Maison à vendre Montpellier
https://example.com/1
Maison 390 000 € près de Castelnau-le-Lez
-
Villa à vendre Lattes
https://example.com/2
400k€ max
"""
results = parse_bing_rss_results(xml)
self.assertEqual(len(results), 2)
self.assertEqual(results[0]["title"], "Maison à vendre Montpellier")
self.assertEqual(results[0]["url"], "https://example.com/1")
self.assertIn("Castelnau", results[0]["snippet"])
def test_extract_price_supports_formats(self):
self.assertEqual(extract_price("Maison à 399 000 € avec jardin"), 399000)
self.assertEqual(extract_price("Prix: 400k€"), 400000)
self.assertIsNone(extract_price("Prix sur demande"))
def test_score_result_prefers_price_and_distance_match(self):
result = {
"title": "Maison Montpellier",
"snippet": "À 12 km de Montpellier, 390 000 €",
"url": "https://example.com",
"source": "PAP",
}
score = score_result(result, center="Montpellier", radius_km=20, max_price=400000, property_type="maison")
self.assertGreater(score, 0)
if __name__ == "__main__":
unittest.main()