greenbone /
python-gvm
| @@ 24-74 (lines=51) @@ | ||
| 21 | from gvm.errors import RequiredArgument |
|
| 22 | ||
| 23 | ||
| 24 | class GmpCreateTicketTestCase: |
|
| 25 | def test_create_ticket(self): |
|
| 26 | self.gmp.create_ticket( |
|
| 27 | result_id='r1', assigned_to_user_id='u1', note='lorem ipsum' |
|
| 28 | ) |
|
| 29 | ||
| 30 | self.connection.send.has_been_called_with( |
|
| 31 | '<create_ticket>' |
|
| 32 | '<result id="r1"/>' |
|
| 33 | '<assigned_to>' |
|
| 34 | '<user id="u1"/>' |
|
| 35 | '</assigned_to>' |
|
| 36 | '<open_note>lorem ipsum</open_note>' |
|
| 37 | '</create_ticket>' |
|
| 38 | ) |
|
| 39 | ||
| 40 | def test_create_ticket_with_comment(self): |
|
| 41 | self.gmp.create_ticket( |
|
| 42 | result_id='r1', |
|
| 43 | assigned_to_user_id='u1', |
|
| 44 | note='lorem ipsum', |
|
| 45 | comment='bar', |
|
| 46 | ) |
|
| 47 | ||
| 48 | self.connection.send.has_been_called_with( |
|
| 49 | '<create_ticket>' |
|
| 50 | '<result id="r1"/>' |
|
| 51 | '<assigned_to>' |
|
| 52 | '<user id="u1"/>' |
|
| 53 | '</assigned_to>' |
|
| 54 | '<open_note>lorem ipsum</open_note>' |
|
| 55 | '<comment>bar</comment>' |
|
| 56 | '</create_ticket>' |
|
| 57 | ) |
|
| 58 | ||
| 59 | def test_create_ticket_missing_result_id(self): |
|
| 60 | with self.assertRaises(RequiredArgument): |
|
| 61 | self.gmp.create_ticket( |
|
| 62 | result_id='', assigned_to_user_id='u1', note='lorem ipsum' |
|
| 63 | ) |
|
| 64 | ||
| 65 | def test_create_ticket_missing_assigned_to_user_id(self): |
|
| 66 | with self.assertRaises(RequiredArgument): |
|
| 67 | self.gmp.create_ticket( |
|
| 68 | result_id='r1', assigned_to_user_id='', note='lorem ipsum' |
|
| 69 | ) |
|
| 70 | ||
| 71 | def test_create_ticket_missing_open_note(self): |
|
| 72 | with self.assertRaises(RequiredArgument): |
|
| 73 | self.gmp.create_ticket( |
|
| 74 | result_id='r1', assigned_to_user_id='u1', note='' |
|
| 75 | ) |
|
| 76 | ||
| 77 | ||
| @@ 24-74 (lines=51) @@ | ||
| 21 | from gvm.errors import RequiredArgument |
|
| 22 | ||
| 23 | ||
| 24 | class GmpCreateTicketTestCase: |
|
| 25 | def test_create_ticket(self): |
|
| 26 | self.gmp.create_ticket( |
|
| 27 | result_id='r1', assigned_to_user_id='u1', note='lorem ipsum' |
|
| 28 | ) |
|
| 29 | ||
| 30 | self.connection.send.has_been_called_with( |
|
| 31 | '<create_ticket>' |
|
| 32 | '<result id="r1"/>' |
|
| 33 | '<assigned_to>' |
|
| 34 | '<user id="u1"/>' |
|
| 35 | '</assigned_to>' |
|
| 36 | '<open_note>lorem ipsum</open_note>' |
|
| 37 | '</create_ticket>' |
|
| 38 | ) |
|
| 39 | ||
| 40 | def test_create_ticket_with_comment(self): |
|
| 41 | self.gmp.create_ticket( |
|
| 42 | result_id='r1', |
|
| 43 | assigned_to_user_id='u1', |
|
| 44 | note='lorem ipsum', |
|
| 45 | comment='bar', |
|
| 46 | ) |
|
| 47 | ||
| 48 | self.connection.send.has_been_called_with( |
|
| 49 | '<create_ticket>' |
|
| 50 | '<result id="r1"/>' |
|
| 51 | '<assigned_to>' |
|
| 52 | '<user id="u1"/>' |
|
| 53 | '</assigned_to>' |
|
| 54 | '<open_note>lorem ipsum</open_note>' |
|
| 55 | '<comment>bar</comment>' |
|
| 56 | '</create_ticket>' |
|
| 57 | ) |
|
| 58 | ||
| 59 | def test_create_ticket_missing_result_id(self): |
|
| 60 | with self.assertRaises(RequiredArgument): |
|
| 61 | self.gmp.create_ticket( |
|
| 62 | result_id='', assigned_to_user_id='u1', note='lorem ipsum' |
|
| 63 | ) |
|
| 64 | ||
| 65 | def test_create_ticket_missing_assigned_to_user_id(self): |
|
| 66 | with self.assertRaises(RequiredArgument): |
|
| 67 | self.gmp.create_ticket( |
|
| 68 | result_id='r1', assigned_to_user_id='', note='lorem ipsum' |
|
| 69 | ) |
|
| 70 | ||
| 71 | def test_create_ticket_missing_open_note(self): |
|
| 72 | with self.assertRaises(RequiredArgument): |
|
| 73 | self.gmp.create_ticket( |
|
| 74 | result_id='r1', assigned_to_user_id='u1', note='' |
|
| 75 | ) |
|
| 76 | ||
| 77 | ||
