Fix bambu client connection.

This commit is contained in:
Anton Skrypnyk
2024-07-24 17:15:47 +03:00
parent ed33fd8fb1
commit 38a6f58306
4 changed files with 65 additions and 61 deletions

View File

@ -124,7 +124,8 @@ def ftps_session_mock(files_info_ftp):
@fixture(scope="function")
def print_job_mock():
print_job = MagicMock()
print_job.get.side_effect = DictGetter({"subtask_name": "", "print_percentage": 0})
print_job.subtask_name = ""
print_job.print_percentage = 0
return print_job
@ -162,7 +163,7 @@ def printer(
async def _mock_connection(self):
pass
BambuVirtualPrinter._create_connection_async = _mock_connection
BambuVirtualPrinter._create_client_connection_async = _mock_connection
serial_obj = BambuVirtualPrinter(
settings,
profile_manager,
@ -224,9 +225,7 @@ def test_print_started_with_selected_file(printer: BambuVirtualPrinter, print_jo
assert printer.file_system.selected_file is not None
assert printer.file_system.selected_file.file_name == "print.3mf"
print_job_mock.get.side_effect = DictGetter(
{"subtask_name": "print.3mf", "print_percentage": 0}
)
print_job_mock.subtask_name = "print.3mf"
printer.write(b"M24\n")
printer.flush()
@ -237,9 +236,7 @@ def test_print_started_with_selected_file(printer: BambuVirtualPrinter, print_jo
def test_pause_print(printer: BambuVirtualPrinter, bambu_client_mock, print_job_mock):
print_job_mock.get.side_effect = DictGetter(
{"subtask_name": "print.3mf", "print_percentage": 0}
)
print_job_mock.subtask_name = "print.3mf"
printer.write(b"M20\n")
printer.write(b"M23 print.3mf\n")