44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
"""Tests for the auth models."""
|
|
from homeassistant.auth import models, permissions
|
|
|
|
|
|
def test_owner_fetching_owner_permissions():
|
|
"""Test we fetch the owner permissions for an owner user."""
|
|
group = models.Group(name="Test Group", policy={})
|
|
owner = models.User(
|
|
name="Test User",
|
|
perm_lookup=None,
|
|
groups=[group],
|
|
is_owner=True
|
|
)
|
|
assert owner.permissions is permissions.OwnerPermissions
|
|
|
|
|
|
def test_permissions_merged():
|
|
"""Test we merge the groups permissions."""
|
|
group = models.Group(name="Test Group", policy={
|
|
'entities': {
|
|
'domains': {
|
|
'switch': True
|
|
}
|
|
}
|
|
})
|
|
group2 = models.Group(name="Test Group", policy={
|
|
'entities': {
|
|
'entity_ids': {
|
|
'light.kitchen': True
|
|
}
|
|
}
|
|
})
|
|
user = models.User(
|
|
name="Test User",
|
|
perm_lookup=None,
|
|
groups=[group, group2]
|
|
)
|
|
# Make sure we cache instance
|
|
assert user.permissions is user.permissions
|
|
|
|
assert user.permissions.check_entity('switch.bla', 'read') is True
|
|
assert user.permissions.check_entity('light.kitchen', 'read') is True
|
|
assert user.permissions.check_entity('light.not_kitchen', 'read') is False
|