greenbone /
python-gvm
| @@ 29-425 (lines=397) @@ | ||
| 26 | ) |
|
| 27 | ||
| 28 | ||
| 29 | class GmpCreateCredentialTestCase: |
|
| 30 | def test_create_up_credential_missing_name(self): |
|
| 31 | with self.assertRaises(RequiredArgument): |
|
| 32 | self.gmp.create_credential( |
|
| 33 | name='', |
|
| 34 | credential_type=CredentialType.USERNAME_PASSWORD, |
|
| 35 | login='foo', |
|
| 36 | ) |
|
| 37 | ||
| 38 | with self.assertRaises(RequiredArgument): |
|
| 39 | self.gmp.create_credential( |
|
| 40 | name=None, |
|
| 41 | credential_type=CredentialType.USERNAME_PASSWORD, |
|
| 42 | login='foo', |
|
| 43 | ) |
|
| 44 | ||
| 45 | def test_create_up_credential(self): |
|
| 46 | self.gmp.create_credential( |
|
| 47 | name='foo', |
|
| 48 | credential_type=CredentialType.USERNAME_PASSWORD, |
|
| 49 | comment='bar', |
|
| 50 | login='Max', |
|
| 51 | password='123', |
|
| 52 | ) |
|
| 53 | ||
| 54 | self.connection.send.has_been_called_with( |
|
| 55 | '<create_credential>' |
|
| 56 | '<name>foo</name>' |
|
| 57 | '<type>up</type>' |
|
| 58 | '<comment>bar</comment>' |
|
| 59 | '<login>Max</login>' |
|
| 60 | '<password>123</password>' |
|
| 61 | '</create_credential>' |
|
| 62 | ) |
|
| 63 | ||
| 64 | def test_create_up_credential_with_allow_insecure(self): |
|
| 65 | self.gmp.create_credential( |
|
| 66 | name='foo', |
|
| 67 | credential_type=CredentialType.USERNAME_PASSWORD, |
|
| 68 | comment='bar', |
|
| 69 | login='Max', |
|
| 70 | password='123', |
|
| 71 | allow_insecure=True, |
|
| 72 | ) |
|
| 73 | ||
| 74 | self.connection.send.has_been_called_with( |
|
| 75 | '<create_credential>' |
|
| 76 | '<name>foo</name>' |
|
| 77 | '<type>up</type>' |
|
| 78 | '<comment>bar</comment>' |
|
| 79 | '<allow_insecure>1</allow_insecure>' |
|
| 80 | '<login>Max</login>' |
|
| 81 | '<password>123</password>' |
|
| 82 | '</create_credential>' |
|
| 83 | ) |
|
| 84 | ||
| 85 | self.gmp.create_credential( |
|
| 86 | name='foo', |
|
| 87 | credential_type=CredentialType.USERNAME_PASSWORD, |
|
| 88 | comment='bar', |
|
| 89 | login='Max', |
|
| 90 | password='123', |
|
| 91 | allow_insecure=False, |
|
| 92 | ) |
|
| 93 | ||
| 94 | self.connection.send.has_been_called_with( |
|
| 95 | '<create_credential>' |
|
| 96 | '<name>foo</name>' |
|
| 97 | '<type>up</type>' |
|
| 98 | '<comment>bar</comment>' |
|
| 99 | '<allow_insecure>0</allow_insecure>' |
|
| 100 | '<login>Max</login>' |
|
| 101 | '<password>123</password>' |
|
| 102 | '</create_credential>' |
|
| 103 | ) |
|
| 104 | ||
| 105 | def test_create_cc_credential_missing_certificate(self): |
|
| 106 | with self.assertRaises(RequiredArgument): |
|
| 107 | self.gmp.create_credential( |
|
| 108 | name='foo', credential_type=CredentialType.CLIENT_CERTIFICATE |
|
| 109 | ) |
|
| 110 | ||
| 111 | def test_create_cc_credential(self): |
|
| 112 | self.gmp.create_credential( |
|
| 113 | name='foo', |
|
| 114 | credential_type=CredentialType.CLIENT_CERTIFICATE, |
|
| 115 | certificate='abcdef', |
|
| 116 | ) |
|
| 117 | ||
| 118 | self.connection.send.has_been_called_with( |
|
| 119 | '<create_credential>' |
|
| 120 | '<name>foo</name>' |
|
| 121 | '<type>cc</type>' |
|
| 122 | '<certificate>abcdef</certificate>' |
|
| 123 | '</create_credential>' |
|
| 124 | ) |
|
| 125 | ||
| 126 | def test_create_cc_credential_with_private_key(self): |
|
| 127 | self.gmp.create_credential( |
|
| 128 | name='foo', |
|
| 129 | credential_type=CredentialType.CLIENT_CERTIFICATE, |
|
| 130 | certificate='abcdef', |
|
| 131 | private_key='123456', |
|
| 132 | ) |
|
| 133 | ||
| 134 | self.connection.send.has_been_called_with( |
|
| 135 | '<create_credential>' |
|
| 136 | '<name>foo</name>' |
|
| 137 | '<type>cc</type>' |
|
| 138 | '<certificate>abcdef</certificate>' |
|
| 139 | '<key>' |
|
| 140 | '<private>123456</private>' |
|
| 141 | '</key>' |
|
| 142 | '</create_credential>' |
|
| 143 | ) |
|
| 144 | ||
| 145 | def test_create_usk_credential_missing_private_key(self): |
|
| 146 | with self.assertRaises(RequiredArgument): |
|
| 147 | self.gmp.create_credential( |
|
| 148 | name='foo', |
|
| 149 | credential_type=CredentialType.USERNAME_SSH_KEY, |
|
| 150 | login='foo', |
|
| 151 | ) |
|
| 152 | ||
| 153 | def test_create_usk_credential_missing_login(self): |
|
| 154 | with self.assertRaises(RequiredArgument): |
|
| 155 | self.gmp.create_credential( |
|
| 156 | name='foo', |
|
| 157 | credential_type=CredentialType.USERNAME_SSH_KEY, |
|
| 158 | private_key='123456', |
|
| 159 | ) |
|
| 160 | ||
| 161 | def test_create_usk_credential(self): |
|
| 162 | self.gmp.create_credential( |
|
| 163 | name='foo', |
|
| 164 | credential_type=CredentialType.USERNAME_SSH_KEY, |
|
| 165 | private_key='123456', |
|
| 166 | login='foo', |
|
| 167 | ) |
|
| 168 | ||
| 169 | self.connection.send.has_been_called_with( |
|
| 170 | '<create_credential>' |
|
| 171 | '<name>foo</name>' |
|
| 172 | '<type>usk</type>' |
|
| 173 | '<login>foo</login>' |
|
| 174 | '<key>' |
|
| 175 | '<private>123456</private>' |
|
| 176 | '</key>' |
|
| 177 | '</create_credential>' |
|
| 178 | ) |
|
| 179 | ||
| 180 | def test_create_usk_credential_with_key_phrase(self): |
|
| 181 | self.gmp.create_credential( |
|
| 182 | name='foo', |
|
| 183 | credential_type=CredentialType.USERNAME_SSH_KEY, |
|
| 184 | private_key='123456', |
|
| 185 | login='foo', |
|
| 186 | key_phrase='abcdef', |
|
| 187 | ) |
|
| 188 | ||
| 189 | self.connection.send.has_been_called_with( |
|
| 190 | '<create_credential>' |
|
| 191 | '<name>foo</name>' |
|
| 192 | '<type>usk</type>' |
|
| 193 | '<login>foo</login>' |
|
| 194 | '<key>' |
|
| 195 | '<private>123456</private>' |
|
| 196 | '<phrase>abcdef</phrase>' |
|
| 197 | '</key>' |
|
| 198 | '</create_credential>' |
|
| 199 | ) |
|
| 200 | ||
| 201 | def test_create_snmp_credential_invalid_auth_algorithm(self): |
|
| 202 | with self.assertRaises(InvalidArgumentType): |
|
| 203 | self.gmp.create_credential( |
|
| 204 | name='foo', credential_type=CredentialType.SNMP, login='foo' |
|
| 205 | ) |
|
| 206 | ||
| 207 | with self.assertRaises(InvalidArgumentType): |
|
| 208 | self.gmp.create_credential( |
|
| 209 | name='foo', |
|
| 210 | credential_type=CredentialType.SNMP, |
|
| 211 | login='foo', |
|
| 212 | auth_algorithm='', |
|
| 213 | ) |
|
| 214 | ||
| 215 | with self.assertRaises(InvalidArgumentType): |
|
| 216 | self.gmp.create_credential( |
|
| 217 | name='foo', |
|
| 218 | credential_type=CredentialType.SNMP, |
|
| 219 | login='foo', |
|
| 220 | auth_algorithm='bar', |
|
| 221 | ) |
|
| 222 | ||
| 223 | def test_create_snmp_credential_auth_algorithm_md5(self): |
|
| 224 | self.gmp.create_credential( |
|
| 225 | name='foo', |
|
| 226 | credential_type=CredentialType.SNMP, |
|
| 227 | login='foo', |
|
| 228 | auth_algorithm=SnmpAuthAlgorithm.MD5, |
|
| 229 | ) |
|
| 230 | ||
| 231 | self.connection.send.has_been_called_with( |
|
| 232 | '<create_credential>' |
|
| 233 | '<name>foo</name>' |
|
| 234 | '<type>snmp</type>' |
|
| 235 | '<login>foo</login>' |
|
| 236 | '<auth_algorithm>md5</auth_algorithm>' |
|
| 237 | '</create_credential>' |
|
| 238 | ) |
|
| 239 | ||
| 240 | def test_create_snmp_credential_auth_algorithm_sha1(self): |
|
| 241 | self.gmp.create_credential( |
|
| 242 | name='foo', |
|
| 243 | credential_type=CredentialType.SNMP, |
|
| 244 | login='foo', |
|
| 245 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 246 | ) |
|
| 247 | ||
| 248 | self.connection.send.has_been_called_with( |
|
| 249 | '<create_credential>' |
|
| 250 | '<name>foo</name>' |
|
| 251 | '<type>snmp</type>' |
|
| 252 | '<login>foo</login>' |
|
| 253 | '<auth_algorithm>sha1</auth_algorithm>' |
|
| 254 | '</create_credential>' |
|
| 255 | ) |
|
| 256 | ||
| 257 | def test_create_snmp_credential_with_community(self): |
|
| 258 | self.gmp.create_credential( |
|
| 259 | name='foo', |
|
| 260 | credential_type=CredentialType.SNMP, |
|
| 261 | login='foo', |
|
| 262 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 263 | community='ipsum', |
|
| 264 | ) |
|
| 265 | ||
| 266 | self.connection.send.has_been_called_with( |
|
| 267 | '<create_credential>' |
|
| 268 | '<name>foo</name>' |
|
| 269 | '<type>snmp</type>' |
|
| 270 | '<login>foo</login>' |
|
| 271 | '<auth_algorithm>sha1</auth_algorithm>' |
|
| 272 | '<community>ipsum</community>' |
|
| 273 | '</create_credential>' |
|
| 274 | ) |
|
| 275 | ||
| 276 | def test_create_snmp_credential_invalid_privacy_algorithm(self): |
|
| 277 | with self.assertRaises(InvalidArgumentType): |
|
| 278 | self.gmp.create_credential( |
|
| 279 | name='foo', |
|
| 280 | credential_type=CredentialType.SNMP, |
|
| 281 | login='foo', |
|
| 282 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 283 | privacy_algorithm='', |
|
| 284 | ) |
|
| 285 | ||
| 286 | with self.assertRaises(InvalidArgumentType): |
|
| 287 | self.gmp.create_credential( |
|
| 288 | name='foo', |
|
| 289 | credential_type=CredentialType.SNMP, |
|
| 290 | login='foo', |
|
| 291 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 292 | privacy_algorithm='foo', |
|
| 293 | ) |
|
| 294 | ||
| 295 | def test_create_snmp_credential_with_privacy_algorithm_aes(self): |
|
| 296 | self.gmp.create_credential( |
|
| 297 | name='foo', |
|
| 298 | credential_type=CredentialType.SNMP, |
|
| 299 | login='foo', |
|
| 300 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 301 | privacy_algorithm=SnmpPrivacyAlgorithm.AES, |
|
| 302 | ) |
|
| 303 | ||
| 304 | self.connection.send.has_been_called_with( |
|
| 305 | '<create_credential>' |
|
| 306 | '<name>foo</name>' |
|
| 307 | '<type>snmp</type>' |
|
| 308 | '<login>foo</login>' |
|
| 309 | '<auth_algorithm>sha1</auth_algorithm>' |
|
| 310 | '<privacy>' |
|
| 311 | '<algorithm>aes</algorithm>' |
|
| 312 | '</privacy>' |
|
| 313 | '</create_credential>' |
|
| 314 | ) |
|
| 315 | ||
| 316 | def test_create_snmp_credential_with_privacy_algorithm_des(self): |
|
| 317 | self.gmp.create_credential( |
|
| 318 | name='foo', |
|
| 319 | credential_type=CredentialType.SNMP, |
|
| 320 | login='foo', |
|
| 321 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 322 | privacy_algorithm=SnmpPrivacyAlgorithm.DES, |
|
| 323 | ) |
|
| 324 | ||
| 325 | self.connection.send.has_been_called_with( |
|
| 326 | '<create_credential>' |
|
| 327 | '<name>foo</name>' |
|
| 328 | '<type>snmp</type>' |
|
| 329 | '<login>foo</login>' |
|
| 330 | '<auth_algorithm>sha1</auth_algorithm>' |
|
| 331 | '<privacy>' |
|
| 332 | '<algorithm>des</algorithm>' |
|
| 333 | '</privacy>' |
|
| 334 | '</create_credential>' |
|
| 335 | ) |
|
| 336 | ||
| 337 | def test_create_snmp_credential_with_privacy_password(self): |
|
| 338 | self.gmp.create_credential( |
|
| 339 | name='foo', |
|
| 340 | credential_type=CredentialType.SNMP, |
|
| 341 | login='foo', |
|
| 342 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 343 | privacy_password='123', |
|
| 344 | ) |
|
| 345 | ||
| 346 | self.connection.send.has_been_called_with( |
|
| 347 | '<create_credential>' |
|
| 348 | '<name>foo</name>' |
|
| 349 | '<type>snmp</type>' |
|
| 350 | '<login>foo</login>' |
|
| 351 | '<auth_algorithm>sha1</auth_algorithm>' |
|
| 352 | '<privacy>' |
|
| 353 | '<password>123</password>' |
|
| 354 | '</privacy>' |
|
| 355 | '</create_credential>' |
|
| 356 | ) |
|
| 357 | ||
| 358 | def test_create_smime_credential(self): |
|
| 359 | self.gmp.create_credential( |
|
| 360 | name='foo', |
|
| 361 | credential_type=CredentialType.SMIME_CERTIFICATE, |
|
| 362 | certificate='ipsum', |
|
| 363 | ) |
|
| 364 | ||
| 365 | self.connection.send.has_been_called_with( |
|
| 366 | '<create_credential>' |
|
| 367 | '<name>foo</name>' |
|
| 368 | '<type>smime</type>' |
|
| 369 | '<certificate>ipsum</certificate>' |
|
| 370 | '</create_credential>' |
|
| 371 | ) |
|
| 372 | ||
| 373 | def test_create_smime_credential_missing_certificate(self): |
|
| 374 | with self.assertRaises(RequiredArgument): |
|
| 375 | self.gmp.create_credential( |
|
| 376 | name='foo', credential_type=CredentialType.SMIME_CERTIFICATE |
|
| 377 | ) |
|
| 378 | ||
| 379 | def test_create_pgp_credential(self): |
|
| 380 | self.gmp.create_credential( |
|
| 381 | name='foo', |
|
| 382 | credential_type=CredentialType.PGP_ENCRYPTION_KEY, |
|
| 383 | public_key='ipsum', |
|
| 384 | ) |
|
| 385 | ||
| 386 | self.connection.send.has_been_called_with( |
|
| 387 | '<create_credential>' |
|
| 388 | '<name>foo</name>' |
|
| 389 | '<type>pgp</type>' |
|
| 390 | '<key>' |
|
| 391 | '<public>ipsum</public>' |
|
| 392 | '</key>' |
|
| 393 | '</create_credential>' |
|
| 394 | ) |
|
| 395 | ||
| 396 | def test_create_pgp_credential_missing_public_key(self): |
|
| 397 | with self.assertRaises(RequiredArgument): |
|
| 398 | self.gmp.create_credential( |
|
| 399 | name='foo', credential_type=CredentialType.PGP_ENCRYPTION_KEY |
|
| 400 | ) |
|
| 401 | ||
| 402 | def test_create_credential_invalid_credential_type(self): |
|
| 403 | with self.assertRaises(InvalidArgumentType): |
|
| 404 | self.gmp.create_credential(name='foo', credential_type=None) |
|
| 405 | ||
| 406 | with self.assertRaises(InvalidArgumentType): |
|
| 407 | self.gmp.create_credential(name='foo', credential_type='') |
|
| 408 | ||
| 409 | with self.assertRaises(InvalidArgumentType): |
|
| 410 | self.gmp.create_credential(name='foo', credential_type='bar') |
|
| 411 | ||
| 412 | def test_create_pw_credential_missing_password(self): |
|
| 413 | with self.assertRaises(RequiredArgument): |
|
| 414 | self.gmp.create_credential( |
|
| 415 | name='foo', credential_type=CredentialType.PASSWORD_ONLY |
|
| 416 | ) |
|
| 417 | ||
| 418 | def test_create_pw_credential(self): |
|
| 419 | self.gmp.create_credential( |
|
| 420 | name='foo', |
|
| 421 | credential_type=CredentialType.PASSWORD_ONLY, |
|
| 422 | password='foo', |
|
| 423 | ) |
|
| 424 | self.connection.send.has_been_called_with( |
|
| 425 | '<create_credential>' |
|
| 426 | '<name>foo</name>' |
|
| 427 | '<type>pw</type>' |
|
| 428 | '<password>foo</password>' |
|
| @@ 29-425 (lines=397) @@ | ||
| 26 | ) |
|
| 27 | ||
| 28 | ||
| 29 | class GmpCreateCredentialTestCase: |
|
| 30 | def test_create_up_credential_missing_name(self): |
|
| 31 | with self.assertRaises(RequiredArgument): |
|
| 32 | self.gmp.create_credential( |
|
| 33 | name='', |
|
| 34 | credential_type=CredentialType.USERNAME_PASSWORD, |
|
| 35 | login='foo', |
|
| 36 | ) |
|
| 37 | ||
| 38 | with self.assertRaises(RequiredArgument): |
|
| 39 | self.gmp.create_credential( |
|
| 40 | name=None, |
|
| 41 | credential_type=CredentialType.USERNAME_PASSWORD, |
|
| 42 | login='foo', |
|
| 43 | ) |
|
| 44 | ||
| 45 | def test_create_up_credential(self): |
|
| 46 | self.gmp.create_credential( |
|
| 47 | name='foo', |
|
| 48 | credential_type=CredentialType.USERNAME_PASSWORD, |
|
| 49 | comment='bar', |
|
| 50 | login='Max', |
|
| 51 | password='123', |
|
| 52 | ) |
|
| 53 | ||
| 54 | self.connection.send.has_been_called_with( |
|
| 55 | '<create_credential>' |
|
| 56 | '<name>foo</name>' |
|
| 57 | '<type>up</type>' |
|
| 58 | '<comment>bar</comment>' |
|
| 59 | '<login>Max</login>' |
|
| 60 | '<password>123</password>' |
|
| 61 | '</create_credential>' |
|
| 62 | ) |
|
| 63 | ||
| 64 | def test_create_up_credential_with_allow_insecure(self): |
|
| 65 | self.gmp.create_credential( |
|
| 66 | name='foo', |
|
| 67 | credential_type=CredentialType.USERNAME_PASSWORD, |
|
| 68 | comment='bar', |
|
| 69 | login='Max', |
|
| 70 | password='123', |
|
| 71 | allow_insecure=True, |
|
| 72 | ) |
|
| 73 | ||
| 74 | self.connection.send.has_been_called_with( |
|
| 75 | '<create_credential>' |
|
| 76 | '<name>foo</name>' |
|
| 77 | '<type>up</type>' |
|
| 78 | '<comment>bar</comment>' |
|
| 79 | '<allow_insecure>1</allow_insecure>' |
|
| 80 | '<login>Max</login>' |
|
| 81 | '<password>123</password>' |
|
| 82 | '</create_credential>' |
|
| 83 | ) |
|
| 84 | ||
| 85 | self.gmp.create_credential( |
|
| 86 | name='foo', |
|
| 87 | credential_type=CredentialType.USERNAME_PASSWORD, |
|
| 88 | comment='bar', |
|
| 89 | login='Max', |
|
| 90 | password='123', |
|
| 91 | allow_insecure=False, |
|
| 92 | ) |
|
| 93 | ||
| 94 | self.connection.send.has_been_called_with( |
|
| 95 | '<create_credential>' |
|
| 96 | '<name>foo</name>' |
|
| 97 | '<type>up</type>' |
|
| 98 | '<comment>bar</comment>' |
|
| 99 | '<allow_insecure>0</allow_insecure>' |
|
| 100 | '<login>Max</login>' |
|
| 101 | '<password>123</password>' |
|
| 102 | '</create_credential>' |
|
| 103 | ) |
|
| 104 | ||
| 105 | def test_create_cc_credential_missing_certificate(self): |
|
| 106 | with self.assertRaises(RequiredArgument): |
|
| 107 | self.gmp.create_credential( |
|
| 108 | name='foo', credential_type=CredentialType.CLIENT_CERTIFICATE |
|
| 109 | ) |
|
| 110 | ||
| 111 | def test_create_cc_credential(self): |
|
| 112 | self.gmp.create_credential( |
|
| 113 | name='foo', |
|
| 114 | credential_type=CredentialType.CLIENT_CERTIFICATE, |
|
| 115 | certificate='abcdef', |
|
| 116 | ) |
|
| 117 | ||
| 118 | self.connection.send.has_been_called_with( |
|
| 119 | '<create_credential>' |
|
| 120 | '<name>foo</name>' |
|
| 121 | '<type>cc</type>' |
|
| 122 | '<certificate>abcdef</certificate>' |
|
| 123 | '</create_credential>' |
|
| 124 | ) |
|
| 125 | ||
| 126 | def test_create_cc_credential_with_private_key(self): |
|
| 127 | self.gmp.create_credential( |
|
| 128 | name='foo', |
|
| 129 | credential_type=CredentialType.CLIENT_CERTIFICATE, |
|
| 130 | certificate='abcdef', |
|
| 131 | private_key='123456', |
|
| 132 | ) |
|
| 133 | ||
| 134 | self.connection.send.has_been_called_with( |
|
| 135 | '<create_credential>' |
|
| 136 | '<name>foo</name>' |
|
| 137 | '<type>cc</type>' |
|
| 138 | '<certificate>abcdef</certificate>' |
|
| 139 | '<key>' |
|
| 140 | '<private>123456</private>' |
|
| 141 | '</key>' |
|
| 142 | '</create_credential>' |
|
| 143 | ) |
|
| 144 | ||
| 145 | def test_create_usk_credential_missing_private_key(self): |
|
| 146 | with self.assertRaises(RequiredArgument): |
|
| 147 | self.gmp.create_credential( |
|
| 148 | name='foo', |
|
| 149 | credential_type=CredentialType.USERNAME_SSH_KEY, |
|
| 150 | login='foo', |
|
| 151 | ) |
|
| 152 | ||
| 153 | def test_create_usk_credential_missing_login(self): |
|
| 154 | with self.assertRaises(RequiredArgument): |
|
| 155 | self.gmp.create_credential( |
|
| 156 | name='foo', |
|
| 157 | credential_type=CredentialType.USERNAME_SSH_KEY, |
|
| 158 | private_key='123456', |
|
| 159 | ) |
|
| 160 | ||
| 161 | def test_create_usk_credential(self): |
|
| 162 | self.gmp.create_credential( |
|
| 163 | name='foo', |
|
| 164 | credential_type=CredentialType.USERNAME_SSH_KEY, |
|
| 165 | private_key='123456', |
|
| 166 | login='foo', |
|
| 167 | ) |
|
| 168 | ||
| 169 | self.connection.send.has_been_called_with( |
|
| 170 | '<create_credential>' |
|
| 171 | '<name>foo</name>' |
|
| 172 | '<type>usk</type>' |
|
| 173 | '<login>foo</login>' |
|
| 174 | '<key>' |
|
| 175 | '<private>123456</private>' |
|
| 176 | '</key>' |
|
| 177 | '</create_credential>' |
|
| 178 | ) |
|
| 179 | ||
| 180 | def test_create_usk_credential_with_key_phrase(self): |
|
| 181 | self.gmp.create_credential( |
|
| 182 | name='foo', |
|
| 183 | credential_type=CredentialType.USERNAME_SSH_KEY, |
|
| 184 | private_key='123456', |
|
| 185 | login='foo', |
|
| 186 | key_phrase='abcdef', |
|
| 187 | ) |
|
| 188 | ||
| 189 | self.connection.send.has_been_called_with( |
|
| 190 | '<create_credential>' |
|
| 191 | '<name>foo</name>' |
|
| 192 | '<type>usk</type>' |
|
| 193 | '<login>foo</login>' |
|
| 194 | '<key>' |
|
| 195 | '<private>123456</private>' |
|
| 196 | '<phrase>abcdef</phrase>' |
|
| 197 | '</key>' |
|
| 198 | '</create_credential>' |
|
| 199 | ) |
|
| 200 | ||
| 201 | def test_create_snmp_credential_invalid_auth_algorithm(self): |
|
| 202 | with self.assertRaises(InvalidArgumentType): |
|
| 203 | self.gmp.create_credential( |
|
| 204 | name='foo', credential_type=CredentialType.SNMP, login='foo' |
|
| 205 | ) |
|
| 206 | ||
| 207 | with self.assertRaises(InvalidArgumentType): |
|
| 208 | self.gmp.create_credential( |
|
| 209 | name='foo', |
|
| 210 | credential_type=CredentialType.SNMP, |
|
| 211 | login='foo', |
|
| 212 | auth_algorithm='', |
|
| 213 | ) |
|
| 214 | ||
| 215 | with self.assertRaises(InvalidArgumentType): |
|
| 216 | self.gmp.create_credential( |
|
| 217 | name='foo', |
|
| 218 | credential_type=CredentialType.SNMP, |
|
| 219 | login='foo', |
|
| 220 | auth_algorithm='bar', |
|
| 221 | ) |
|
| 222 | ||
| 223 | def test_create_snmp_credential_auth_algorithm_md5(self): |
|
| 224 | self.gmp.create_credential( |
|
| 225 | name='foo', |
|
| 226 | credential_type=CredentialType.SNMP, |
|
| 227 | login='foo', |
|
| 228 | auth_algorithm=SnmpAuthAlgorithm.MD5, |
|
| 229 | ) |
|
| 230 | ||
| 231 | self.connection.send.has_been_called_with( |
|
| 232 | '<create_credential>' |
|
| 233 | '<name>foo</name>' |
|
| 234 | '<type>snmp</type>' |
|
| 235 | '<login>foo</login>' |
|
| 236 | '<auth_algorithm>md5</auth_algorithm>' |
|
| 237 | '</create_credential>' |
|
| 238 | ) |
|
| 239 | ||
| 240 | def test_create_snmp_credential_auth_algorithm_sha1(self): |
|
| 241 | self.gmp.create_credential( |
|
| 242 | name='foo', |
|
| 243 | credential_type=CredentialType.SNMP, |
|
| 244 | login='foo', |
|
| 245 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 246 | ) |
|
| 247 | ||
| 248 | self.connection.send.has_been_called_with( |
|
| 249 | '<create_credential>' |
|
| 250 | '<name>foo</name>' |
|
| 251 | '<type>snmp</type>' |
|
| 252 | '<login>foo</login>' |
|
| 253 | '<auth_algorithm>sha1</auth_algorithm>' |
|
| 254 | '</create_credential>' |
|
| 255 | ) |
|
| 256 | ||
| 257 | def test_create_snmp_credential_with_community(self): |
|
| 258 | self.gmp.create_credential( |
|
| 259 | name='foo', |
|
| 260 | credential_type=CredentialType.SNMP, |
|
| 261 | login='foo', |
|
| 262 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 263 | community='ipsum', |
|
| 264 | ) |
|
| 265 | ||
| 266 | self.connection.send.has_been_called_with( |
|
| 267 | '<create_credential>' |
|
| 268 | '<name>foo</name>' |
|
| 269 | '<type>snmp</type>' |
|
| 270 | '<login>foo</login>' |
|
| 271 | '<auth_algorithm>sha1</auth_algorithm>' |
|
| 272 | '<community>ipsum</community>' |
|
| 273 | '</create_credential>' |
|
| 274 | ) |
|
| 275 | ||
| 276 | def test_create_snmp_credential_invalid_privacy_algorithm(self): |
|
| 277 | with self.assertRaises(InvalidArgumentType): |
|
| 278 | self.gmp.create_credential( |
|
| 279 | name='foo', |
|
| 280 | credential_type=CredentialType.SNMP, |
|
| 281 | login='foo', |
|
| 282 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 283 | privacy_algorithm='', |
|
| 284 | ) |
|
| 285 | ||
| 286 | with self.assertRaises(InvalidArgumentType): |
|
| 287 | self.gmp.create_credential( |
|
| 288 | name='foo', |
|
| 289 | credential_type=CredentialType.SNMP, |
|
| 290 | login='foo', |
|
| 291 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 292 | privacy_algorithm='foo', |
|
| 293 | ) |
|
| 294 | ||
| 295 | def test_create_snmp_credential_with_privacy_algorithm_aes(self): |
|
| 296 | self.gmp.create_credential( |
|
| 297 | name='foo', |
|
| 298 | credential_type=CredentialType.SNMP, |
|
| 299 | login='foo', |
|
| 300 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 301 | privacy_algorithm=SnmpPrivacyAlgorithm.AES, |
|
| 302 | ) |
|
| 303 | ||
| 304 | self.connection.send.has_been_called_with( |
|
| 305 | '<create_credential>' |
|
| 306 | '<name>foo</name>' |
|
| 307 | '<type>snmp</type>' |
|
| 308 | '<login>foo</login>' |
|
| 309 | '<auth_algorithm>sha1</auth_algorithm>' |
|
| 310 | '<privacy>' |
|
| 311 | '<algorithm>aes</algorithm>' |
|
| 312 | '</privacy>' |
|
| 313 | '</create_credential>' |
|
| 314 | ) |
|
| 315 | ||
| 316 | def test_create_snmp_credential_with_privacy_algorithm_des(self): |
|
| 317 | self.gmp.create_credential( |
|
| 318 | name='foo', |
|
| 319 | credential_type=CredentialType.SNMP, |
|
| 320 | login='foo', |
|
| 321 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 322 | privacy_algorithm=SnmpPrivacyAlgorithm.DES, |
|
| 323 | ) |
|
| 324 | ||
| 325 | self.connection.send.has_been_called_with( |
|
| 326 | '<create_credential>' |
|
| 327 | '<name>foo</name>' |
|
| 328 | '<type>snmp</type>' |
|
| 329 | '<login>foo</login>' |
|
| 330 | '<auth_algorithm>sha1</auth_algorithm>' |
|
| 331 | '<privacy>' |
|
| 332 | '<algorithm>des</algorithm>' |
|
| 333 | '</privacy>' |
|
| 334 | '</create_credential>' |
|
| 335 | ) |
|
| 336 | ||
| 337 | def test_create_snmp_credential_with_privacy_password(self): |
|
| 338 | self.gmp.create_credential( |
|
| 339 | name='foo', |
|
| 340 | credential_type=CredentialType.SNMP, |
|
| 341 | login='foo', |
|
| 342 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 343 | privacy_password='123', |
|
| 344 | ) |
|
| 345 | ||
| 346 | self.connection.send.has_been_called_with( |
|
| 347 | '<create_credential>' |
|
| 348 | '<name>foo</name>' |
|
| 349 | '<type>snmp</type>' |
|
| 350 | '<login>foo</login>' |
|
| 351 | '<auth_algorithm>sha1</auth_algorithm>' |
|
| 352 | '<privacy>' |
|
| 353 | '<password>123</password>' |
|
| 354 | '</privacy>' |
|
| 355 | '</create_credential>' |
|
| 356 | ) |
|
| 357 | ||
| 358 | def test_create_smime_credential(self): |
|
| 359 | self.gmp.create_credential( |
|
| 360 | name='foo', |
|
| 361 | credential_type=CredentialType.SMIME_CERTIFICATE, |
|
| 362 | certificate='ipsum', |
|
| 363 | ) |
|
| 364 | ||
| 365 | self.connection.send.has_been_called_with( |
|
| 366 | '<create_credential>' |
|
| 367 | '<name>foo</name>' |
|
| 368 | '<type>smime</type>' |
|
| 369 | '<certificate>ipsum</certificate>' |
|
| 370 | '</create_credential>' |
|
| 371 | ) |
|
| 372 | ||
| 373 | def test_create_smime_credential_missing_certificate(self): |
|
| 374 | with self.assertRaises(RequiredArgument): |
|
| 375 | self.gmp.create_credential( |
|
| 376 | name='foo', credential_type=CredentialType.SMIME_CERTIFICATE |
|
| 377 | ) |
|
| 378 | ||
| 379 | def test_create_pgp_credential(self): |
|
| 380 | self.gmp.create_credential( |
|
| 381 | name='foo', |
|
| 382 | credential_type=CredentialType.PGP_ENCRYPTION_KEY, |
|
| 383 | public_key='ipsum', |
|
| 384 | ) |
|
| 385 | ||
| 386 | self.connection.send.has_been_called_with( |
|
| 387 | '<create_credential>' |
|
| 388 | '<name>foo</name>' |
|
| 389 | '<type>pgp</type>' |
|
| 390 | '<key>' |
|
| 391 | '<public>ipsum</public>' |
|
| 392 | '</key>' |
|
| 393 | '</create_credential>' |
|
| 394 | ) |
|
| 395 | ||
| 396 | def test_create_pgp_credential_missing_public_key(self): |
|
| 397 | with self.assertRaises(RequiredArgument): |
|
| 398 | self.gmp.create_credential( |
|
| 399 | name='foo', credential_type=CredentialType.PGP_ENCRYPTION_KEY |
|
| 400 | ) |
|
| 401 | ||
| 402 | def test_create_credential_invalid_credential_type(self): |
|
| 403 | with self.assertRaises(InvalidArgumentType): |
|
| 404 | self.gmp.create_credential(name='foo', credential_type=None) |
|
| 405 | ||
| 406 | with self.assertRaises(InvalidArgumentType): |
|
| 407 | self.gmp.create_credential(name='foo', credential_type='') |
|
| 408 | ||
| 409 | with self.assertRaises(InvalidArgumentType): |
|
| 410 | self.gmp.create_credential(name='foo', credential_type='bar') |
|
| 411 | ||
| 412 | def test_create_pw_credential_missing_password(self): |
|
| 413 | with self.assertRaises(RequiredArgument): |
|
| 414 | self.gmp.create_credential( |
|
| 415 | name='foo', credential_type=CredentialType.PASSWORD_ONLY |
|
| 416 | ) |
|
| 417 | ||
| 418 | def test_create_pw_credential(self): |
|
| 419 | self.gmp.create_credential( |
|
| 420 | name='foo', |
|
| 421 | credential_type=CredentialType.PASSWORD_ONLY, |
|
| 422 | password='foo', |
|
| 423 | ) |
|
| 424 | self.connection.send.has_been_called_with( |
|
| 425 | '<create_credential>' |
|
| 426 | '<name>foo</name>' |
|
| 427 | '<type>pw</type>' |
|
| 428 | '<password>foo</password>' |
|
| @@ 29-425 (lines=397) @@ | ||
| 26 | ) |
|
| 27 | ||
| 28 | ||
| 29 | class GmpCreateCredentialTestCase: |
|
| 30 | def test_create_up_credential_missing_name(self): |
|
| 31 | with self.assertRaises(RequiredArgument): |
|
| 32 | self.gmp.create_credential( |
|
| 33 | name='', |
|
| 34 | credential_type=CredentialType.USERNAME_PASSWORD, |
|
| 35 | login='foo', |
|
| 36 | ) |
|
| 37 | ||
| 38 | with self.assertRaises(RequiredArgument): |
|
| 39 | self.gmp.create_credential( |
|
| 40 | name=None, |
|
| 41 | credential_type=CredentialType.USERNAME_PASSWORD, |
|
| 42 | login='foo', |
|
| 43 | ) |
|
| 44 | ||
| 45 | def test_create_up_credential(self): |
|
| 46 | self.gmp.create_credential( |
|
| 47 | name='foo', |
|
| 48 | credential_type=CredentialType.USERNAME_PASSWORD, |
|
| 49 | comment='bar', |
|
| 50 | login='Max', |
|
| 51 | password='123', |
|
| 52 | ) |
|
| 53 | ||
| 54 | self.connection.send.has_been_called_with( |
|
| 55 | '<create_credential>' |
|
| 56 | '<name>foo</name>' |
|
| 57 | '<type>up</type>' |
|
| 58 | '<comment>bar</comment>' |
|
| 59 | '<login>Max</login>' |
|
| 60 | '<password>123</password>' |
|
| 61 | '</create_credential>' |
|
| 62 | ) |
|
| 63 | ||
| 64 | def test_create_up_credential_with_allow_insecure(self): |
|
| 65 | self.gmp.create_credential( |
|
| 66 | name='foo', |
|
| 67 | credential_type=CredentialType.USERNAME_PASSWORD, |
|
| 68 | comment='bar', |
|
| 69 | login='Max', |
|
| 70 | password='123', |
|
| 71 | allow_insecure=True, |
|
| 72 | ) |
|
| 73 | ||
| 74 | self.connection.send.has_been_called_with( |
|
| 75 | '<create_credential>' |
|
| 76 | '<name>foo</name>' |
|
| 77 | '<type>up</type>' |
|
| 78 | '<comment>bar</comment>' |
|
| 79 | '<allow_insecure>1</allow_insecure>' |
|
| 80 | '<login>Max</login>' |
|
| 81 | '<password>123</password>' |
|
| 82 | '</create_credential>' |
|
| 83 | ) |
|
| 84 | ||
| 85 | self.gmp.create_credential( |
|
| 86 | name='foo', |
|
| 87 | credential_type=CredentialType.USERNAME_PASSWORD, |
|
| 88 | comment='bar', |
|
| 89 | login='Max', |
|
| 90 | password='123', |
|
| 91 | allow_insecure=False, |
|
| 92 | ) |
|
| 93 | ||
| 94 | self.connection.send.has_been_called_with( |
|
| 95 | '<create_credential>' |
|
| 96 | '<name>foo</name>' |
|
| 97 | '<type>up</type>' |
|
| 98 | '<comment>bar</comment>' |
|
| 99 | '<allow_insecure>0</allow_insecure>' |
|
| 100 | '<login>Max</login>' |
|
| 101 | '<password>123</password>' |
|
| 102 | '</create_credential>' |
|
| 103 | ) |
|
| 104 | ||
| 105 | def test_create_cc_credential_missing_certificate(self): |
|
| 106 | with self.assertRaises(RequiredArgument): |
|
| 107 | self.gmp.create_credential( |
|
| 108 | name='foo', credential_type=CredentialType.CLIENT_CERTIFICATE |
|
| 109 | ) |
|
| 110 | ||
| 111 | def test_create_cc_credential(self): |
|
| 112 | self.gmp.create_credential( |
|
| 113 | name='foo', |
|
| 114 | credential_type=CredentialType.CLIENT_CERTIFICATE, |
|
| 115 | certificate='abcdef', |
|
| 116 | ) |
|
| 117 | ||
| 118 | self.connection.send.has_been_called_with( |
|
| 119 | '<create_credential>' |
|
| 120 | '<name>foo</name>' |
|
| 121 | '<type>cc</type>' |
|
| 122 | '<certificate>abcdef</certificate>' |
|
| 123 | '</create_credential>' |
|
| 124 | ) |
|
| 125 | ||
| 126 | def test_create_cc_credential_with_private_key(self): |
|
| 127 | self.gmp.create_credential( |
|
| 128 | name='foo', |
|
| 129 | credential_type=CredentialType.CLIENT_CERTIFICATE, |
|
| 130 | certificate='abcdef', |
|
| 131 | private_key='123456', |
|
| 132 | ) |
|
| 133 | ||
| 134 | self.connection.send.has_been_called_with( |
|
| 135 | '<create_credential>' |
|
| 136 | '<name>foo</name>' |
|
| 137 | '<type>cc</type>' |
|
| 138 | '<certificate>abcdef</certificate>' |
|
| 139 | '<key>' |
|
| 140 | '<private>123456</private>' |
|
| 141 | '</key>' |
|
| 142 | '</create_credential>' |
|
| 143 | ) |
|
| 144 | ||
| 145 | def test_create_usk_credential_missing_private_key(self): |
|
| 146 | with self.assertRaises(RequiredArgument): |
|
| 147 | self.gmp.create_credential( |
|
| 148 | name='foo', |
|
| 149 | credential_type=CredentialType.USERNAME_SSH_KEY, |
|
| 150 | login='foo', |
|
| 151 | ) |
|
| 152 | ||
| 153 | def test_create_usk_credential_missing_login(self): |
|
| 154 | with self.assertRaises(RequiredArgument): |
|
| 155 | self.gmp.create_credential( |
|
| 156 | name='foo', |
|
| 157 | credential_type=CredentialType.USERNAME_SSH_KEY, |
|
| 158 | private_key='123456', |
|
| 159 | ) |
|
| 160 | ||
| 161 | def test_create_usk_credential(self): |
|
| 162 | self.gmp.create_credential( |
|
| 163 | name='foo', |
|
| 164 | credential_type=CredentialType.USERNAME_SSH_KEY, |
|
| 165 | private_key='123456', |
|
| 166 | login='foo', |
|
| 167 | ) |
|
| 168 | ||
| 169 | self.connection.send.has_been_called_with( |
|
| 170 | '<create_credential>' |
|
| 171 | '<name>foo</name>' |
|
| 172 | '<type>usk</type>' |
|
| 173 | '<login>foo</login>' |
|
| 174 | '<key>' |
|
| 175 | '<private>123456</private>' |
|
| 176 | '</key>' |
|
| 177 | '</create_credential>' |
|
| 178 | ) |
|
| 179 | ||
| 180 | def test_create_usk_credential_with_key_phrase(self): |
|
| 181 | self.gmp.create_credential( |
|
| 182 | name='foo', |
|
| 183 | credential_type=CredentialType.USERNAME_SSH_KEY, |
|
| 184 | private_key='123456', |
|
| 185 | login='foo', |
|
| 186 | key_phrase='abcdef', |
|
| 187 | ) |
|
| 188 | ||
| 189 | self.connection.send.has_been_called_with( |
|
| 190 | '<create_credential>' |
|
| 191 | '<name>foo</name>' |
|
| 192 | '<type>usk</type>' |
|
| 193 | '<login>foo</login>' |
|
| 194 | '<key>' |
|
| 195 | '<private>123456</private>' |
|
| 196 | '<phrase>abcdef</phrase>' |
|
| 197 | '</key>' |
|
| 198 | '</create_credential>' |
|
| 199 | ) |
|
| 200 | ||
| 201 | def test_create_snmp_credential_invalid_auth_algorithm(self): |
|
| 202 | with self.assertRaises(InvalidArgumentType): |
|
| 203 | self.gmp.create_credential( |
|
| 204 | name='foo', credential_type=CredentialType.SNMP, login='foo' |
|
| 205 | ) |
|
| 206 | ||
| 207 | with self.assertRaises(InvalidArgumentType): |
|
| 208 | self.gmp.create_credential( |
|
| 209 | name='foo', |
|
| 210 | credential_type=CredentialType.SNMP, |
|
| 211 | login='foo', |
|
| 212 | auth_algorithm='', |
|
| 213 | ) |
|
| 214 | ||
| 215 | with self.assertRaises(InvalidArgumentType): |
|
| 216 | self.gmp.create_credential( |
|
| 217 | name='foo', |
|
| 218 | credential_type=CredentialType.SNMP, |
|
| 219 | login='foo', |
|
| 220 | auth_algorithm='bar', |
|
| 221 | ) |
|
| 222 | ||
| 223 | def test_create_snmp_credential_auth_algorithm_md5(self): |
|
| 224 | self.gmp.create_credential( |
|
| 225 | name='foo', |
|
| 226 | credential_type=CredentialType.SNMP, |
|
| 227 | login='foo', |
|
| 228 | auth_algorithm=SnmpAuthAlgorithm.MD5, |
|
| 229 | ) |
|
| 230 | ||
| 231 | self.connection.send.has_been_called_with( |
|
| 232 | '<create_credential>' |
|
| 233 | '<name>foo</name>' |
|
| 234 | '<type>snmp</type>' |
|
| 235 | '<login>foo</login>' |
|
| 236 | '<auth_algorithm>md5</auth_algorithm>' |
|
| 237 | '</create_credential>' |
|
| 238 | ) |
|
| 239 | ||
| 240 | def test_create_snmp_credential_auth_algorithm_sha1(self): |
|
| 241 | self.gmp.create_credential( |
|
| 242 | name='foo', |
|
| 243 | credential_type=CredentialType.SNMP, |
|
| 244 | login='foo', |
|
| 245 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 246 | ) |
|
| 247 | ||
| 248 | self.connection.send.has_been_called_with( |
|
| 249 | '<create_credential>' |
|
| 250 | '<name>foo</name>' |
|
| 251 | '<type>snmp</type>' |
|
| 252 | '<login>foo</login>' |
|
| 253 | '<auth_algorithm>sha1</auth_algorithm>' |
|
| 254 | '</create_credential>' |
|
| 255 | ) |
|
| 256 | ||
| 257 | def test_create_snmp_credential_with_community(self): |
|
| 258 | self.gmp.create_credential( |
|
| 259 | name='foo', |
|
| 260 | credential_type=CredentialType.SNMP, |
|
| 261 | login='foo', |
|
| 262 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 263 | community='ipsum', |
|
| 264 | ) |
|
| 265 | ||
| 266 | self.connection.send.has_been_called_with( |
|
| 267 | '<create_credential>' |
|
| 268 | '<name>foo</name>' |
|
| 269 | '<type>snmp</type>' |
|
| 270 | '<login>foo</login>' |
|
| 271 | '<auth_algorithm>sha1</auth_algorithm>' |
|
| 272 | '<community>ipsum</community>' |
|
| 273 | '</create_credential>' |
|
| 274 | ) |
|
| 275 | ||
| 276 | def test_create_snmp_credential_invalid_privacy_algorithm(self): |
|
| 277 | with self.assertRaises(InvalidArgumentType): |
|
| 278 | self.gmp.create_credential( |
|
| 279 | name='foo', |
|
| 280 | credential_type=CredentialType.SNMP, |
|
| 281 | login='foo', |
|
| 282 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 283 | privacy_algorithm='', |
|
| 284 | ) |
|
| 285 | ||
| 286 | with self.assertRaises(InvalidArgumentType): |
|
| 287 | self.gmp.create_credential( |
|
| 288 | name='foo', |
|
| 289 | credential_type=CredentialType.SNMP, |
|
| 290 | login='foo', |
|
| 291 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 292 | privacy_algorithm='foo', |
|
| 293 | ) |
|
| 294 | ||
| 295 | def test_create_snmp_credential_with_privacy_algorithm_aes(self): |
|
| 296 | self.gmp.create_credential( |
|
| 297 | name='foo', |
|
| 298 | credential_type=CredentialType.SNMP, |
|
| 299 | login='foo', |
|
| 300 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 301 | privacy_algorithm=SnmpPrivacyAlgorithm.AES, |
|
| 302 | ) |
|
| 303 | ||
| 304 | self.connection.send.has_been_called_with( |
|
| 305 | '<create_credential>' |
|
| 306 | '<name>foo</name>' |
|
| 307 | '<type>snmp</type>' |
|
| 308 | '<login>foo</login>' |
|
| 309 | '<auth_algorithm>sha1</auth_algorithm>' |
|
| 310 | '<privacy>' |
|
| 311 | '<algorithm>aes</algorithm>' |
|
| 312 | '</privacy>' |
|
| 313 | '</create_credential>' |
|
| 314 | ) |
|
| 315 | ||
| 316 | def test_create_snmp_credential_with_privacy_algorithm_des(self): |
|
| 317 | self.gmp.create_credential( |
|
| 318 | name='foo', |
|
| 319 | credential_type=CredentialType.SNMP, |
|
| 320 | login='foo', |
|
| 321 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 322 | privacy_algorithm=SnmpPrivacyAlgorithm.DES, |
|
| 323 | ) |
|
| 324 | ||
| 325 | self.connection.send.has_been_called_with( |
|
| 326 | '<create_credential>' |
|
| 327 | '<name>foo</name>' |
|
| 328 | '<type>snmp</type>' |
|
| 329 | '<login>foo</login>' |
|
| 330 | '<auth_algorithm>sha1</auth_algorithm>' |
|
| 331 | '<privacy>' |
|
| 332 | '<algorithm>des</algorithm>' |
|
| 333 | '</privacy>' |
|
| 334 | '</create_credential>' |
|
| 335 | ) |
|
| 336 | ||
| 337 | def test_create_snmp_credential_with_privacy_password(self): |
|
| 338 | self.gmp.create_credential( |
|
| 339 | name='foo', |
|
| 340 | credential_type=CredentialType.SNMP, |
|
| 341 | login='foo', |
|
| 342 | auth_algorithm=SnmpAuthAlgorithm.SHA1, |
|
| 343 | privacy_password='123', |
|
| 344 | ) |
|
| 345 | ||
| 346 | self.connection.send.has_been_called_with( |
|
| 347 | '<create_credential>' |
|
| 348 | '<name>foo</name>' |
|
| 349 | '<type>snmp</type>' |
|
| 350 | '<login>foo</login>' |
|
| 351 | '<auth_algorithm>sha1</auth_algorithm>' |
|
| 352 | '<privacy>' |
|
| 353 | '<password>123</password>' |
|
| 354 | '</privacy>' |
|
| 355 | '</create_credential>' |
|
| 356 | ) |
|
| 357 | ||
| 358 | def test_create_smime_credential(self): |
|
| 359 | self.gmp.create_credential( |
|
| 360 | name='foo', |
|
| 361 | credential_type=CredentialType.SMIME_CERTIFICATE, |
|
| 362 | certificate='ipsum', |
|
| 363 | ) |
|
| 364 | ||
| 365 | self.connection.send.has_been_called_with( |
|
| 366 | '<create_credential>' |
|
| 367 | '<name>foo</name>' |
|
| 368 | '<type>smime</type>' |
|
| 369 | '<certificate>ipsum</certificate>' |
|
| 370 | '</create_credential>' |
|
| 371 | ) |
|
| 372 | ||
| 373 | def test_create_smime_credential_missing_certificate(self): |
|
| 374 | with self.assertRaises(RequiredArgument): |
|
| 375 | self.gmp.create_credential( |
|
| 376 | name='foo', credential_type=CredentialType.SMIME_CERTIFICATE |
|
| 377 | ) |
|
| 378 | ||
| 379 | def test_create_pgp_credential(self): |
|
| 380 | self.gmp.create_credential( |
|
| 381 | name='foo', |
|
| 382 | credential_type=CredentialType.PGP_ENCRYPTION_KEY, |
|
| 383 | public_key='ipsum', |
|
| 384 | ) |
|
| 385 | ||
| 386 | self.connection.send.has_been_called_with( |
|
| 387 | '<create_credential>' |
|
| 388 | '<name>foo</name>' |
|
| 389 | '<type>pgp</type>' |
|
| 390 | '<key>' |
|
| 391 | '<public>ipsum</public>' |
|
| 392 | '</key>' |
|
| 393 | '</create_credential>' |
|
| 394 | ) |
|
| 395 | ||
| 396 | def test_create_pgp_credential_missing_public_key(self): |
|
| 397 | with self.assertRaises(RequiredArgument): |
|
| 398 | self.gmp.create_credential( |
|
| 399 | name='foo', credential_type=CredentialType.PGP_ENCRYPTION_KEY |
|
| 400 | ) |
|
| 401 | ||
| 402 | def test_create_credential_invalid_credential_type(self): |
|
| 403 | with self.assertRaises(InvalidArgumentType): |
|
| 404 | self.gmp.create_credential(name='foo', credential_type=None) |
|
| 405 | ||
| 406 | with self.assertRaises(InvalidArgumentType): |
|
| 407 | self.gmp.create_credential(name='foo', credential_type='') |
|
| 408 | ||
| 409 | with self.assertRaises(InvalidArgumentType): |
|
| 410 | self.gmp.create_credential(name='foo', credential_type='bar') |
|
| 411 | ||
| 412 | def test_create_pw_credential_missing_password(self): |
|
| 413 | with self.assertRaises(RequiredArgument): |
|
| 414 | self.gmp.create_credential( |
|
| 415 | name='foo', credential_type=CredentialType.PASSWORD_ONLY |
|
| 416 | ) |
|
| 417 | ||
| 418 | def test_create_pw_credential(self): |
|
| 419 | self.gmp.create_credential( |
|
| 420 | name='foo', |
|
| 421 | credential_type=CredentialType.PASSWORD_ONLY, |
|
| 422 | password='foo', |
|
| 423 | ) |
|
| 424 | self.connection.send.has_been_called_with( |
|
| 425 | '<create_credential>' |
|
| 426 | '<name>foo</name>' |
|
| 427 | '<type>pw</type>' |
|
| 428 | '<password>foo</password>' |
|
