1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
# SPDX-FileCopyrightText: 2018-2024 Greenbone AG
#
# SPDX-License-Identifier: GPL-3.0-or-later
#
from gvm.errors import InvalidArgument, RequiredArgument
from gvm.protocols.gmp.requests.v224 import CredentialFormat
class GmpGetCredentialTestMixin:
def test_get_credential(self):
self.gmp.get_credential("id")
self.connection.send.has_been_called_with(
b'<get_credentials credential_id="id"/>'
)
def test_get_credential_missing_credential_id(self):
with self.assertRaises(RequiredArgument):
self.gmp.get_credential(None)
def test_get_credential_invalid_credential_id(self):
with self.assertRaises(RequiredArgument):
self.gmp.get_credential(credential_id=None)
with self.assertRaises(RequiredArgument):
self.gmp.get_credential("")
def test_get_credential_with_credential_format(self):
self.gmp.get_credential("id", credential_format=CredentialFormat.KEY)
self.connection.send.has_been_called_with(
b'<get_credentials credential_id="id" format="key"/>'
)
self.gmp.get_credential("id", credential_format=CredentialFormat.RPM)
self.connection.send.has_been_called_with(
b'<get_credentials credential_id="id" format="rpm"/>'
)
self.gmp.get_credential("id", credential_format=CredentialFormat.DEB)
self.connection.send.has_been_called_with(
b'<get_credentials credential_id="id" format="deb"/>'
)
self.gmp.get_credential("id", credential_format=CredentialFormat.EXE)
self.connection.send.has_been_called_with(
b'<get_credentials credential_id="id" format="exe"/>'
)
self.gmp.get_credential("id", credential_format=CredentialFormat.PEM)
self.connection.send.has_been_called_with(
b'<get_credentials credential_id="id" format="pem"/>'
)
def test_get_credential_with_invalid_credential_format(self):
with self.assertRaises(InvalidArgument):
self.gmp.get_credential("id", credential_format="foo")
def test_get_credential_with_scanners(self):
self.gmp.get_credential("id", scanners=True)
self.connection.send.has_been_called_with(
b'<get_credentials credential_id="id" scanners="1"/>'
)
self.gmp.get_credential("id", scanners=False)
self.connection.send.has_been_called_with(
b'<get_credentials credential_id="id" scanners="0"/>'
)
def test_get_credential_with_targets(self):
self.gmp.get_credential("id", targets=True)
self.connection.send.has_been_called_with(
b'<get_credentials credential_id="id" targets="1"/>'
)
self.gmp.get_credential("id", targets=False)
self.connection.send.has_been_called_with(
b'<get_credentials credential_id="id" targets="0"/>'
)
|
