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