def test_get_incident_status_team(self):
     """Get incident status summary with team_id.
     """
     t1 = factories.TeamFactory.create()
     t2 = factories.TeamFactory.create()
     e1 = factories.EventFactory.create(team=t1)
     e2 = factories.EventFactory.create(team=t2)
     t1_incidents = []
     t2_incidents = []
     [t1_incidents.append(factories.IncidentFactory.create(event=e1)) for i in range(10)]
     [t2_incidents.append(factories.IncidentFactory.create(event=e2)) for i in range(10)]
     i = incidents.get_incident_stats()
     # Make sure we have 20 active incidents
     self.assertEquals(i['active'], 20)
     self.assertEquals(Incident.objects.filter().count(), 20)
     # Lets change the status of a few incidents by team
     t2_incidents[0].status = 'Acknowledged'
     t2_incidents[1].status = 'Acknowledged'
     t2_incidents[2].status = 'Acknowledged'
     t2_incidents[3].status = 'Cleared'
     t2_incidents[4].status = 'Cleared'
     t2_incidents[5].status = 'Cleared'
     [t2_incidents[i].save() for i in range(10)]
     i = incidents.get_incident_stats(team_id=t2.id)
     self.assertEquals(i['active'], 4, msg="Team (t2) incidents do not validate after update")
     self.assertEquals(i['acknowledged'], 3)
     self.assertEquals(i['cleared'], 3)
 def test_get_incident_status_all(self):
     """Get incident status summary for all incidents.
     """
     test_incidents = []
     [test_incidents.append(factories.IncidentFactory.create()) for i in range(10)]
     i = incidents.get_incident_stats()
     # Make sure we have 10 active incidents
     self.assertEquals(i['active'], 10)
     self.assertEquals(Incident.objects.filter().count(), 10)
     # Lets change the status of a few incidents
     test_incidents[0].status = 'Acknowledged'
     test_incidents[1].status = 'Acknowledged'
     test_incidents[2].status = 'Acknowledged'
     test_incidents[3].status = 'Cleared'
     test_incidents[4].status = 'Cleared'
     test_incidents[5].status = 'Cleared'
     [test_incidents[i].save() for i in range(10)]
     i = incidents.get_incident_stats()
     self.assertEquals(i['active'], 4)
     self.assertEquals(i['acknowledged'], 3)
     self.assertEquals(i['cleared'], 3)
 def test_get_incident_status_team(self):
     """Get incident status summary with team_id.
     """
     t1 = factories.TeamFactory.create()
     t2 = factories.TeamFactory.create()
     e1 = factories.EventFactory.create(team=t1)
     e2 = factories.EventFactory.create(team=t2)
     t1_incidents = []
     t2_incidents = []
     [
         t1_incidents.append(factories.IncidentFactory.create(event=e1))
         for i in range(10)
     ]
     [
         t2_incidents.append(factories.IncidentFactory.create(event=e2))
         for i in range(10)
     ]
     i = incidents.get_incident_stats()
     # Make sure we have 20 active incidents
     self.assertEquals(i['active'], 20)
     self.assertEquals(Incident.objects.filter().count(), 20)
     # Lets change the status of a few incidents by team
     t2_incidents[0].status = 'Acknowledged'
     t2_incidents[1].status = 'Acknowledged'
     t2_incidents[2].status = 'Acknowledged'
     t2_incidents[3].status = 'Cleared'
     t2_incidents[4].status = 'Cleared'
     t2_incidents[5].status = 'Cleared'
     [t2_incidents[i].save() for i in range(10)]
     i = incidents.get_incident_stats(team_id=t2.id)
     self.assertEquals(
         i['active'],
         4,
         msg="Team (t2) incidents do not validate after update")
     self.assertEquals(i['acknowledged'], 3)
     self.assertEquals(i['cleared'], 3)
 def test_get_incident_status_all(self):
     """Get incident status summary for all incidents.
     """
     test_incidents = []
     [
         test_incidents.append(factories.IncidentFactory.create())
         for i in range(10)
     ]
     i = incidents.get_incident_stats()
     # Make sure we have 10 active incidents
     self.assertEquals(i['active'], 10)
     self.assertEquals(Incident.objects.filter().count(), 10)
     # Lets change the status of a few incidents
     test_incidents[0].status = 'Acknowledged'
     test_incidents[1].status = 'Acknowledged'
     test_incidents[2].status = 'Acknowledged'
     test_incidents[3].status = 'Cleared'
     test_incidents[4].status = 'Cleared'
     test_incidents[5].status = 'Cleared'
     [test_incidents[i].save() for i in range(10)]
     i = incidents.get_incident_stats()
     self.assertEquals(i['active'], 4)
     self.assertEquals(i['acknowledged'], 3)
     self.assertEquals(i['cleared'], 3)