Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +00:00
Code Activity

Black

Browse Source
This commit is contained in:
Paulus Schoutsen
2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
Download Patch File Download Diff File Expand all files Collapse all files

179
tests/components/frontend/test_storage.py
View File

@@ -8,74 +8,61 @@ from homeassistant.components.frontend import storage
@pytest.fixture(autouse=True)
def setup_frontend(hass):
"""Fixture to setup the frontend."""
hass.loop.run_until_complete(async_setup_component(hass, 'frontend', {}))
hass.loop.run_until_complete(async_setup_component(hass, "frontend", {}))
async def test_get_user_data_empty(hass, hass_ws_client, hass_storage):
"""Test get_user_data command."""
client = await hass_ws_client(hass)
await client.send_json({
'id': 5,
'type': 'frontend/get_user_data',
'key': 'non-existing-key',
})
await client.send_json(
{"id": 5, "type": "frontend/get_user_data", "key": "non-existing-key"}
)
res = await client.receive_json()
assert res['success'], res
assert res['result']['value'] is None
assert res["success"], res
assert res["result"]["value"] is None
async def test_get_user_data(hass, hass_ws_client, hass_admin_user,
hass_storage):
async def test_get_user_data(hass, hass_ws_client, hass_admin_user, hass_storage):
"""Test get_user_data command."""
storage_key = storage.STORAGE_KEY_USER_DATA.format(hass_admin_user.id)
hass_storage[storage_key] = {
'key': storage_key,
'version': 1,
'data': {
'test-key': 'test-value',
'test-complex': [{'foo': 'bar'}]
}
"key": storage_key,
"version": 1,
"data": {"test-key": "test-value", "test-complex": [{"foo": "bar"}]},
}
client = await hass_ws_client(hass)
# Get a simple string key
await client.send_json({
'id': 6,
'type': 'frontend/get_user_data',
'key': 'test-key',
})
await client.send_json(
{"id": 6, "type": "frontend/get_user_data", "key": "test-key"}
)
res = await client.receive_json()
assert res['success'], res
assert res['result']['value'] == 'test-value'
assert res["success"], res
assert res["result"]["value"] == "test-value"
# Get a more complex key
await client.send_json({
'id': 7,
'type': 'frontend/get_user_data',
'key': 'test-complex',
})
await client.send_json(
{"id": 7, "type": "frontend/get_user_data", "key": "test-complex"}
)
res = await client.receive_json()
assert res['success'], res
assert res['result']['value'][0]['foo'] == 'bar'
assert res["success"], res
assert res["result"]["value"][0]["foo"] == "bar"
# Get all data (no key)
await client.send_json({
'id': 8,
'type': 'frontend/get_user_data',
})
await client.send_json({"id": 8, "type": "frontend/get_user_data"})
res = await client.receive_json()
assert res['success'], res
assert res['result']['value']['test-key'] == 'test-value'
assert res['result']['value']['test-complex'][0]['foo'] == 'bar'
assert res["success"], res
assert res["result"]["value"]["test-key"] == "test-value"
assert res["result"]["value"]["test-complex"][0]["foo"] == "bar"
async def test_set_user_data_empty(hass, hass_ws_client, hass_storage):
@@ -84,103 +71,95 @@ async def test_set_user_data_empty(hass, hass_ws_client, hass_storage):
# test creating
await client.send_json({
'id': 6,
'type': 'frontend/get_user_data',
'key': 'test-key',
})
await client.send_json(
{"id": 6, "type": "frontend/get_user_data", "key": "test-key"}
)
res = await client.receive_json()
assert res['success'], res
assert res['result']['value'] is None
assert res["success"], res
assert res["result"]["value"] is None
await client.send_json({
'id': 7,
'type': 'frontend/set_user_data',
'key': 'test-key',
'value': 'test-value'
})
await client.send_json(
{
"id": 7,
"type": "frontend/set_user_data",
"key": "test-key",
"value": "test-value",
}
)
res = await client.receive_json()
assert res['success'], res
assert res["success"], res
await client.send_json({
'id': 8,
'type': 'frontend/get_user_data',
'key': 'test-key',
})
await client.send_json(
{"id": 8, "type": "frontend/get_user_data", "key": "test-key"}
)
res = await client.receive_json()
assert res['success'], res
assert res['result']['value'] == 'test-value'
assert res["success"], res
assert res["result"]["value"] == "test-value"
async def test_set_user_data(hass, hass_ws_client, hass_storage,
hass_admin_user):
async def test_set_user_data(hass, hass_ws_client, hass_storage, hass_admin_user):
"""Test set_user_data command with initial data."""
storage_key = storage.STORAGE_KEY_USER_DATA.format(hass_admin_user.id)
hass_storage[storage_key] = {
'version': 1,
'data': {
'test-key': 'test-value',
'test-complex': 'string',
}
"version": 1,
"data": {"test-key": "test-value", "test-complex": "string"},
}
client = await hass_ws_client(hass)
# test creating
await client.send_json({
'id': 5,
'type': 'frontend/set_user_data',
'key': 'test-non-existent-key',
'value': 'test-value-new'
})
await client.send_json(
{
"id": 5,
"type": "frontend/set_user_data",
"key": "test-non-existent-key",
"value": "test-value-new",
}
)
res = await client.receive_json()
assert res['success'], res
assert res["success"], res
await client.send_json({
'id': 6,
'type': 'frontend/get_user_data',
'key': 'test-non-existent-key',
})
await client.send_json(
{"id": 6, "type": "frontend/get_user_data", "key": "test-non-existent-key"}
)
res = await client.receive_json()
assert res['success'], res
assert res['result']['value'] == 'test-value-new'
assert res["success"], res
assert res["result"]["value"] == "test-value-new"
# test updating with complex data
await client.send_json({
'id': 7,
'type': 'frontend/set_user_data',
'key': 'test-complex',
'value': [{'foo': 'bar'}]
})
await client.send_json(
{
"id": 7,
"type": "frontend/set_user_data",
"key": "test-complex",
"value": [{"foo": "bar"}],
}
)
res = await client.receive_json()
assert res['success'], res
assert res["success"], res
await client.send_json({
'id': 8,
'type': 'frontend/get_user_data',
'key': 'test-complex',
})
await client.send_json(
{"id": 8, "type": "frontend/get_user_data", "key": "test-complex"}
)
res = await client.receive_json()
assert res['success'], res
assert res['result']['value'][0]['foo'] == 'bar'
assert res["success"], res
assert res["result"]["value"][0]["foo"] == "bar"
# ensure other existing key was not modified
await client.send_json({
'id': 9,
'type': 'frontend/get_user_data',
'key': 'test-key',
})
await client.send_json(
{"id": 9, "type": "frontend/get_user_data", "key": "test-key"}
)
res = await client.receive_json()
assert res['success'], res
assert res['result']['value'] == 'test-value'
assert res["success"], res
assert res["result"]["value"] == "test-value"