async def test__job_service(caplog, method): caplog.set_level(logging.INFO) with patch("middlewared.main.multiprocessing"): middleware = Middleware() middleware.loop = asyncio.get_event_loop() middleware.add_service(CoreService(middleware)) middleware.add_service(DatastoreService(middleware)) middleware.add_service(JobService(middleware)) middleware._resolve_methods() fut = asyncio.Future() application = Application(middleware, asyncio.get_event_loop(), Mock(), Mock(send_str=AsyncMock(side_effect=fut.set_result))) application.authenticated = True application.authenticated_credentials = Mock(authorize=Mock(return_value=True)) application.handshake = True await application.on_message({"id": "1", "msg": "method", "method": f"job.{method}", "params": [{"password": "******"}]}) await fut fut = asyncio.Future() application.response.send_str = AsyncMock(side_effect=fut.set_result) await application.on_message({"id": "1", "msg": "method", "method": "core.get_jobs", "params": []}) result = json.loads(await fut) assert result["result"][0]["arguments"] == [{"password": "******"}]
def __init_services(self): from middlewared.service import CoreService self.add_service(CoreService(self))
def test_cli_args_descriptions(doc, names, descriptions): assert CoreService(None)._cli_args_descriptions(doc, names) == descriptions