greenbone /
python-gvm
| @@ 23-150 (lines=128) @@ | ||
| 20 | from gvm.protocols.gmpv214 import ScannerType |
|
| 21 | ||
| 22 | ||
| 23 | class GmpModifyScannerTestMixin: |
|
| 24 | def test_modify_scanner(self): |
|
| 25 | self.gmp.modify_scanner(scanner_id='s1') |
|
| 26 | ||
| 27 | self.connection.send.has_been_called_with( |
|
| 28 | '<modify_scanner scanner_id="s1"/>' |
|
| 29 | ) |
|
| 30 | ||
| 31 | def test_modify_scanner_missing_scanner_id(self): |
|
| 32 | with self.assertRaises(RequiredArgument): |
|
| 33 | self.gmp.modify_scanner(scanner_id=None) |
|
| 34 | ||
| 35 | with self.assertRaises(RequiredArgument): |
|
| 36 | self.gmp.modify_scanner(scanner_id='') |
|
| 37 | ||
| 38 | def test_modify_scanner_with_comment(self): |
|
| 39 | self.gmp.modify_scanner(scanner_id='s1', comment='foo') |
|
| 40 | ||
| 41 | self.connection.send.has_been_called_with( |
|
| 42 | '<modify_scanner scanner_id="s1">' |
|
| 43 | '<comment>foo</comment>' |
|
| 44 | '</modify_scanner>' |
|
| 45 | ) |
|
| 46 | ||
| 47 | def test_modify_scanner_with_host(self): |
|
| 48 | self.gmp.modify_scanner(scanner_id='s1', host='foo') |
|
| 49 | ||
| 50 | self.connection.send.has_been_called_with( |
|
| 51 | '<modify_scanner scanner_id="s1">' |
|
| 52 | '<host>foo</host>' |
|
| 53 | '</modify_scanner>' |
|
| 54 | ) |
|
| 55 | ||
| 56 | def test_modify_scanner_with_port(self): |
|
| 57 | self.gmp.modify_scanner(scanner_id='s1', port=1234) |
|
| 58 | ||
| 59 | self.connection.send.has_been_called_with( |
|
| 60 | '<modify_scanner scanner_id="s1">' |
|
| 61 | '<port>1234</port>' |
|
| 62 | '</modify_scanner>' |
|
| 63 | ) |
|
| 64 | ||
| 65 | self.gmp.modify_scanner(scanner_id='s1', port='1234') |
|
| 66 | ||
| 67 | self.connection.send.has_been_called_with( |
|
| 68 | '<modify_scanner scanner_id="s1">' |
|
| 69 | '<port>1234</port>' |
|
| 70 | '</modify_scanner>' |
|
| 71 | ) |
|
| 72 | ||
| 73 | def test_modify_scanner_with_name(self): |
|
| 74 | self.gmp.modify_scanner(scanner_id='s1', name='foo') |
|
| 75 | ||
| 76 | self.connection.send.has_been_called_with( |
|
| 77 | '<modify_scanner scanner_id="s1">' |
|
| 78 | '<name>foo</name>' |
|
| 79 | '</modify_scanner>' |
|
| 80 | ) |
|
| 81 | ||
| 82 | def test_modify_scanner_with_ca_pub(self): |
|
| 83 | self.gmp.modify_scanner(scanner_id='s1', ca_pub='foo') |
|
| 84 | ||
| 85 | self.connection.send.has_been_called_with( |
|
| 86 | '<modify_scanner scanner_id="s1">' |
|
| 87 | '<ca_pub>foo</ca_pub>' |
|
| 88 | '</modify_scanner>' |
|
| 89 | ) |
|
| 90 | ||
| 91 | def test_modify_scanner_with_credential_id(self): |
|
| 92 | self.gmp.modify_scanner(scanner_id='s1', credential_id='c1') |
|
| 93 | ||
| 94 | self.connection.send.has_been_called_with( |
|
| 95 | '<modify_scanner scanner_id="s1">' |
|
| 96 | '<credential id="c1"/>' |
|
| 97 | '</modify_scanner>' |
|
| 98 | ) |
|
| 99 | ||
| 100 | def test_modify_scanner_with_scanner_type(self): |
|
| 101 | self.gmp.modify_scanner( |
|
| 102 | scanner_id='s1', scanner_type=ScannerType.OSP_SCANNER_TYPE |
|
| 103 | ) |
|
| 104 | ||
| 105 | self.connection.send.has_been_called_with( |
|
| 106 | '<modify_scanner scanner_id="s1">' |
|
| 107 | '<type>1</type>' |
|
| 108 | '</modify_scanner>' |
|
| 109 | ) |
|
| 110 | ||
| 111 | self.gmp.modify_scanner( |
|
| 112 | scanner_id='s1', scanner_type=ScannerType.OPENVAS_SCANNER_TYPE |
|
| 113 | ) |
|
| 114 | ||
| 115 | self.connection.send.has_been_called_with( |
|
| 116 | '<modify_scanner scanner_id="s1">' |
|
| 117 | '<type>2</type>' |
|
| 118 | '</modify_scanner>' |
|
| 119 | ) |
|
| 120 | ||
| 121 | self.gmp.modify_scanner( |
|
| 122 | scanner_id='s1', scanner_type=ScannerType.CVE_SCANNER_TYPE |
|
| 123 | ) |
|
| 124 | ||
| 125 | self.connection.send.has_been_called_with( |
|
| 126 | '<modify_scanner scanner_id="s1">' |
|
| 127 | '<type>3</type>' |
|
| 128 | '</modify_scanner>' |
|
| 129 | ) |
|
| 130 | ||
| 131 | self.gmp.modify_scanner( |
|
| 132 | scanner_id='s1', |
|
| 133 | scanner_type=ScannerType.GREENBONE_SENSOR_SCANNER_TYPE, |
|
| 134 | ) |
|
| 135 | ||
| 136 | self.connection.send.has_been_called_with( |
|
| 137 | '<modify_scanner scanner_id="s1">' |
|
| 138 | '<type>5</type>' |
|
| 139 | '</modify_scanner>' |
|
| 140 | ) |
|
| 141 | ||
| 142 | def test_modify_scanner_invalid_scanner_type(self): |
|
| 143 | with self.assertRaises(InvalidArgumentType): |
|
| 144 | self.gmp.modify_scanner(scanner_id='s1', scanner_type='') |
|
| 145 | ||
| 146 | with self.assertRaises(InvalidArgumentType): |
|
| 147 | self.gmp.modify_scanner(scanner_id='s1', scanner_type='-1') |
|
| 148 | ||
| 149 | with self.assertRaises(InvalidArgumentType): |
|
| 150 | self.gmp.modify_scanner(scanner_id='s1', scanner_type=1) |
|
| 151 | ||
| @@ 23-149 (lines=127) @@ | ||
| 20 | from gvm.protocols.gmpv208 import ScannerType |
|
| 21 | ||
| 22 | ||
| 23 | class GmpModifyScannerTestMixin: |
|
| 24 | def test_modify_scanner(self): |
|
| 25 | self.gmp.modify_scanner(scanner_id='s1') |
|
| 26 | ||
| 27 | self.connection.send.has_been_called_with( |
|
| 28 | '<modify_scanner scanner_id="s1"/>' |
|
| 29 | ) |
|
| 30 | ||
| 31 | def test_modify_scanner_missing_scanner_id(self): |
|
| 32 | with self.assertRaises(RequiredArgument): |
|
| 33 | self.gmp.modify_scanner(scanner_id=None) |
|
| 34 | ||
| 35 | with self.assertRaises(RequiredArgument): |
|
| 36 | self.gmp.modify_scanner(scanner_id='') |
|
| 37 | ||
| 38 | def test_modify_scanner_with_comment(self): |
|
| 39 | self.gmp.modify_scanner(scanner_id='s1', comment='foo') |
|
| 40 | ||
| 41 | self.connection.send.has_been_called_with( |
|
| 42 | '<modify_scanner scanner_id="s1">' |
|
| 43 | '<comment>foo</comment>' |
|
| 44 | '</modify_scanner>' |
|
| 45 | ) |
|
| 46 | ||
| 47 | def test_modify_scanner_with_host(self): |
|
| 48 | self.gmp.modify_scanner(scanner_id='s1', host='foo') |
|
| 49 | ||
| 50 | self.connection.send.has_been_called_with( |
|
| 51 | '<modify_scanner scanner_id="s1">' |
|
| 52 | '<host>foo</host>' |
|
| 53 | '</modify_scanner>' |
|
| 54 | ) |
|
| 55 | ||
| 56 | def test_modify_scanner_with_port(self): |
|
| 57 | self.gmp.modify_scanner(scanner_id='s1', port=1234) |
|
| 58 | ||
| 59 | self.connection.send.has_been_called_with( |
|
| 60 | '<modify_scanner scanner_id="s1">' |
|
| 61 | '<port>1234</port>' |
|
| 62 | '</modify_scanner>' |
|
| 63 | ) |
|
| 64 | ||
| 65 | self.gmp.modify_scanner(scanner_id='s1', port='1234') |
|
| 66 | ||
| 67 | self.connection.send.has_been_called_with( |
|
| 68 | '<modify_scanner scanner_id="s1">' |
|
| 69 | '<port>1234</port>' |
|
| 70 | '</modify_scanner>' |
|
| 71 | ) |
|
| 72 | ||
| 73 | def test_modify_scanner_with_name(self): |
|
| 74 | self.gmp.modify_scanner(scanner_id='s1', name='foo') |
|
| 75 | ||
| 76 | self.connection.send.has_been_called_with( |
|
| 77 | '<modify_scanner scanner_id="s1">' |
|
| 78 | '<name>foo</name>' |
|
| 79 | '</modify_scanner>' |
|
| 80 | ) |
|
| 81 | ||
| 82 | def test_modify_scanner_with_ca_pub(self): |
|
| 83 | self.gmp.modify_scanner(scanner_id='s1', ca_pub='foo') |
|
| 84 | ||
| 85 | self.connection.send.has_been_called_with( |
|
| 86 | '<modify_scanner scanner_id="s1">' |
|
| 87 | '<ca_pub>foo</ca_pub>' |
|
| 88 | '</modify_scanner>' |
|
| 89 | ) |
|
| 90 | ||
| 91 | def test_modify_scanner_with_credential_id(self): |
|
| 92 | self.gmp.modify_scanner(scanner_id='s1', credential_id='c1') |
|
| 93 | ||
| 94 | self.connection.send.has_been_called_with( |
|
| 95 | '<modify_scanner scanner_id="s1">' |
|
| 96 | '<credential id="c1"/>' |
|
| 97 | '</modify_scanner>' |
|
| 98 | ) |
|
| 99 | ||
| 100 | def test_modify_scanner_with_scanner_type(self): |
|
| 101 | self.gmp.modify_scanner( |
|
| 102 | scanner_id='s1', scanner_type=ScannerType.OSP_SCANNER_TYPE |
|
| 103 | ) |
|
| 104 | ||
| 105 | self.connection.send.has_been_called_with( |
|
| 106 | '<modify_scanner scanner_id="s1">' |
|
| 107 | '<type>1</type>' |
|
| 108 | '</modify_scanner>' |
|
| 109 | ) |
|
| 110 | ||
| 111 | self.gmp.modify_scanner( |
|
| 112 | scanner_id='s1', scanner_type=ScannerType.OPENVAS_SCANNER_TYPE |
|
| 113 | ) |
|
| 114 | ||
| 115 | self.connection.send.has_been_called_with( |
|
| 116 | '<modify_scanner scanner_id="s1">' |
|
| 117 | '<type>2</type>' |
|
| 118 | '</modify_scanner>' |
|
| 119 | ) |
|
| 120 | ||
| 121 | self.gmp.modify_scanner( |
|
| 122 | scanner_id='s1', scanner_type=ScannerType.CVE_SCANNER_TYPE |
|
| 123 | ) |
|
| 124 | ||
| 125 | self.connection.send.has_been_called_with( |
|
| 126 | '<modify_scanner scanner_id="s1">' |
|
| 127 | '<type>3</type>' |
|
| 128 | '</modify_scanner>' |
|
| 129 | ) |
|
| 130 | ||
| 131 | self.gmp.modify_scanner( |
|
| 132 | scanner_id='s1', scanner_type=ScannerType.GMP_SCANNER_TYPE |
|
| 133 | ) |
|
| 134 | ||
| 135 | self.connection.send.has_been_called_with( |
|
| 136 | '<modify_scanner scanner_id="s1">' |
|
| 137 | '<type>4</type>' |
|
| 138 | '</modify_scanner>' |
|
| 139 | ) |
|
| 140 | ||
| 141 | def test_modify_scanner_invalid_scanner_type(self): |
|
| 142 | with self.assertRaises(InvalidArgumentType): |
|
| 143 | self.gmp.modify_scanner(scanner_id='s1', scanner_type='') |
|
| 144 | ||
| 145 | with self.assertRaises(InvalidArgumentType): |
|
| 146 | self.gmp.modify_scanner(scanner_id='s1', scanner_type='-1') |
|
| 147 | ||
| 148 | with self.assertRaises(InvalidArgumentType): |
|
| 149 | self.gmp.modify_scanner(scanner_id='s1', scanner_type=1) |
|
| 150 | ||
