def __init__( self, server: Server, dest_path: Optional[StrOrPath], download_size: int, progress_callback: Optional[Callable[[int, int, int], None]]) -> None: self.server = server self.event_loop = server.get_event_loop() self.need_content_length: bool = True self.need_content_disposition: bool = False self.request_ok: bool = False if dest_path is None: # If no destination is provided initialize to a procedurally # generated temp file. We will attempt to extract the filename # from the Content-Disposition Header tmp_dir = tempfile.gettempdir() loop_time = int(self.event_loop.get_loop_time()) tmp_fname = f"moonraker.download-{loop_time}.mrd" self.dest_file = pathlib.Path(tmp_dir).joinpath(tmp_fname) self.need_content_disposition = True elif isinstance(dest_path, str): self.dest_file = pathlib.Path(dest_path) else: self.dest_file = dest_path self.filename = self.dest_file.name self.file_hdl: Optional[BufferedWriter] = None self.total_recd: int = 0 self.download_size: int = download_size self.pct_done: int = 0 self.chunk_buffer: List[bytes] = [] self.progress_callback = progress_callback self.busy_evt: asyncio.Event = asyncio.Event() self.busy_evt.set()
async def test_send_async_event(full_server: Server): evtloop = full_server.get_event_loop() fut = evtloop.create_future() async def test_func(arg): fut.set_result(arg) full_server.register_event_handler("test:my_test", test_func) full_server.send_event("test:my_test", "test") result = await fut assert result == "test"
async def test_klippy_reconnect(ready_server: Server, klippy: KlippyProcess): evtloop = ready_server.get_event_loop() futs = [evtloop.create_future() for _ in range(2)] events = { "server:klippy_disconnect": lambda: futs[0].set_result("disconnect"), "server:klippy_ready": lambda: futs[1].set_result("ready") } for name, func in events.items(): ready_server.register_event_handler(name, func) klippy.restart() ret = await asyncio.wait_for(asyncio.gather(*futs), 6.) assert ret == ["disconnect", "ready"]
async def test_klippy_shutdown(ready_server: Server, klippy: KlippyProcess): evtloop = ready_server.get_event_loop() fut = evtloop.create_future() def on_shutdown(): if not fut.done(): fut.set_result("shutdown") ready_server.register_event_handler("server:klippy_shutdown", on_shutdown) klippy.send_gcode("M112") await asyncio.wait_for(fut, 2.) assert fut.result() == "shutdown"
async def test_gcode_response(ready_server: Server, klippy: KlippyProcess): evtloop = ready_server.get_event_loop() fut = evtloop.create_future() def on_gc_resp(resp: str): if not fut.done(): fut.set_result(resp) ready_server.register_event_handler("server:gcode_response", on_gc_resp) klippy.send_gcode("M118 Moonraker Test") await asyncio.wait_for(fut, 1.) assert "Moonraker Test" in fut.result()
async def test_klippy_startup(full_server: Server): evtloop = full_server.get_event_loop() futs = [evtloop.create_future() for _ in range(3)] events = { "server:klippy_identified": lambda: futs[0].set_result("id"), "server:klippy_started": lambda x: futs[1].set_result("started"), "server:klippy_ready": lambda: futs[2].set_result("ready") } for name, func in events.items(): full_server.register_event_handler(name, func) await full_server.start_server() ret = await asyncio.wait_for(asyncio.gather(*futs), 4.) assert (ret == ["id", "started", "ready"] and full_server.klippy_connection.is_connected())
async def test_status_update(ready_server: Server, klippy: KlippyProcess): evtloop = ready_server.get_event_loop() fut = evtloop.create_future() def on_status_update(data): if not fut.done(): fut.set_result(data) ready_server.register_event_handler("server:status_update", on_status_update) kapis = ready_server.klippy_connection.klippy_apis await kapis.subscribe_objects({"toolhead": None}) klippy.send_gcode("G28") await asyncio.wait_for(fut, 2.) assert isinstance(fut.result(), dict)
async def test_call_remote_method(base_server: Server, klippy: KlippyProcess): fut = base_server.get_event_loop().create_future() def method_test(result): fut.set_result(result) base_server.register_remote_method("moonraker_test", method_test) base_server.load_components() await base_server.server_init() ret = base_server.klippy_connection.wait_connected() await asyncio.wait_for(ret, 4.) klippy.send_gcode("TEST_REMOTE_METHOD") await fut await base_server._stop_server("terminate") assert fut.result() == "test"
def test_pending_tasks(self, base_server: Server): loop = base_server.get_event_loop().aioloop assert len(asyncio.all_tasks(loop)) == 0