8
8
from unittest .mock import Mock , patch
9
9
10
10
from gvm .errors import InvalidArgumentType
11
- from gvm .http .core .headers import ContentType
12
- from gvm .http .core .response import HttpResponse
13
- from gvm .http .openvasd .openvasd1 import OpenvasdHttpApiV1
11
+ from gvm .protocols . http .core .headers import ContentType
12
+ from gvm .protocols . http .core .response import HttpResponse
13
+ from gvm .protocols . http .openvasd .openvasd1 import OpenvasdHttpApiV1
14
14
15
15
16
16
def new_mock_empty_response (
@@ -29,21 +29,21 @@ def new_mock_empty_response(
29
29
30
30
class OpenvasdHttpApiV1TestCase (unittest .TestCase ):
31
31
32
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
32
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
33
33
def test_init (self , mock_connector : Mock ):
34
34
api = OpenvasdHttpApiV1 (mock_connector )
35
35
mock_connector .update_headers .assert_not_called ()
36
36
self .assertIsNotNone (api )
37
37
38
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
38
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
39
39
def test_init_with_api_key (self , mock_connector : Mock ):
40
40
api = OpenvasdHttpApiV1 (mock_connector , api_key = "my-API-key" )
41
41
mock_connector .update_headers .assert_called_once_with (
42
42
{"X-API-KEY" : "my-API-key" }
43
43
)
44
44
self .assertIsNotNone (api )
45
45
46
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
46
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
47
47
def test_get_health_alive (self , mock_connector : Mock ):
48
48
expected_response = new_mock_empty_response ()
49
49
mock_connector .get .return_value = expected_response
@@ -55,7 +55,7 @@ def test_get_health_alive(self, mock_connector: Mock):
55
55
)
56
56
self .assertEqual (expected_response , response )
57
57
58
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
58
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
59
59
def test_get_health_ready (self , mock_connector : Mock ):
60
60
expected_response = new_mock_empty_response ()
61
61
mock_connector .get .return_value = expected_response
@@ -67,7 +67,7 @@ def test_get_health_ready(self, mock_connector: Mock):
67
67
)
68
68
self .assertEqual (expected_response , response )
69
69
70
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
70
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
71
71
def test_get_health_started (self , mock_connector : Mock ):
72
72
expected_response = new_mock_empty_response ()
73
73
mock_connector .get .return_value = expected_response
@@ -79,7 +79,7 @@ def test_get_health_started(self, mock_connector: Mock):
79
79
)
80
80
self .assertEqual (expected_response , response )
81
81
82
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
82
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
83
83
def test_get_notus_os_list (self , mock_connector : Mock ):
84
84
expected_response = new_mock_empty_response ()
85
85
mock_connector .get .return_value = expected_response
@@ -91,7 +91,7 @@ def test_get_notus_os_list(self, mock_connector: Mock):
91
91
)
92
92
self .assertEqual (expected_response , response )
93
93
94
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
94
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
95
95
def test_run_notus_scan (self , mock_connector : Mock ):
96
96
expected_response = new_mock_empty_response ()
97
97
mock_connector .post_json .return_value = expected_response
@@ -105,7 +105,7 @@ def test_run_notus_scan(self, mock_connector: Mock):
105
105
)
106
106
self .assertEqual (expected_response , response )
107
107
108
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
108
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
109
109
def test_get_scan_preferences (self , mock_connector : Mock ):
110
110
expected_response = new_mock_empty_response ()
111
111
mock_connector .get .return_value = expected_response
@@ -117,7 +117,7 @@ def test_get_scan_preferences(self, mock_connector: Mock):
117
117
)
118
118
self .assertEqual (expected_response , response )
119
119
120
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
120
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
121
121
def test_create_scan (self , mock_connector : Mock ):
122
122
expected_response = new_mock_empty_response ()
123
123
mock_connector .post_json .return_value = expected_response
@@ -156,7 +156,7 @@ def test_create_scan(self, mock_connector: Mock):
156
156
)
157
157
self .assertEqual (expected_response , response )
158
158
159
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
159
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
160
160
def test_delete_scan (self , mock_connector : Mock ):
161
161
expected_response = new_mock_empty_response ()
162
162
mock_connector .delete .return_value = expected_response
@@ -168,7 +168,7 @@ def test_delete_scan(self, mock_connector: Mock):
168
168
)
169
169
self .assertEqual (expected_response , response )
170
170
171
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
171
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
172
172
def test_get_scans (self , mock_connector : Mock ):
173
173
expected_response = new_mock_empty_response ()
174
174
mock_connector .get .return_value = expected_response
@@ -180,7 +180,7 @@ def test_get_scans(self, mock_connector: Mock):
180
180
)
181
181
self .assertEqual (expected_response , response )
182
182
183
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
183
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
184
184
def test_get_scan (self , mock_connector : Mock ):
185
185
expected_response = new_mock_empty_response ()
186
186
mock_connector .get .return_value = expected_response
@@ -192,7 +192,7 @@ def test_get_scan(self, mock_connector: Mock):
192
192
)
193
193
self .assertEqual (expected_response , response )
194
194
195
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
195
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
196
196
def test_get_scan_results (self , mock_connector : Mock ):
197
197
expected_response = new_mock_empty_response ()
198
198
mock_connector .get .return_value = expected_response
@@ -204,7 +204,7 @@ def test_get_scan_results(self, mock_connector: Mock):
204
204
)
205
205
self .assertEqual (expected_response , response )
206
206
207
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
207
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
208
208
def test_get_scan_results_with_ranges (self , mock_connector : Mock ):
209
209
expected_response = new_mock_empty_response ()
210
210
mock_connector .get .return_value = expected_response
@@ -239,7 +239,7 @@ def test_get_scan_results_with_ranges(self, mock_connector: Mock):
239
239
)
240
240
self .assertEqual (expected_response , response )
241
241
242
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
242
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
243
243
def test_get_scan_results_with_invalid_ranges (self , mock_connector : Mock ):
244
244
expected_response = new_mock_empty_response ()
245
245
mock_connector .get .return_value = expected_response
@@ -263,7 +263,7 @@ def test_get_scan_results_with_invalid_ranges(self, mock_connector: Mock):
263
263
range_end = "invalid" ,
264
264
)
265
265
266
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
266
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
267
267
def test_get_scan_result (self , mock_connector : Mock ):
268
268
expected_response = new_mock_empty_response ()
269
269
mock_connector .get .return_value = expected_response
@@ -275,7 +275,7 @@ def test_get_scan_result(self, mock_connector: Mock):
275
275
)
276
276
self .assertEqual (expected_response , response )
277
277
278
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
278
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
279
279
def test_get_scan_status (self , mock_connector : Mock ):
280
280
expected_response = new_mock_empty_response ()
281
281
mock_connector .get .return_value = expected_response
@@ -287,7 +287,7 @@ def test_get_scan_status(self, mock_connector: Mock):
287
287
)
288
288
self .assertEqual (expected_response , response )
289
289
290
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
290
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
291
291
def test_run_scan_action (self , mock_connector : Mock ):
292
292
expected_response = new_mock_empty_response ()
293
293
mock_connector .post_json .return_value = expected_response
@@ -301,7 +301,7 @@ def test_run_scan_action(self, mock_connector: Mock):
301
301
)
302
302
self .assertEqual (expected_response , response )
303
303
304
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
304
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
305
305
def test_start_scan (self , mock_connector : Mock ):
306
306
expected_response = new_mock_empty_response ()
307
307
mock_connector .post_json .return_value = expected_response
@@ -313,7 +313,7 @@ def test_start_scan(self, mock_connector: Mock):
313
313
)
314
314
self .assertEqual (expected_response , response )
315
315
316
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
316
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
317
317
def test_stop_scan (self , mock_connector : Mock ):
318
318
expected_response = new_mock_empty_response ()
319
319
mock_connector .post_json .return_value = expected_response
@@ -325,7 +325,7 @@ def test_stop_scan(self, mock_connector: Mock):
325
325
)
326
326
self .assertEqual (expected_response , response )
327
327
328
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
328
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
329
329
def test_get_vts (self , mock_connector : Mock ):
330
330
expected_response = new_mock_empty_response ()
331
331
mock_connector .get .return_value = expected_response
@@ -337,7 +337,7 @@ def test_get_vts(self, mock_connector: Mock):
337
337
)
338
338
self .assertEqual (expected_response , response )
339
339
340
- @patch ("gvm.http.core.connector.HttpApiConnector" , autospec = True )
340
+ @patch ("gvm.protocols. http.core.connector.HttpApiConnector" , autospec = True )
341
341
def test_get_vt (self , mock_connector : Mock ):
342
342
expected_response = new_mock_empty_response ()
343
343
mock_connector .get .return_value = expected_response
0 commit comments