f7f0a4e811
* Add read only and admin policies * Migrate to 2 system groups * Add system groups * Add system groups admin & read only * Dont' mutate parameters * Fix types
35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
"""Tests for the auth permission system."""
|
|
from homeassistant.core import State
|
|
from homeassistant.auth import permissions
|
|
|
|
|
|
def test_policy_perm_filter_states():
|
|
"""Test filtering entitites."""
|
|
states = [
|
|
State('light.kitchen', 'on'),
|
|
State('light.living_room', 'off'),
|
|
State('light.balcony', 'on'),
|
|
]
|
|
perm = permissions.PolicyPermissions({
|
|
'entities': {
|
|
'entity_ids': {
|
|
'light.kitchen': True,
|
|
'light.balcony': True,
|
|
}
|
|
}
|
|
})
|
|
filtered = perm.filter_states(states)
|
|
assert len(filtered) == 2
|
|
assert filtered == [states[0], states[2]]
|
|
|
|
|
|
def test_owner_permissions():
|
|
"""Test owner permissions access all."""
|
|
assert permissions.OwnerPermissions.check_entity('light.kitchen', 'write')
|
|
states = [
|
|
State('light.kitchen', 'on'),
|
|
State('light.living_room', 'off'),
|
|
State('light.balcony', 'on'),
|
|
]
|
|
assert permissions.OwnerPermissions.filter_states(states) == states
|