Example #1
0
    def setUp(self):  # pylint: disable=invalid-name
        """Set up things to be run when tests are started."""
        self.hass = get_test_home_assistant()
        self.events = []
        self.service1 = demo.DemoNotificationService(self.hass)
        self.service2 = demo.DemoNotificationService(self.hass)

        self.service1.send_message = MagicMock(autospec=True)
        self.service2.send_message = MagicMock(autospec=True)

        def mock_get_service(hass, config, discovery_info=None):
            if config["name"] == "demo1":
                return self.service1
            return self.service2

        with assert_setup_component(2, notify.DOMAIN), patch.object(
                demo, "get_service", mock_get_service):
            setup_component(
                self.hass,
                notify.DOMAIN,
                {
                    "notify": [
                        {
                            "name": "demo1",
                            "platform": "demo"
                        },
                        {
                            "name": "demo2",
                            "platform": "demo"
                        },
                    ]
                },
            )

        self.service = asyncio.run_coroutine_threadsafe(
            group.async_get_service(
                self.hass,
                {
                    "services": [
                        {
                            "service": "demo1"
                        },
                        {
                            "service": "demo2",
                            "data": {
                                "target": "unnamed device",
                                "data": {
                                    "test": "message"
                                },
                            },
                        },
                    ]
                },
            ),
            self.hass.loop,
        ).result()

        assert self.service is not None
        self.addCleanup(self.tear_down_cleanup)
Example #2
0
    def setUp(self):  # pylint: disable=invalid-name
        """Set up things to be run when tests are started."""
        self.hass = get_test_home_assistant()
        self.events = []
        self.service1 = demo.DemoNotificationService(self.hass)
        self.service2 = demo.DemoNotificationService(self.hass)

        self.service1.send_message = MagicMock(autospec=True)
        self.service2.send_message = MagicMock(autospec=True)

        def mock_get_service(hass, config, discovery_info=None):
            if config['name'] == 'demo1':
                return self.service1
            return self.service2

        with assert_setup_component(2), \
                patch.object(demo, 'get_service', mock_get_service):
            setup_component(
                self.hass, notify.DOMAIN, {
                    'notify': [{
                        'name': 'demo1',
                        'platform': 'demo'
                    }, {
                        'name': 'demo2',
                        'platform': 'demo'
                    }]
                })

        self.service = run_coroutine_threadsafe(
            group.async_get_service(
                self.hass, {
                    'services': [{
                        'service': 'demo1'
                    }, {
                        'service': 'demo2',
                        'data': {
                            'target': 'unnamed device',
                            'data': {
                                'test': 'message'
                            }
                        }
                    }]
                }), self.hass.loop).result()

        assert self.service is not None
Example #3
0
    def setUp(self):  # pylint: disable=invalid-name
        """Set up things to be run when tests are started."""
        self.hass = get_test_home_assistant()
        self.events = []
        self.service1 = demo.DemoNotificationService(self.hass)
        self.service2 = demo.DemoNotificationService(self.hass)

        self.service1.send_message = MagicMock(autospec=True)
        self.service2.send_message = MagicMock(autospec=True)

        def mock_get_service(hass, config, discovery_info=None):
            if config['name'] == 'demo1':
                return self.service1
            return self.service2

        with assert_setup_component(2), \
                patch.object(demo, 'get_service', mock_get_service):
            setup_component(self.hass, notify.DOMAIN, {
                'notify': [{
                    'name': 'demo1',
                    'platform': 'demo'
                }, {
                    'name': 'demo2',
                    'platform': 'demo'
                }]
            })

        self.service = run_coroutine_threadsafe(
            group.async_get_service(self.hass, {'services': [
                {'service': 'demo1'},
                {'service': 'demo2',
                 'data': {'target': 'unnamed device',
                          'data': {'test': 'message'}}}]}),
            self.hass.loop
        ).result()

        assert self.service is not None