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 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114
|
# SPDX-FileCopyrightText: 2024 Greenbone AG
#
# SPDX-License-Identifier: GPL-3.0-or-later
import unittest
from gvm.errors import RequiredArgument
from gvm.protocols.gmp.requests.v224 import OperatingSystems
class OperatingSystemsTestCase(unittest.TestCase):
def test_delete_operating_system(self):
request = OperatingSystems.delete_operating_system(
"operating_system_id"
)
self.assertEqual(
bytes(request),
b'<delete_asset asset_id="operating_system_id"/>',
)
def test_delete_operating_system_missing_operating_system_id(self):
with self.assertRaises(RequiredArgument):
OperatingSystems.delete_operating_system(None)
with self.assertRaises(RequiredArgument):
OperatingSystems.delete_operating_system("")
def test_get_operating_systems(self):
request = OperatingSystems.get_operating_systems()
self.assertEqual(
bytes(request),
b'<get_assets type="os"/>',
)
def test_get_operating_systems_with_filter_string(self):
request = OperatingSystems.get_operating_systems(
filter_string="filter_string"
)
self.assertEqual(
bytes(request),
b'<get_assets type="os" filter="filter_string"/>',
)
def test_get_operating_systems_with_filter_id(self):
request = OperatingSystems.get_operating_systems(filter_id="filter_id")
self.assertEqual(
bytes(request),
b'<get_assets type="os" filt_id="filter_id"/>',
)
def test_get_operating_systems_with_details(self):
request = OperatingSystems.get_operating_systems(details=True)
self.assertEqual(
bytes(request),
b'<get_assets type="os" details="1"/>',
)
request = OperatingSystems.get_operating_systems(details=False)
self.assertEqual(
bytes(request),
b'<get_assets type="os" details="0"/>',
)
def test_get_operating_system(self):
request = OperatingSystems.get_operating_system("operating_system_id")
self.assertEqual(
bytes(request),
b'<get_assets asset_id="operating_system_id" type="os"/>',
)
def test_get_operating_system_with_details(self):
request = OperatingSystems.get_operating_system(
"operating_system_id", details=True
)
self.assertEqual(
bytes(request),
b'<get_assets asset_id="operating_system_id" type="os" details="1"/>',
)
request = OperatingSystems.get_operating_system(
"operating_system_id", details=False
)
self.assertEqual(
bytes(request),
b'<get_assets asset_id="operating_system_id" type="os" details="0"/>',
)
def test_modify_operating_system(self):
request = OperatingSystems.modify_operating_system(
"operating_system_id"
)
self.assertEqual(
bytes(request),
b'<modify_asset asset_id="operating_system_id">'
b"<comment></comment>"
b"</modify_asset>",
)
def test_modify_operating_system_with_comment(self):
request = OperatingSystems.modify_operating_system(
"operating_system_id", comment="comment"
)
self.assertEqual(
bytes(request),
b'<modify_asset asset_id="operating_system_id">'
b"<comment>comment</comment>"
b"</modify_asset>",
)
def test_modify_operating_system_missing_operating_system_id(self):
with self.assertRaises(RequiredArgument):
OperatingSystems.modify_operating_system(None)
with self.assertRaises(RequiredArgument):
OperatingSystems.modify_operating_system("")
|