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_search(self): provider = CloudAmberBusRouteProvider('foo') provider.url = 'molly/apps/places/tests/data/cloudamber-search.html' provider._scrape_route = mock.Mock() Route.objects.get_or_create = mock.Mock(return_value=[mock.Mock(), True]) provider._scrape_search() # Attempts to create 200 routes self.assertEqual(Route.objects.get_or_create.call_count, 201)
def test_scrape_search(self): provider = CloudAmberBusRouteProvider('foo') provider.url = 'molly/apps/places/tests/data/cloudamber-search.html' provider._scrape_route = mock.Mock() Route.objects.get_or_create = mock.Mock( return_value=[mock.Mock(), True]) provider._scrape_search() # Attempts to create 200 routes self.assertEqual(Route.objects.get_or_create.call_count, 201)
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)