evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Import websocket_api (part 3) (#63907)

Browse source
This commit is contained in:
Erik Montnemery 2022-01-11 17:26:37 +01:00 • committed by GitHub
parent 3b7448bb1c
commit 6eb0447566
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 32 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/components/websocket_api/test_init.py
View file

@ -4,7 +4,11 @@ from unittest.mock import Mock, patch
from aiohttp import WSMsgType
import voluptuous as vol
from homeassistant.components.websocket_api import const, messages
from homeassistant.components.websocket_api import (
async_register_command,
const,
messages,
)
async def test_invalid_message_format(websocket_client):
@ -49,7 +53,8 @@ async def test_unknown_command(websocket_client):
async def test_handler_failing(hass, websocket_client):
"""Test a command that raises."""
hass.components.websocket_api.async_register_command(
async_register_command(
hass,
"bla",
Mock(side_effect=TypeError),
messages.BASE_COMMAND_MESSAGE_SCHEMA.extend({"type": "bla"}),
@ -65,7 +70,8 @@ async def test_handler_failing(hass, websocket_client):
async def test_invalid_vol(hass, websocket_client):
"""Test a command that raises invalid vol error."""
hass.components.websocket_api.async_register_command(
async_register_command(
hass,
"bla",
Mock(side_effect=TypeError),
messages.BASE_COMMAND_MESSAGE_SCHEMA.extend(