Completed
Push — master ( da306e...c5fe8d )
by Jaspar
17s queued 15s
created

GmpDeleteAlertTestMixin.test_delete_ultimate()   A

Complexity

Conditions 1

Size

Total Lines 5
Code Lines 4

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 1
eloc 4
nop 1
dl 0
loc 5
rs 10
c 0
b 0
f 0
1
# -*- coding: utf-8 -*-
2
# Copyright (C) 2018-2021 Greenbone Networks GmbH
3
#
4
# SPDX-License-Identifier: GPL-3.0-or-later
5
#
6
# This program is free software: you can redistribute it and/or modify
7
# it under the terms of the GNU General Public License as published by
8
# the Free Software Foundation, either version 3 of the License, or
9
# (at your option) any later version.
10
#
11
# This program is distributed in the hope that it will be useful,
12
# but WITHOUT ANY WARRANTY; without even the implied warranty of
13
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14
# GNU General Public License for more details.
15
#
16
# You should have received a copy of the GNU General Public License
17
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
18
19
from gvm.errors import RequiredArgument
20
21
22
class GmpDeleteAlertTestMixin:
23
    def test_delete(self):
24
        self.gmp.delete_alert('a1')
25
26
        self.connection.send.has_been_called_with(
27
            '<delete_alert alert_id="a1" ultimate="0"/>'
28
        )
29
30
    def test_delete_ultimate(self):
31
        self.gmp.delete_alert('a1', ultimate=True)
32
33
        self.connection.send.has_been_called_with(
34
            '<delete_alert alert_id="a1" ultimate="1"/>'
35
        )
36
37
    def test_missing_alert_id(self):
38
        with self.assertRaises(RequiredArgument):
39
            self.gmp.delete_alert(None)
40
41
        with self.assertRaises(RequiredArgument):
42
            self.gmp.delete_alert('')
43