68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
from octoprint_bambu_printer.bambu_print_plugin import BambuPrintPlugin
|
|
from octoprint_bambu_printer.printer.bambu_virtual_printer import BambuVirtualPrinter
|
|
from octoprint_bambu_printer.printer.states.idle_state import IdleState
|
|
from octoprint_bambu_printer.printer.states.paused_state import PausedState
|
|
from octoprint_bambu_printer.printer.states.print_finished_state import (
|
|
PrintFinishedState,
|
|
)
|
|
from octoprint_bambu_printer.printer.states.printing_state import PrintingState
|
|
from pytest import fixture
|
|
|
|
|
|
@fixture
|
|
def printer():
|
|
printer = BambuPrintPlugin().virtual_printer_factory(None, 5000, 115200, 5)
|
|
assert printer is not None
|
|
printer._bambu_client
|
|
return printer
|
|
|
|
|
|
def test_initial_state(printer: BambuVirtualPrinter):
|
|
assert isinstance(printer.current_state, IdleState)
|
|
|
|
|
|
def test_list_sd_card(printer: BambuVirtualPrinter):
|
|
printer.write(b"M20\n") # GCode for listing SD card
|
|
result = printer.readline()
|
|
assert result == "" # Replace with the actual expected result
|
|
|
|
|
|
def test_start_print(printer: BambuVirtualPrinter):
|
|
gcode = b"G28\nG1 X10 Y10\n"
|
|
printer.write(gcode)
|
|
result = printer.readline()
|
|
assert isinstance(printer.current_state, PrintingState)
|
|
|
|
|
|
def test_pause_print(printer: BambuVirtualPrinter):
|
|
gcode = b"G28\nG1 X10 Y10\n"
|
|
printer.write(gcode)
|
|
printer.write(b"M25\n") # GCode for pausing the print
|
|
result = printer.readline()
|
|
assert isinstance(printer.current_state, PausedState)
|
|
|
|
|
|
def test_get_printing_info(printer: BambuVirtualPrinter):
|
|
gcode = b"G28\nG1 X10 Y10\n"
|
|
printer.write(gcode)
|
|
printer.write(b"M27\n") # GCode for getting printing info
|
|
result = printer.readline()
|
|
assert result == ""
|
|
|
|
|
|
def test_abort_print(printer: BambuVirtualPrinter):
|
|
gcode = b"G28\nG1 X10 Y10\n"
|
|
printer.write(gcode)
|
|
printer.write(b"M26\n") # GCode for aborting the print
|
|
result = printer.readline()
|
|
assert isinstance(printer.current_state, IdleState)
|
|
|
|
|
|
def test_print_finished(printer: BambuVirtualPrinter):
|
|
gcode = b"G28\nG1 X10 Y10\n"
|
|
printer.write(gcode)
|
|
result = printer.readline()
|
|
assert isinstance(printer.current_state, PrintFinishedState)
|