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 19import unittest 20 21from gvm.errors import RequiredArgument 22from gvm.protocols.ospv1 import Osp 23 24from .. import MockConnection 25 26 27class OSPStopScanTestCase(unittest.TestCase): 28 def setUp(self): 29 self.connection = MockConnection() 30 self.osp = Osp(self.connection) 31 32 def test_stop_scan(self): 33 self.osp.stop_scan(scan_id="123-456") 34 35 self.connection.send.has_been_called_with( 36 '<stop_scan scan_id="123-456"/>' 37 ) 38 39 def test_stop_scan_without_id(self): 40 with self.assertRaises(RequiredArgument): 41 self.osp.stop_scan(None) 42 43 with self.assertRaises(RequiredArgument): 44 self.osp.stop_scan('') 45 46 47if __name__ == '__main__': 48 unittest.main() 49