Python EntityComponent.extract_from_service Examples

Example #1

File: test_entity_component.py Project: Rosiak/home-assistant

    def test_extract_from_service_filter_out_non_existing_entities(self):
        """Test the extraction of non existing entities from service."""
        component = EntityComponent(_LOGGER, DOMAIN, self.hass)
        component.add_entities([EntityTest(name="test_1"), EntityTest(name="test_2")])

        call = ha.ServiceCall("test", "service", {"entity_id": ["test_domain.test_2", "test_domain.non_exist"]})

        assert ["test_domain.test_2"] == [ent.entity_id for ent in component.extract_from_service(call)]

Example #2

File: test_entity_component.py Project: Rosiak/home-assistant

    def test_extract_from_service_returns_all_if_no_entity_id(self):
        """Test the extraction of everything from service."""
        component = EntityComponent(_LOGGER, DOMAIN, self.hass)
        component.add_entities([EntityTest(name="test_1"), EntityTest(name="test_2")])

        call = ha.ServiceCall("test", "service")

        assert ["test_domain.test_1", "test_domain.test_2"] == sorted(
            ent.entity_id for ent in component.extract_from_service(call)
        )

Example #3

File: test_entity_component.py Project: cyrus19901/home-assistant-1

    def test_extract_from_service_no_group_expand(self):
        """Test not expanding a group."""
        component = EntityComponent(_LOGGER, DOMAIN, self.hass)
        test_group = group.Group.create_group(
            self.hass, 'test_group', ['light.Ceiling', 'light.Kitchen'])
        component.add_entities([test_group])

        call = ha.ServiceCall('test', 'service',
                              {'entity_id': ['group.test_group']})

        extracted = component.extract_from_service(call, expand_group=False)
        self.assertEqual([test_group], extracted)

Example #4

File: test_entity_component.py Project: x86Labs/home-assistant

    def test_extract_from_service_returns_all_if_no_entity_id(self):
        component = EntityComponent(_LOGGER, DOMAIN, self.hass)
        component.add_entities([
            EntityTest(name='test_1'),
            EntityTest(name='test_2'),
        ])

        call = ha.ServiceCall('test', 'service')

        assert ['test_domain.test_1', 'test_domain.test_2'] == \
            sorted(ent.entity_id for ent in
                   component.extract_from_service(call))

Example #5

File: test_entity_component.py Project: FanaHOVA/home-assistant

    def test_extract_from_service_returns_all_if_no_entity_id(self):
        component = EntityComponent(_LOGGER, DOMAIN, self.hass)
        component.add_entities([
            EntityTest(name='test_1'),
            EntityTest(name='test_2'),
        ])

        call = ha.ServiceCall('test', 'service')

        assert ['test_domain.test_1', 'test_domain.test_2'] == \
            sorted(ent.entity_id for ent in
                   component.extract_from_service(call))

Example #6

File: test_entity_component.py Project: Khabi/home-assistant

    def test_extract_from_service_no_group_expand(self):
        """Test not expanding a group."""
        component = EntityComponent(_LOGGER, DOMAIN, self.hass)
        test_group = group.Group.create_group(
            self.hass, 'test_group', ['light.Ceiling', 'light.Kitchen'])
        component.add_entities([test_group])

        call = ha.ServiceCall('test', 'service', {
            'entity_id': ['group.test_group']
        })

        extracted = component.extract_from_service(call, expand_group=False)
        self.assertEqual([test_group], extracted)

Example #7

File: test_entity_component.py Project: x86Labs/home-assistant

    def test_extract_from_service_filter_out_non_existing_entities(self):
        component = EntityComponent(_LOGGER, DOMAIN, self.hass)
        component.add_entities([
            EntityTest(name='test_1'),
            EntityTest(name='test_2'),
        ])

        call = ha.ServiceCall(
            'test', 'service',
            {'entity_id': ['test_domain.test_2', 'test_domain.non_exist']})

        assert ['test_domain.test_2'] == \
               [ent.entity_id for ent in component.extract_from_service(call)]

Example #8

File: test_entity_component.py Project: FanaHOVA/home-assistant

    def test_extract_from_service_filter_out_non_existing_entities(self):
        component = EntityComponent(_LOGGER, DOMAIN, self.hass)
        component.add_entities([
            EntityTest(name='test_1'),
            EntityTest(name='test_2'),
        ])

        call = ha.ServiceCall('test', 'service', {
            'entity_id': ['test_domain.test_2', 'test_domain.non_exist']
        })

        assert ['test_domain.test_2'] == \
               [ent.entity_id for ent in component.extract_from_service(call)]