def test_data(self):
sc = Scraper()
sc.store_data(INCIDENTS[0])
self.assertEqual("F150020627", Incident.objects.all()[0].incident_id)
self.assertEqual("2717 Dexter Av N", Incident.objects.all()[0].location_text)
self.assertEqual("Medic Response", Incident.objects.all()[0].type.type_name)
self.assertEqual("E9", Vehicle.objects.get(id=1).name)
def test_followup_dispatch(self):
sc = Scraper()
sc.store_data(INCIDENTS[5])
self.assertEqual(1, len(Incident.objects.all()))
self.assertEqual(1, len(Dispatch.objects.all()))
sc.store_data(INCIDENTS[0])
self.assertEqual(1, len(Incident.objects.all()))
self.assertEqual(2, len(Dispatch.objects.all()))
def test_closed(self):
sc = Scraper()
sc.store_data(INCIDENTS[1])
self.assertEqual(0, len(Incident.objects.all()))
def test_multi_vehic(self):
sc = Scraper()
sc.store_data(INCIDENTS[6])
self.assertEqual(2, len(Vehicle.objects.all()))
def test_dupe_vehic(self):
sc = Scraper()
sc.store_data(INCIDENTS[0])
self.assertEqual(1, len(Vehicle.objects.all()))
sc.store_data(INCIDENTS[2])
self.assertEqual(1, len(Vehicle.objects.all()))
def test_open_close(self):
sc = Scraper()
sc.store_data(INCIDENTS[0])
self.assertIsNone(Incident.objects.all()[0].end)
sc.store_data(INCIDENTS[1])
self.assertIsNotNone(Incident.objects.all()[0].end)
def test_no_dupes(self):
sc = Scraper()
sc.store_data(INCIDENTS[0])
self.assertEqual(1, len(Incident.objects.all()))
sc.store_data(INCIDENTS[0])
self.assertEqual(1, len(Incident.objects.all()))
