greenbone /
python-gvm
| @@ 24-72 (lines=49) @@ | ||
| 21 | from gvm.errors import RequiredArgument |
|
| 22 | ||
| 23 | ||
| 24 | class GmpModifyAssetTestCase: |
|
| 25 | def test_modify_asset(self): |
|
| 26 | self.gmp.modify_asset(asset_id='a1') |
|
| 27 | ||
| 28 | self.connection.send.has_been_called_with( |
|
| 29 | '<modify_asset asset_id="a1">' |
|
| 30 | '<comment></comment>' |
|
| 31 | '</modify_asset>' |
|
| 32 | ) |
|
| 33 | ||
| 34 | def test_modify_asset_without_asset_id(self): |
|
| 35 | with self.assertRaises(RequiredArgument): |
|
| 36 | self.gmp.modify_asset(asset_id=None, comment='foo') |
|
| 37 | ||
| 38 | with self.assertRaises(RequiredArgument): |
|
| 39 | self.gmp.modify_asset(asset_id='', comment='foo') |
|
| 40 | ||
| 41 | with self.assertRaises(RequiredArgument): |
|
| 42 | self.gmp.modify_asset('', comment='foo') |
|
| 43 | ||
| 44 | def test_modify_asset_with_comment(self): |
|
| 45 | self.gmp.modify_asset('a1', 'foo') |
|
| 46 | ||
| 47 | self.connection.send.has_been_called_with( |
|
| 48 | '<modify_asset asset_id="a1">' |
|
| 49 | '<comment>foo</comment>' |
|
| 50 | '</modify_asset>' |
|
| 51 | ) |
|
| 52 | ||
| 53 | self.gmp.modify_asset('a1', comment='foo') |
|
| 54 | ||
| 55 | self.connection.send.has_been_called_with( |
|
| 56 | '<modify_asset asset_id="a1">' |
|
| 57 | '<comment>foo</comment>' |
|
| 58 | '</modify_asset>' |
|
| 59 | ) |
|
| 60 | ||
| 61 | self.gmp.modify_asset('a1', '') |
|
| 62 | ||
| 63 | self.connection.send.has_been_called_with( |
|
| 64 | '<modify_asset asset_id="a1">' |
|
| 65 | '<comment></comment>' |
|
| 66 | '</modify_asset>' |
|
| 67 | ) |
|
| 68 | ||
| 69 | self.gmp.modify_asset('a1', None) |
|
| 70 | ||
| 71 | self.connection.send.has_been_called_with( |
|
| 72 | '<modify_asset asset_id="a1">' '<comment/>' '</modify_asset>' |
|
| 73 | ) |
|
| 74 | ||
| 75 | ||
| @@ 24-72 (lines=49) @@ | ||
| 21 | from gvm.errors import RequiredArgument |
|
| 22 | ||
| 23 | ||
| 24 | class GmpModifyAssetTestCase: |
|
| 25 | def test_modify_asset(self): |
|
| 26 | self.gmp.modify_asset(asset_id='a1') |
|
| 27 | ||
| 28 | self.connection.send.has_been_called_with( |
|
| 29 | '<modify_asset asset_id="a1">' |
|
| 30 | '<comment></comment>' |
|
| 31 | '</modify_asset>' |
|
| 32 | ) |
|
| 33 | ||
| 34 | def test_modify_asset_without_asset_id(self): |
|
| 35 | with self.assertRaises(RequiredArgument): |
|
| 36 | self.gmp.modify_asset(asset_id=None, comment='foo') |
|
| 37 | ||
| 38 | with self.assertRaises(RequiredArgument): |
|
| 39 | self.gmp.modify_asset(asset_id='', comment='foo') |
|
| 40 | ||
| 41 | with self.assertRaises(RequiredArgument): |
|
| 42 | self.gmp.modify_asset('', comment='foo') |
|
| 43 | ||
| 44 | def test_modify_asset_with_comment(self): |
|
| 45 | self.gmp.modify_asset('a1', 'foo') |
|
| 46 | ||
| 47 | self.connection.send.has_been_called_with( |
|
| 48 | '<modify_asset asset_id="a1">' |
|
| 49 | '<comment>foo</comment>' |
|
| 50 | '</modify_asset>' |
|
| 51 | ) |
|
| 52 | ||
| 53 | self.gmp.modify_asset('a1', comment='foo') |
|
| 54 | ||
| 55 | self.connection.send.has_been_called_with( |
|
| 56 | '<modify_asset asset_id="a1">' |
|
| 57 | '<comment>foo</comment>' |
|
| 58 | '</modify_asset>' |
|
| 59 | ) |
|
| 60 | ||
| 61 | self.gmp.modify_asset('a1', '') |
|
| 62 | ||
| 63 | self.connection.send.has_been_called_with( |
|
| 64 | '<modify_asset asset_id="a1">' |
|
| 65 | '<comment></comment>' |
|
| 66 | '</modify_asset>' |
|
| 67 | ) |
|
| 68 | ||
| 69 | self.gmp.modify_asset('a1', None) |
|
| 70 | ||
| 71 | self.connection.send.has_been_called_with( |
|
| 72 | '<modify_asset asset_id="a1">' '<comment/>' '</modify_asset>' |
|
| 73 | ) |
|
| 74 | ||
| 75 | ||
