Add vehicle color parsing
This commit is contained in:
@@ -126,6 +126,42 @@ class TestEncarMapper(unittest.TestCase):
|
||||
with self.subTest(raw=raw):
|
||||
self.assertEqual(mapper._translate_model(raw), expected)
|
||||
|
||||
def test_translate_color_normalizes_api_spacing(self):
|
||||
mapper = EncarMapper()
|
||||
|
||||
self.assertEqual(mapper._translate_color("흰색"), "white")
|
||||
self.assertEqual(mapper._translate_color("흰 색"), "white")
|
||||
self.assertEqual(mapper._translate_color("검정색"), "black")
|
||||
self.assertEqual(mapper._translate_color("진주색"), "pearl")
|
||||
self.assertEqual(mapper._translate_color("쥐색"), "mouse gray")
|
||||
self.assertEqual(mapper._translate_color("은회색"), "silver-gray")
|
||||
self.assertEqual(mapper._translate_color("하늘색"), "sky blue")
|
||||
self.assertEqual(mapper._translate_color("자주색"), "burgundy")
|
||||
|
||||
def test_apply_batch_color_uses_spec_color_name(self):
|
||||
payload = {
|
||||
"Id": 42456458,
|
||||
"Manufacturer": "Lincoln",
|
||||
"Model": "Corsair",
|
||||
"FormYear": "2022",
|
||||
"Price": 2690,
|
||||
"Mileage": 26600,
|
||||
}
|
||||
|
||||
mapper = EncarMapper()
|
||||
record = mapper.map_to_car_record(
|
||||
ENCAR_DETAIL_URL_TEMPLATE.format(vehicle_id="42456458"),
|
||||
payload,
|
||||
)
|
||||
mapper.apply_batch_color(record, {"spec": {"colorName": "흰 색"}})
|
||||
|
||||
self.assertEqual(record.brand, "Lincoln")
|
||||
self.assertEqual(record.model, "Corsair")
|
||||
self.assertEqual(record.year, 2022)
|
||||
self.assertEqual(record.price, 26900000)
|
||||
self.assertEqual(record.mileage, 26600)
|
||||
self.assertEqual(record.color, "white")
|
||||
|
||||
|
||||
class TestEncarScraper(unittest.TestCase):
|
||||
def test_collect_listing_limits_results(self):
|
||||
|
||||
Reference in New Issue
Block a user