33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from server.config import ServerSettings
|
|
from server.grid import GridManager
|
|
from server.planner import GridPlanner
|
|
from server.models import ActionType, GridInitRequest
|
|
|
|
|
|
def test_planner_preferred_label(default_grid_request):
|
|
settings = ServerSettings()
|
|
manager = GridManager(settings)
|
|
request = GridInitRequest(**default_grid_request)
|
|
grid = manager.create_grid(request)
|
|
descriptor = grid.describe()
|
|
descriptor.cells[0].label = "target"
|
|
|
|
planner = GridPlanner()
|
|
payload = planner.build_payload(descriptor, preferred_label="target", action=ActionType.CLICK)
|
|
|
|
assert payload.target_cell == descriptor.cells[0].cell_id
|
|
|
|
|
|
def test_planner_falls_back_to_center(default_grid_request):
|
|
settings = ServerSettings()
|
|
manager = GridManager(settings)
|
|
request = GridInitRequest(**default_grid_request)
|
|
grid = manager.create_grid(request)
|
|
descriptor = grid.describe()
|
|
|
|
planner = GridPlanner()
|
|
payload = planner.build_payload(descriptor, action=ActionType.CLICK)
|
|
|
|
assert payload.target_cell is not None
|
|
assert payload.grid_id == descriptor.grid_id
|