greenbone /
python-gvm
| @@ 26-46 (lines=21) @@ | ||
| 23 | from . import Gmpv9TestCase |
|
| 24 | ||
| 25 | ||
| 26 | class GmpDeleteTaskTestCase(Gmpv9TestCase): |
|
| 27 | def test_delete(self): |
|
| 28 | self.gmp.delete_task('a1') |
|
| 29 | ||
| 30 | self.connection.send.has_been_called_with( |
|
| 31 | '<delete_task task_id="a1" ultimate="0"/>' |
|
| 32 | ) |
|
| 33 | ||
| 34 | def test_delete_ultimate(self): |
|
| 35 | self.gmp.delete_task('a1', ultimate=True) |
|
| 36 | ||
| 37 | self.connection.send.has_been_called_with( |
|
| 38 | '<delete_task task_id="a1" ultimate="1"/>' |
|
| 39 | ) |
|
| 40 | ||
| 41 | def test_missing_id(self): |
|
| 42 | with self.assertRaises(GvmError): |
|
| 43 | self.gmp.delete_task(None) |
|
| 44 | ||
| 45 | with self.assertRaises(GvmError): |
|
| 46 | self.gmp.delete_task('') |
|
| 47 | ||
| 48 | ||
| 49 | if __name__ == '__main__': |
|
| @@ 26-46 (lines=21) @@ | ||
| 23 | from . import Gmpv7TestCase |
|
| 24 | ||
| 25 | ||
| 26 | class GmpDeleteTaskTestCase(Gmpv7TestCase): |
|
| 27 | def test_delete(self): |
|
| 28 | self.gmp.delete_task('a1') |
|
| 29 | ||
| 30 | self.connection.send.has_been_called_with( |
|
| 31 | '<delete_task task_id="a1" ultimate="0"/>' |
|
| 32 | ) |
|
| 33 | ||
| 34 | def test_delete_ultimate(self): |
|
| 35 | self.gmp.delete_task('a1', ultimate=True) |
|
| 36 | ||
| 37 | self.connection.send.has_been_called_with( |
|
| 38 | '<delete_task task_id="a1" ultimate="1"/>' |
|
| 39 | ) |
|
| 40 | ||
| 41 | def test_missing_id(self): |
|
| 42 | with self.assertRaises(GvmError): |
|
| 43 | self.gmp.delete_task(None) |
|
| 44 | ||
| 45 | with self.assertRaises(GvmError): |
|
| 46 | self.gmp.delete_task('') |
|
| 47 | ||
| 48 | ||
| 49 | if __name__ == '__main__': |
|
