63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
import unittest
|
||
from unittest.mock import MagicMock
|
||
|
||
from encar_scraper.encar import EncarMapper, EncarScraper, ENCAR_DETAIL_URL_TEMPLATE
|
||
|
||
|
||
class TestEncarMapper(unittest.TestCase):
|
||
def test_map_to_car_record(self):
|
||
payload = {
|
||
"Id": 41421262,
|
||
"Manufacturer": "Hyundai",
|
||
"Model": "G90 (RS4)",
|
||
"FormYear": "2023",
|
||
"Price": 8590,
|
||
"Mileage": 37923,
|
||
"Photos": [
|
||
{"location": "/carpicture01/pic4141/41412663_001.jpg", "ordering": 1},
|
||
{"location": "/carpicture01/pic4141/41412663_003.jpg", "ordering": 3},
|
||
],
|
||
}
|
||
|
||
record = EncarMapper().map_to_car_record(
|
||
ENCAR_DETAIL_URL_TEMPLATE.format(vehicle_id="41421262"),
|
||
payload,
|
||
)
|
||
|
||
self.assertEqual(record.origin_id, "encar:41421262")
|
||
self.assertEqual(record.parser_id, "encar:41421262")
|
||
self.assertEqual(record.brand, "Hyundai")
|
||
self.assertEqual(record.model, "G90 (RS4)")
|
||
self.assertEqual(record.year, 2023)
|
||
self.assertEqual(record.price, 85900000) # 8590 만원 × 10000
|
||
self.assertEqual(record.mileage, 37923)
|
||
self.assertEqual(record.country, "KR")
|
||
self.assertEqual(record.currency, "KRW")
|
||
self.assertEqual(len(record.images), 2)
|
||
self.assertTrue(record.images[0].fullres_image.endswith("_001.jpg"))
|
||
|
||
def test_extract_vehicle_id_from_url(self):
|
||
url = ENCAR_DETAIL_URL_TEMPLATE.format(vehicle_id="41421262")
|
||
self.assertEqual(EncarMapper()._extract_vehicle_id(url), "41421262")
|
||
|
||
|
||
class TestEncarScraper(unittest.TestCase):
|
||
def test_collect_listing_limits_results(self):
|
||
scraper = EncarScraper()
|
||
scraper._fetch_listing_page = MagicMock(side_effect=[
|
||
{"SearchResults": [{"Id": 1}, {"Id": 2}]},
|
||
])
|
||
|
||
result = scraper.collect_listing(limit=1, page_size=2)
|
||
self.assertEqual(result["items_collected"], 1)
|
||
self.assertEqual(result["vehicle_urls"], [ENCAR_DETAIL_URL_TEMPLATE.format(vehicle_id="1")])
|
||
|
||
def test_scrape_vehicle_detail_raises_without_id(self):
|
||
scraper = EncarScraper()
|
||
with self.assertRaises(ValueError):
|
||
scraper.scrape_vehicle_detail("https://www.encar.com/invalid")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|