93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
"""The tests for the hassio component."""
|
|
from unittest.mock import Mock, patch
|
|
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
|
|
from tests.common import mock_coro
|
|
|
|
|
|
async def test_login_success(hass, hassio_client):
|
|
"""Test no auth needed for ."""
|
|
with patch(
|
|
"homeassistant.auth.providers.homeassistant."
|
|
"HassAuthProvider.async_validate_login",
|
|
Mock(return_value=mock_coro()),
|
|
) as mock_login:
|
|
resp = await hassio_client.post(
|
|
"/api/hassio_auth",
|
|
json={"username": "test", "password": "123456", "addon": "samba"},
|
|
)
|
|
|
|
# Check we got right response
|
|
assert resp.status == 200
|
|
mock_login.assert_called_with("test", "123456")
|
|
|
|
|
|
async def test_login_error(hass, hassio_client):
|
|
"""Test no auth needed for error."""
|
|
with patch(
|
|
"homeassistant.auth.providers.homeassistant."
|
|
"HassAuthProvider.async_validate_login",
|
|
Mock(side_effect=HomeAssistantError()),
|
|
) as mock_login:
|
|
resp = await hassio_client.post(
|
|
"/api/hassio_auth",
|
|
json={"username": "test", "password": "123456", "addon": "samba"},
|
|
)
|
|
|
|
# Check we got right response
|
|
assert resp.status == 403
|
|
mock_login.assert_called_with("test", "123456")
|
|
|
|
|
|
async def test_login_no_data(hass, hassio_client):
|
|
"""Test auth with no data -> error."""
|
|
with patch(
|
|
"homeassistant.auth.providers.homeassistant."
|
|
"HassAuthProvider.async_validate_login",
|
|
Mock(side_effect=HomeAssistantError()),
|
|
) as mock_login:
|
|
resp = await hassio_client.post("/api/hassio_auth")
|
|
|
|
# Check we got right response
|
|
assert resp.status == 400
|
|
assert not mock_login.called
|
|
|
|
|
|
async def test_login_no_username(hass, hassio_client):
|
|
"""Test auth with no username in data -> error."""
|
|
with patch(
|
|
"homeassistant.auth.providers.homeassistant."
|
|
"HassAuthProvider.async_validate_login",
|
|
Mock(side_effect=HomeAssistantError()),
|
|
) as mock_login:
|
|
resp = await hassio_client.post(
|
|
"/api/hassio_auth", json={"password": "123456", "addon": "samba"}
|
|
)
|
|
|
|
# Check we got right response
|
|
assert resp.status == 400
|
|
assert not mock_login.called
|
|
|
|
|
|
async def test_login_success_extra(hass, hassio_client):
|
|
"""Test auth with extra data."""
|
|
with patch(
|
|
"homeassistant.auth.providers.homeassistant."
|
|
"HassAuthProvider.async_validate_login",
|
|
Mock(return_value=mock_coro()),
|
|
) as mock_login:
|
|
resp = await hassio_client.post(
|
|
"/api/hassio_auth",
|
|
json={
|
|
"username": "test",
|
|
"password": "123456",
|
|
"addon": "samba",
|
|
"path": "/share",
|
|
},
|
|
)
|
|
|
|
# Check we got right response
|
|
assert resp.status == 200
|
|
mock_login.assert_called_with("test", "123456")
|