91 lines
2.9 KiB
Python
91 lines
2.9 KiB
Python
"""The tests for the hassio component."""
|
|
import asyncio
|
|
|
|
import aiohttp
|
|
|
|
|
|
@asyncio.coroutine
|
|
def test_api_ping(hassio_handler, aioclient_mock):
|
|
"""Test setup with API ping."""
|
|
aioclient_mock.get(
|
|
"http://127.0.0.1/supervisor/ping", json={'result': 'ok'})
|
|
|
|
assert (yield from hassio_handler.is_connected())
|
|
assert aioclient_mock.call_count == 1
|
|
|
|
|
|
@asyncio.coroutine
|
|
def test_api_ping_error(hassio_handler, aioclient_mock):
|
|
"""Test setup with API ping error."""
|
|
aioclient_mock.get(
|
|
"http://127.0.0.1/supervisor/ping", json={'result': 'error'})
|
|
|
|
assert not (yield from hassio_handler.is_connected())
|
|
assert aioclient_mock.call_count == 1
|
|
|
|
|
|
@asyncio.coroutine
|
|
def test_api_ping_exeption(hassio_handler, aioclient_mock):
|
|
"""Test setup with API ping exception."""
|
|
aioclient_mock.get(
|
|
"http://127.0.0.1/supervisor/ping", exc=aiohttp.ClientError())
|
|
|
|
assert not (yield from hassio_handler.is_connected())
|
|
assert aioclient_mock.call_count == 1
|
|
|
|
|
|
@asyncio.coroutine
|
|
def test_api_homeassistant_info(hassio_handler, aioclient_mock):
|
|
"""Test setup with API homeassistant info."""
|
|
aioclient_mock.get(
|
|
"http://127.0.0.1/homeassistant/info", json={
|
|
'result': 'ok', 'data': {'last_version': '10.0'}})
|
|
|
|
data = yield from hassio_handler.get_homeassistant_info()
|
|
assert aioclient_mock.call_count == 1
|
|
assert data['last_version'] == "10.0"
|
|
|
|
|
|
@asyncio.coroutine
|
|
def test_api_homeassistant_info_error(hassio_handler, aioclient_mock):
|
|
"""Test setup with API homeassistant info error."""
|
|
aioclient_mock.get(
|
|
"http://127.0.0.1/homeassistant/info", json={
|
|
'result': 'error', 'message': None})
|
|
|
|
data = yield from hassio_handler.get_homeassistant_info()
|
|
assert aioclient_mock.call_count == 1
|
|
assert data is None
|
|
|
|
|
|
@asyncio.coroutine
|
|
def test_api_homeassistant_stop(hassio_handler, aioclient_mock):
|
|
"""Test setup with API HomeAssistant stop."""
|
|
aioclient_mock.post(
|
|
"http://127.0.0.1/homeassistant/stop", json={'result': 'ok'})
|
|
|
|
assert (yield from hassio_handler.stop_homeassistant())
|
|
assert aioclient_mock.call_count == 1
|
|
|
|
|
|
@asyncio.coroutine
|
|
def test_api_homeassistant_restart(hassio_handler, aioclient_mock):
|
|
"""Test setup with API HomeAssistant restart."""
|
|
aioclient_mock.post(
|
|
"http://127.0.0.1/homeassistant/restart", json={'result': 'ok'})
|
|
|
|
assert (yield from hassio_handler.restart_homeassistant())
|
|
assert aioclient_mock.call_count == 1
|
|
|
|
|
|
@asyncio.coroutine
|
|
def test_api_homeassistant_config(hassio_handler, aioclient_mock):
|
|
"""Test setup with API HomeAssistant restart."""
|
|
aioclient_mock.post(
|
|
"http://127.0.0.1/homeassistant/check", json={
|
|
'result': 'ok', 'data': {'test': 'bla'}})
|
|
|
|
data = yield from hassio_handler.check_homeassistant_config()
|
|
assert data['data']['test'] == 'bla'
|
|
assert aioclient_mock.call_count == 1
|