From e5eef7c6ddc69b2c3b73186d041fd0e0c7e2babb Mon Sep 17 00:00:00 2001 From: Joost Lekkerkerker Date: Mon, 17 Jun 2024 19:17:52 +0200 Subject: [PATCH] Fix Dremel 3D printer tests (#119853) --- .../components/dremel_3d_printer/conftest.py | 32 +++++++++---------- 1 file changed, 16 insertions(+), 16 deletions(-) diff --git a/tests/components/dremel_3d_printer/conftest.py b/tests/components/dremel_3d_printer/conftest.py index 0284d8baebf..6490b844dc0 100644 --- a/tests/components/dremel_3d_printer/conftest.py +++ b/tests/components/dremel_3d_printer/conftest.py @@ -32,23 +32,23 @@ def config_entry(hass: HomeAssistant) -> MockConfigEntry: @pytest.fixture def connection() -> None: """Mock Dremel 3D Printer connection.""" - mock = requests_mock.Mocker() - mock.post( - f"http://{HOST}:80/command", - response_list=[ - {"text": load_fixture("dremel_3d_printer/command_1.json")}, - {"text": load_fixture("dremel_3d_printer/command_2.json")}, - {"text": load_fixture("dremel_3d_printer/command_1.json")}, - {"text": load_fixture("dremel_3d_printer/command_2.json")}, - ], - ) + with requests_mock.Mocker() as mock: + mock.post( + f"http://{HOST}:80/command", + response_list=[ + {"text": load_fixture("dremel_3d_printer/command_1.json")}, + {"text": load_fixture("dremel_3d_printer/command_2.json")}, + {"text": load_fixture("dremel_3d_printer/command_1.json")}, + {"text": load_fixture("dremel_3d_printer/command_2.json")}, + ], + ) - mock.post( - f"https://{HOST}:11134/getHomeMessage", - text=load_fixture("dremel_3d_printer/get_home_message.json"), - status_code=HTTPStatus.OK, - ) - mock.start() + mock.post( + f"https://{HOST}:11134/getHomeMessage", + text=load_fixture("dremel_3d_printer/get_home_message.json"), + status_code=HTTPStatus.OK, + ) + yield def patch_async_setup_entry():