diff --git a/rocky/tests/conftest.py b/rocky/tests/conftest.py index 8b0ce8ae23b..a40d907a5f9 100644 --- a/rocky/tests/conftest.py +++ b/rocky/tests/conftest.py @@ -1,4 +1,5 @@ import binascii +import io import json import logging from os import urandom @@ -527,37 +528,39 @@ def mock_mixins_katalogus(mocker): def mock_scheduler_client_task_list(mocker): mock_scheduler_client_session = mocker.patch("rocky.scheduler.client.session") response = Response() - response._content = json.dumps( - { - "count": 1, - "next": "http://scheduler:8000/tasks?scheduler_id=boefje-test&type=boefje&plugin_id=test_plugin&limit=10&offset=10", - "previous": None, - "results": [ - { - "id": "2e757dd3-66c7-46b8-9987-7cd18252cc6d", - "scheduler_id": "boefje-test", - "type": "boefje", - "p_item": { + response.raw = io.BytesIO( + json.dumps( + { + "count": 1, + "next": "http://scheduler:8000/tasks?scheduler_id=boefje-test&type=boefje&plugin_id=test_plugin&limit=10&offset=10", + "previous": None, + "results": [ + { "id": "2e757dd3-66c7-46b8-9987-7cd18252cc6d", "scheduler_id": "boefje-test", - "hash": "416aa907e0b2a16c1b324f7d3261c5a4", - "priority": 631, - "data": { - "id": "2e757dd366c746b899877cd18252cc6d", - "boefje": {"id": "test-plugin", "version": None}, - "input_ooi": "Hostname|internet|example.com", - "organization": "test", - "dispatches": [], + "type": "boefje", + "p_item": { + "id": "2e757dd3-66c7-46b8-9987-7cd18252cc6d", + "scheduler_id": "boefje-test", + "hash": "416aa907e0b2a16c1b324f7d3261c5a4", + "priority": 631, + "data": { + "id": "2e757dd366c746b899877cd18252cc6d", + "boefje": {"id": "test-plugin", "version": None}, + "input_ooi": "Hostname|internet|example.com", + "organization": "test", + "dispatches": [], + }, + "created_at": "2023-05-09T09:37:20.899668+00:00", + "modified_at": "2023-05-09T09:37:20.899675+00:00", }, - "created_at": "2023-05-09T09:37:20.899668+00:00", - "modified_at": "2023-05-09T09:37:20.899675+00:00", - }, - "status": "completed", - "created_at": "2023-05-09T09:37:20.909069+00:00", - "modified_at": "2023-05-09T09:37:20.909071+00:00", - } - ], - } + "status": "completed", + "created_at": "2023-05-09T09:37:20.909069+00:00", + "modified_at": "2023-05-09T09:37:20.909071+00:00", + } + ], + } + ).encode() ) mock_scheduler_client_session.get.return_value = response