greenbone /
python-gvm
| @@ 24-60 (lines=37) @@ | ||
| 21 | from gvm.errors import RequiredArgument |
|
| 22 | ||
| 23 | ||
| 24 | class GmpCreateConfigFromOSPScannerTestCase: |
|
| 25 | def test_create_config_from_osp_scanner(self): |
|
| 26 | self.gmp.create_config_from_osp_scanner('a1', 'foo') |
|
| 27 | ||
| 28 | self.connection.send.has_been_called_with( |
|
| 29 | '<create_config>' |
|
| 30 | '<scanner>a1</scanner>' |
|
| 31 | '<name>foo</name>' |
|
| 32 | '<usage_type>scan</usage_type>' |
|
| 33 | '</create_config>' |
|
| 34 | ) |
|
| 35 | ||
| 36 | def test_create_config_from_osp_scanner_with_comment(self): |
|
| 37 | self.gmp.create_config_from_osp_scanner('a1', 'foo', comment='comment') |
|
| 38 | ||
| 39 | self.connection.send.has_been_called_with( |
|
| 40 | '<create_config>' |
|
| 41 | '<comment>comment</comment>' |
|
| 42 | '<scanner>a1</scanner>' |
|
| 43 | '<name>foo</name>' |
|
| 44 | '<usage_type>scan</usage_type>' |
|
| 45 | '</create_config>' |
|
| 46 | ) |
|
| 47 | ||
| 48 | def test_missing_scanner_id(self): |
|
| 49 | with self.assertRaises(RequiredArgument): |
|
| 50 | self.gmp.create_config_from_osp_scanner(scanner_id='', name='foo') |
|
| 51 | ||
| 52 | with self.assertRaises(RequiredArgument): |
|
| 53 | self.gmp.create_config_from_osp_scanner(scanner_id=None, name='foo') |
|
| 54 | ||
| 55 | def test_missing_name(self): |
|
| 56 | with self.assertRaises(RequiredArgument): |
|
| 57 | self.gmp.create_config_from_osp_scanner(scanner_id='c1', name=None) |
|
| 58 | ||
| 59 | with self.assertRaises(RequiredArgument): |
|
| 60 | self.gmp.create_config_from_osp_scanner(scanner_id='c1', name='') |
|
| 61 | ||
| 62 | ||
| 63 | if __name__ == '__main__': |
|
| @@ 24-60 (lines=37) @@ | ||
| 21 | from gvm.errors import RequiredArgument |
|
| 22 | ||
| 23 | ||
| 24 | class GmpCreateConfigFromOSPScannerTestCase: |
|
| 25 | def test_create_config_from_osp_scanner(self): |
|
| 26 | self.gmp.create_config_from_osp_scanner('a1', 'foo') |
|
| 27 | ||
| 28 | self.connection.send.has_been_called_with( |
|
| 29 | '<create_config>' |
|
| 30 | '<scanner>a1</scanner>' |
|
| 31 | '<name>foo</name>' |
|
| 32 | '<usage_type>scan</usage_type>' |
|
| 33 | '</create_config>' |
|
| 34 | ) |
|
| 35 | ||
| 36 | def test_create_config_from_osp_scanner_with_comment(self): |
|
| 37 | self.gmp.create_config_from_osp_scanner('a1', 'foo', comment='comment') |
|
| 38 | ||
| 39 | self.connection.send.has_been_called_with( |
|
| 40 | '<create_config>' |
|
| 41 | '<comment>comment</comment>' |
|
| 42 | '<scanner>a1</scanner>' |
|
| 43 | '<name>foo</name>' |
|
| 44 | '<usage_type>scan</usage_type>' |
|
| 45 | '</create_config>' |
|
| 46 | ) |
|
| 47 | ||
| 48 | def test_missing_scanner_id(self): |
|
| 49 | with self.assertRaises(RequiredArgument): |
|
| 50 | self.gmp.create_config_from_osp_scanner(scanner_id='', name='foo') |
|
| 51 | ||
| 52 | with self.assertRaises(RequiredArgument): |
|
| 53 | self.gmp.create_config_from_osp_scanner(scanner_id=None, name='foo') |
|
| 54 | ||
| 55 | def test_missing_name(self): |
|
| 56 | with self.assertRaises(RequiredArgument): |
|
| 57 | self.gmp.create_config_from_osp_scanner(scanner_id='c1', name=None) |
|
| 58 | ||
| 59 | with self.assertRaises(RequiredArgument): |
|
| 60 | self.gmp.create_config_from_osp_scanner(scanner_id='c1', name='') |
|
| 61 | ||
| 62 | ||
| 63 | if __name__ == '__main__': |
|
