def setUp(self): """Set up things to be run when tests are started.""" self.hass = get_test_home_assistant() controller_mock = mock.MagicMock() dev_dict = {"address": "a1", "name": "fake_switch"} self.switch = mochad.MochadSwitch(self.hass, controller_mock, dev_dict)
def switch_mock(hass): """Mock switch.""" controller_mock = mock.MagicMock() dev_dict = {"address": "a1", "name": "fake_switch"} return mochad.MochadSwitch(hass, controller_mock, dev_dict)