class CdsBusArrivalTestCases(unittest.TestCase):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
logging.basicConfig(format='%(asctime)s - %(levelname)s [%(filename)s:%(lineno)s %(funcName)20s] %(message)s',
level=logging.INFO,
handlers=[logging.StreamHandler()])
self.logger = logging.getLogger("vrnbus")
self.tracker = EventTracker(logger, [])
self.mock_provider = CdsTestDataProvider(logger)
self.cds = CdsRequest(logger, self.mock_provider)
self.processor = WebDataProcessor(self.cds, self.logger, self.tracker)
def test_arrival(self):
result = self.processor.get_arrival("про 49 5а", 51.692727, 39.18297)
stops = result['bus_stops']
counts = 0
for k, v in stops.items():
self.logger.info(k)
self.logger.info(v)
counts += len(v.split('\n'))
break
self.logger.info(result)
self.logger.info(counts)
def test_businfo(self):
result = self.processor.get_bus_info("про 49 5а", 51.692727, 39.18297, True)
self.logger.info(result)
def test_arrival_distance(self):
src = 'Центральный автовокзал (в центр)'
dst = 'Площадь Застава (в центр)'
wrong_direction = self.cds.get_dist("27", dst, src)
right_direction = self.cds.get_dist("27", src, dst)
self.assertTrue(wrong_direction == 0)
self.assertGreater(right_direction, 0)
def test_arrival_by_id(self):
for i in range(500):
result = self.processor.get_arrival_by_id("pro", i)
self.assertTrue(result['arrival_info']['found'])
class CdsRouteTestCase(unittest.TestCase):
def __init__(self, *args, **kwargs):
super(CdsRouteTestCase, self).__init__(*args, **kwargs)
self.mock_provider = CdsTestDataProvider(logger)
self.cds = CdsRequest(logger, self.mock_provider)
self.date_time = datetime.datetime(2018, 2, 15, 19, 56, 53)
def test_routes_on_bus_stop(self):
result = self.cds.get_routes_on_bus_stop(57)
self.assertTrue(result)
def test_bus_stop_distance(self):
route_name = "5А"
stop_1 = "у-м Молодежный (ул. Лизюкова в центр)"
stop_2 = "ул. Лизюкова (ул. Жукова в центр)"
with self.subTest('Normal bus station order'):
result = self.cds.get_dist(route_name, stop_1, stop_2)
self.assertTrue(result)
with self.subTest('Reverse bus station order'):
result = self.cds.get_dist(route_name, stop_2, stop_1)
self.assertFalse(result)
def test_closest_bus_stop_checked(self):
route_name = '5А'
pos_1 = CdsBusPosition(51.705497, 39.149543, self.date_time) # у-м Молодёжный
pos_2 = CdsBusPosition(51.705763, 39.155278, self.date_time) # 60 лет ВЛКСМ
with self.subTest('From city center '):
result = self.cds.get_closest_bus_stop_checked(route_name, (pos_2, pos_1))
self.assertTrue(result.NAME_ == 'у-м Молодежный (ул. Лизюкова из центра)')
self.assertEqual(result.NUMBER_, 62)
with self.subTest('To city center '):
result = self.cds.get_closest_bus_stop_checked(route_name, (pos_1, pos_2))
self.assertEqual(result.NUMBER_, 5)
def test_closest_bus_stop_same_stations(self):
positions = [CdsBusPosition(51.667033, 39.193648, self.date_time),
CdsBusPosition(51.672135, 39.187541, self.date_time),
CdsBusPosition(51.675065, 39.185286, self.date_time),
CdsBusPosition(51.677922, 39.184953, self.date_time),
CdsBusPosition(51.677922, 39.184953, self.date_time),
CdsBusPosition(51.680843, 39.184798, self.date_time)]
result = self.cds.get_closest_bus_stop_checked("90", positions)
self.assertTrue(result.NUMBER_ == 40)
self.assertTrue(result.NAME_ == 'Проспект Труда (Московский проспект из центра)')
def test_closest_bus_stop(self):
route_bus = CdsRouteBus.make(*[
51.625537, 39.177478,
16,
"2018-02-15T19:57:47",
"М617АК136",
834,
20,
"80",
0,
"2018-02-15T19:54:56",
"Рабочий проспект (из центра)"
])
station = self.cds.get_closest_bus_stop(route_bus)
logger.info(f"{station}; {route_bus.distance_km(station):.4f} {route_bus.distance(station):.4f}")