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
|
# SPDX-FileCopyrightText: 2020-2024 Greenbone AG
#
# SPDX-License-Identifier: GPL-3.0-or-later
#
import unittest
from unittest.mock import MagicMock
from gvm.connections import (
DEFAULT_TIMEOUT,
GvmConnection,
)
from gvm.connections._connection import AbstractGvmConnection
from gvm.errors import GvmError
class TestConnection(AbstractGvmConnection):
def connect(self) -> None:
pass
class GvmConnectionTestCase(unittest.TestCase):
# pylint: disable=protected-access
def test_init_no_args(self):
connection = TestConnection()
self.assertIsNone(connection._socket)
self.assertEqual(connection._timeout, DEFAULT_TIMEOUT)
def test_init_with_none(self):
connection = TestConnection(timeout=None)
self.assertIsNone(connection._socket)
self.assertEqual(connection._timeout, DEFAULT_TIMEOUT)
def test_read_no_data(self):
read_mock = MagicMock()
read_mock.return_value = None
connection = TestConnection()
connection._read = read_mock
with self.assertRaises(GvmError, msg="Remote closed the connection"):
connection.read()
def test_read_trigger_timeout(self):
# mocking the response into two parts, so we run into the timeout
# check in the loop
read_mock = MagicMock()
read_mock.side_effect = [b"<foo>xyz<bar></bar>", b"</foo>"]
connection = TestConnection(timeout=0)
connection._read = read_mock
with self.assertRaises(
GvmError, msg="Timeout while reading the response"
):
connection.read()
def test_is_gvm_connection(self):
connection = TestConnection()
self.assertTrue(isinstance(connection, GvmConnection))
|
