68 lines
2.0 KiB
Python
68 lines
2.0 KiB
Python
"""Test the OctoPrint services."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.components.octoprint.const import (
|
|
CONF_BAUDRATE,
|
|
DOMAIN,
|
|
SERVICE_CONNECT,
|
|
)
|
|
from homeassistant.const import ATTR_DEVICE_ID, CONF_PORT, CONF_PROFILE_NAME
|
|
from homeassistant.helpers.device_registry import (
|
|
async_entries_for_config_entry,
|
|
async_get as async_get_dev_reg,
|
|
)
|
|
|
|
from . import init_integration
|
|
|
|
|
|
async def test_connect_default(hass) -> None:
|
|
"""Test the connect to printer service."""
|
|
await init_integration(hass, "sensor")
|
|
|
|
dev_reg = async_get_dev_reg(hass)
|
|
device = async_entries_for_config_entry(dev_reg, "uuid")[0]
|
|
|
|
# Test pausing the printer when it is printing
|
|
with patch("pyoctoprintapi.OctoprintClient.connect") as connect_command:
|
|
await hass.services.async_call(
|
|
DOMAIN,
|
|
SERVICE_CONNECT,
|
|
{
|
|
ATTR_DEVICE_ID: device.id,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
assert len(connect_command.mock_calls) == 1
|
|
connect_command.assert_called_with(
|
|
port=None, printer_profile=None, baud_rate=None
|
|
)
|
|
|
|
|
|
async def test_connect_all_arguments(hass) -> None:
|
|
"""Test the connect to printer service."""
|
|
await init_integration(hass, "sensor")
|
|
|
|
dev_reg = async_get_dev_reg(hass)
|
|
device = async_entries_for_config_entry(dev_reg, "uuid")[0]
|
|
|
|
# Test pausing the printer when it is printing
|
|
with patch("pyoctoprintapi.OctoprintClient.connect") as connect_command:
|
|
await hass.services.async_call(
|
|
DOMAIN,
|
|
SERVICE_CONNECT,
|
|
{
|
|
ATTR_DEVICE_ID: device.id,
|
|
CONF_PROFILE_NAME: "Test Profile",
|
|
CONF_PORT: "VIRTUAL",
|
|
CONF_BAUDRATE: 9600,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
assert len(connect_command.mock_calls) == 1
|
|
connect_command.assert_called_with(
|
|
port="VIRTUAL", printer_profile="Test Profile", baud_rate=9600
|
|
)
|