greenbone /
python-gvm
| @@ 25-58 (lines=34) @@ | ||
| 22 | from gvm.protocols.gmpv208 import AlertEvent, get_alert_event_from_string |
|
| 23 | ||
| 24 | ||
| 25 | class GetAlertEventFromStringTestCase(unittest.TestCase): |
|
| 26 | def test_invalid(self): |
|
| 27 | with self.assertRaises(InvalidArgument): |
|
| 28 | get_alert_event_from_string('foo') |
|
| 29 | ||
| 30 | def test_none_or_empty(self): |
|
| 31 | ct = get_alert_event_from_string(None) |
|
| 32 | self.assertIsNone(ct) |
|
| 33 | ct = get_alert_event_from_string('') |
|
| 34 | self.assertIsNone(ct) |
|
| 35 | ||
| 36 | def test_task_run_status_changed(self): |
|
| 37 | ct = get_alert_event_from_string('Task run status changed') |
|
| 38 | self.assertEqual(ct, AlertEvent.TASK_RUN_STATUS_CHANGED) |
|
| 39 | ||
| 40 | def test_new_secinfo_arrived(self): |
|
| 41 | ct = get_alert_event_from_string('New SecInfo arrived') |
|
| 42 | self.assertEqual(ct, AlertEvent.NEW_SECINFO_ARRIVED) |
|
| 43 | ||
| 44 | def test_updated_secinfo_arrived(self): |
|
| 45 | ct = get_alert_event_from_string('Updated SecInfo arrived') |
|
| 46 | self.assertEqual(ct, AlertEvent.UPDATED_SECINFO_ARRIVED) |
|
| 47 | ||
| 48 | def test_ticket_received(self): |
|
| 49 | ct = get_alert_event_from_string('ticket received') |
|
| 50 | self.assertEqual(ct, AlertEvent.TICKET_RECEIVED) |
|
| 51 | ||
| 52 | def test_assigned_ticket_changed(self): |
|
| 53 | ct = get_alert_event_from_string('assigned ticket changed') |
|
| 54 | self.assertEqual(ct, AlertEvent.ASSIGNED_TICKET_CHANGED) |
|
| 55 | ||
| 56 | def test_owned_ticket_changed(self): |
|
| 57 | ct = get_alert_event_from_string('owned ticket changed') |
|
| 58 | self.assertEqual(ct, AlertEvent.OWNED_TICKET_CHANGED) |
|
| 59 | ||
| 60 | ||
| 61 | if __name__ == '__main__': |
|
| @@ 25-58 (lines=34) @@ | ||
| 22 | from gvm.protocols.gmpv214 import AlertEvent, get_alert_event_from_string |
|
| 23 | ||
| 24 | ||
| 25 | class GetAlertEventFromStringTestCase(unittest.TestCase): |
|
| 26 | def test_invalid(self): |
|
| 27 | with self.assertRaises(InvalidArgument): |
|
| 28 | get_alert_event_from_string('foo') |
|
| 29 | ||
| 30 | def test_none_or_empty(self): |
|
| 31 | ct = get_alert_event_from_string(None) |
|
| 32 | self.assertIsNone(ct) |
|
| 33 | ct = get_alert_event_from_string('') |
|
| 34 | self.assertIsNone(ct) |
|
| 35 | ||
| 36 | def test_task_run_status_changed(self): |
|
| 37 | ct = get_alert_event_from_string('Task run status changed') |
|
| 38 | self.assertEqual(ct, AlertEvent.TASK_RUN_STATUS_CHANGED) |
|
| 39 | ||
| 40 | def test_new_secinfo_arrived(self): |
|
| 41 | ct = get_alert_event_from_string('New SecInfo arrived') |
|
| 42 | self.assertEqual(ct, AlertEvent.NEW_SECINFO_ARRIVED) |
|
| 43 | ||
| 44 | def test_updated_secinfo_arrived(self): |
|
| 45 | ct = get_alert_event_from_string('Updated SecInfo arrived') |
|
| 46 | self.assertEqual(ct, AlertEvent.UPDATED_SECINFO_ARRIVED) |
|
| 47 | ||
| 48 | def test_ticket_received(self): |
|
| 49 | ct = get_alert_event_from_string('ticket received') |
|
| 50 | self.assertEqual(ct, AlertEvent.TICKET_RECEIVED) |
|
| 51 | ||
| 52 | def test_assigned_ticket_changed(self): |
|
| 53 | ct = get_alert_event_from_string('assigned ticket changed') |
|
| 54 | self.assertEqual(ct, AlertEvent.ASSIGNED_TICKET_CHANGED) |
|
| 55 | ||
| 56 | def test_owned_ticket_changed(self): |
|
| 57 | ct = get_alert_event_from_string('owned ticket changed') |
|
| 58 | self.assertEqual(ct, AlertEvent.OWNED_TICKET_CHANGED) |
|
| 59 | ||
| 60 | ||
| 61 | if __name__ == '__main__': |
|
