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 19from gvm.errors import GvmError 20 21 22class GmpGetPolicyTestMixin: 23 def test_get_policy(self): 24 self.gmp.get_policy('a1') 25 26 self.connection.send.has_been_called_with( 27 '<get_configs config_id="a1" usage_type="policy" details="1"/>' 28 ) 29 30 def test_get_policy_with_audits(self): 31 self.gmp.get_policy('a1', audits=True) 32 33 self.connection.send.has_been_called_with( 34 '<get_configs config_id="a1" ' 35 'usage_type="policy" tasks="1" details="1"/>' 36 ) 37 38 def test_fail_without_policy_id(self): 39 with self.assertRaises(GvmError): 40 self.gmp.get_policy(None) 41 42 with self.assertRaises(GvmError): 43 self.gmp.get_policy('') 44