def test_scrape_route(self): provider = CloudAmberBusRouteProvider('foo') provider._get_entity = mock.Mock(return_value='bar') StopOnRoute.objects.create = mock.Mock() provider._scrape_route( 6, 'molly/apps/places/tests/data/cloudamber-route.html') self.assertEqual(StopOnRoute.objects.create.call_count, 18)
def test_scrape_route(self): provider = CloudAmberBusRouteProvider('foo') provider._get_entity = mock.Mock(return_value='bar') StopOnRoute.objects.create = mock.Mock() provider._scrape_route(6, 'molly/apps/places/tests/data/cloudamber-route.html') self.assertEqual(StopOnRoute.objects.create.call_count, 18)