"""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