Esempio n. 1
0
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": "******"}]
Esempio n. 2
0
 def __init_services(self):
     from middlewared.service import CoreService
     self.add_service(CoreService(self))
Esempio n. 3
0
def test_cli_args_descriptions(doc, names, descriptions):
    assert CoreService(None)._cli_args_descriptions(doc, names) == descriptions