hass-core/tests/auth/test_models.py

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