def test_active_child_state(self): """Test active child state property.""" config = self.config_children_only universal.validate_config(config) ump = universal.UniversalMediaPlayer(self.hass, **config) ump.entity_id = media_player.ENTITY_ID_FORMAT.format(config['name']) ump.update() self.assertEqual(None, ump._child_state) self.mock_mp_1._state = STATE_PLAYING self.mock_mp_1.update_ha_state() ump.update() self.assertEqual(self.mock_mp_1.entity_id, ump._child_state.entity_id) self.mock_mp_2._state = STATE_PLAYING self.mock_mp_2.update_ha_state() ump.update() self.assertEqual(self.mock_mp_1.entity_id, ump._child_state.entity_id) self.mock_mp_1._state = STATE_OFF self.mock_mp_1.update_ha_state() ump.update() self.assertEqual(self.mock_mp_2.entity_id, ump._child_state.entity_id)
def test_service_call_to_child(self): """Test service calls that should be routed to a child.""" config = self.config_children_only universal.validate_config(config) ump = universal.UniversalMediaPlayer(self.hass, **config) ump.entity_id = media_player.ENTITY_ID_FORMAT.format(config['name']) ump.update() self.mock_mp_2._state = STATE_PLAYING self.mock_mp_2.update_ha_state() ump.update() ump.turn_off() self.assertEqual(1, len(self.mock_mp_2.service_calls['turn_off'])) ump.turn_on() self.assertEqual(1, len(self.mock_mp_2.service_calls['turn_on'])) ump.mute_volume(True) self.assertEqual(1, len(self.mock_mp_2.service_calls['mute_volume'])) ump.set_volume_level(0.5) self.assertEqual( 1, len(self.mock_mp_2.service_calls['set_volume_level'])) ump.media_play() self.assertEqual(1, len(self.mock_mp_2.service_calls['media_play'])) ump.media_pause() self.assertEqual(1, len(self.mock_mp_2.service_calls['media_pause'])) ump.media_previous_track() self.assertEqual( 1, len(self.mock_mp_2.service_calls['media_previous_track'])) ump.media_next_track() self.assertEqual( 1, len(self.mock_mp_2.service_calls['media_next_track'])) ump.media_seek(100) self.assertEqual(1, len(self.mock_mp_2.service_calls['media_seek'])) ump.play_media('movie', 'batman') self.assertEqual(1, len(self.mock_mp_2.service_calls['play_media'])) ump.volume_up() self.assertEqual(1, len(self.mock_mp_2.service_calls['volume_up'])) ump.volume_down() self.assertEqual(1, len(self.mock_mp_2.service_calls['volume_down'])) ump.media_play_pause() self.assertEqual( 1, len(self.mock_mp_2.service_calls['media_play_pause'])) ump.select_source('dvd') self.assertEqual( 1, len(self.mock_mp_2.service_calls['select_source']))
def test_polling(self): """Test should_poll property.""" config = self.config_children_only universal.validate_config(config) ump = universal.UniversalMediaPlayer(self.hass, **config) self.assertEqual(False, ump.should_poll)
def test_name(self): """Test name property.""" config = self.config_children_only universal.validate_config(config) ump = universal.UniversalMediaPlayer(self.hass, **config) self.assertEqual(config['name'], ump.name)
def test_master_state(self): """Test master state property.""" config = self.config_children_only universal.validate_config(config) ump = universal.UniversalMediaPlayer(self.hass, **config) self.assertEqual(None, ump.master_state)
def test_service_call_to_child(self): """Test service calls that should be routed to a child.""" config = self.config_children_only universal.validate_config(config) ump = universal.UniversalMediaPlayer(self.hass, **config) ump.entity_id = media_player.ENTITY_ID_FORMAT.format(config['name']) ump.update() self.mock_mp_2._state = STATE_PLAYING self.mock_mp_2.update_ha_state() ump.update() ump.turn_off() self.assertEqual(1, len(self.mock_mp_2.service_calls['turn_off'])) ump.turn_on() self.assertEqual(1, len(self.mock_mp_2.service_calls['turn_on'])) ump.mute_volume(True) self.assertEqual(1, len(self.mock_mp_2.service_calls['mute_volume'])) ump.set_volume_level(0.5) self.assertEqual(1, len(self.mock_mp_2.service_calls['set_volume_level'])) ump.media_play() self.assertEqual(1, len(self.mock_mp_2.service_calls['media_play'])) ump.media_pause() self.assertEqual(1, len(self.mock_mp_2.service_calls['media_pause'])) ump.media_previous_track() self.assertEqual( 1, len(self.mock_mp_2.service_calls['media_previous_track'])) ump.media_next_track() self.assertEqual(1, len(self.mock_mp_2.service_calls['media_next_track'])) ump.media_seek(100) self.assertEqual(1, len(self.mock_mp_2.service_calls['media_seek'])) ump.play_media('movie', 'batman') self.assertEqual(1, len(self.mock_mp_2.service_calls['play_media'])) ump.volume_up() self.assertEqual(1, len(self.mock_mp_2.service_calls['volume_up'])) ump.volume_down() self.assertEqual(1, len(self.mock_mp_2.service_calls['volume_down'])) ump.media_play_pause() self.assertEqual(1, len(self.mock_mp_2.service_calls['media_play_pause'])) ump.select_source('dvd') self.assertEqual(1, len(self.mock_mp_2.service_calls['select_source']))
def test_master_state_with_bad_attrs(self): """Test master state property.""" config = self.config_children_and_attr config['attributes']['state'] = 'bad.entity_id' universal.validate_config(config) ump = universal.UniversalMediaPlayer(self.hass, **config) self.assertEqual(STATE_OFF, ump.master_state)
def test_master_state_with_attrs(self): """Test master state property.""" config = self.config_children_and_attr universal.validate_config(config) ump = universal.UniversalMediaPlayer(self.hass, **config) self.assertEqual(STATE_OFF, ump.master_state) self.hass.states.set(self.mock_state_switch_id, STATE_ON) self.assertEqual(STATE_ON, ump.master_state)
def test_is_volume_muted_children_and_attr(self): """Test is volume muted property w/ children and attrs.""" config = self.config_children_and_attr universal.validate_config(config) ump = universal.UniversalMediaPlayer(self.hass, **config) self.assertFalse(ump.is_volume_muted) self.hass.states.set(self.mock_mute_switch_id, STATE_ON) self.assertTrue(ump.is_volume_muted)
def test_config_bad_children(self): """Check config with bad children entry.""" config_no_children = {'name': 'test', 'platform': 'universal'} config_bad_children = {'name': 'test', 'children': {}, 'platform': 'universal'} response = universal.validate_config(config_no_children) self.assertTrue(response) self.assertEqual([], config_no_children['children']) response = universal.validate_config(config_bad_children) self.assertTrue(response) self.assertEqual([], config_bad_children['children'])
def test_state_children_only(self): """Test media player state with only children.""" config = self.config_children_only universal.validate_config(config) ump = universal.UniversalMediaPlayer(self.hass, **config) ump.entity_id = media_player.ENTITY_ID_FORMAT.format(config['name']) ump.update() self.assertTrue(ump.state, STATE_OFF) self.mock_mp_1._state = STATE_PLAYING self.mock_mp_1.update_ha_state() ump.update() self.assertEqual(STATE_PLAYING, ump.state)
def test_config_bad_commands(self): """Check config with bad commands entry.""" config = {'name': 'test', 'commands': [], 'platform': 'universal'} response = universal.validate_config(config) self.assertTrue(response) self.assertEqual({}, config['commands'])
def test_config_bad_key(self): """Check config with bad key.""" config = {'name': 'test', 'asdf': 5, 'platform': 'universal'} response = universal.validate_config(config) self.assertTrue(response) self.assertFalse('asdf' in config)
def test_config_bad_attributes(self): """Check config with bad attributes.""" config = {'name': 'test', 'attributes': [], 'platform': 'universal'} response = universal.validate_config(config) self.assertTrue(response) self.assertEqual({}, config['attributes'])
def test_config_bad_children(self): """Check config with bad children entry.""" config_no_children = {'name': 'test', 'platform': 'universal'} config_bad_children = { 'name': 'test', 'children': {}, 'platform': 'universal' } response = universal.validate_config(config_no_children) self.assertTrue(response) self.assertEqual([], config_no_children['children']) response = universal.validate_config(config_bad_children) self.assertTrue(response) self.assertEqual([], config_bad_children['children'])
def test_supported_media_commands_children_only(self): """Test supported media commands with only children.""" config = self.config_children_only universal.validate_config(config) ump = universal.UniversalMediaPlayer(self.hass, **config) ump.entity_id = media_player.ENTITY_ID_FORMAT.format(config['name']) ump.update() self.assertEqual(0, ump.supported_media_commands) self.mock_mp_1._supported_media_commands = 512 self.mock_mp_1._state = STATE_PLAYING self.mock_mp_1.update_ha_state() ump.update() self.assertEqual(512, ump.supported_media_commands)
def test_config_children_and_attr(self): """Check config with children and attributes.""" config_start = copy(self.config_children_and_attr) del config_start['platform'] config_start['commands'] = {} response = universal.validate_config(self.config_children_and_attr) self.assertTrue(response) self.assertEqual(config_start, self.config_children_and_attr)
def test_service_call_to_command(self): """Test service call to command.""" config = self.config_children_only config['commands'] = {'turn_off': { 'service': 'test.turn_off', 'data': {}}} universal.validate_config(config) service = mock_service(self.hass, 'test', 'turn_off') ump = universal.UniversalMediaPlayer(self.hass, **config) ump.entity_id = media_player.ENTITY_ID_FORMAT.format(config['name']) ump.update() self.mock_mp_2._state = STATE_PLAYING self.mock_mp_2.update_ha_state() ump.update() ump.turn_off() self.assertEqual(1, len(service))
def test_volume_level(self): """Test volume level property.""" config = self.config_children_only universal.validate_config(config) ump = universal.UniversalMediaPlayer(self.hass, **config) ump.entity_id = media_player.ENTITY_ID_FORMAT.format(config['name']) ump.update() self.assertEqual(None, ump.volume_level) self.mock_mp_1._state = STATE_PLAYING self.mock_mp_1.update_ha_state() ump.update() self.assertEqual(0, ump.volume_level) self.mock_mp_1._volume_level = 1 self.mock_mp_1.update_ha_state() ump.update() self.assertEqual(1, ump.volume_level)
def test_is_volume_muted_children_only(self): """Test is volume muted property w/ children only.""" config = self.config_children_only universal.validate_config(config) ump = universal.UniversalMediaPlayer(self.hass, **config) ump.entity_id = media_player.ENTITY_ID_FORMAT.format(config['name']) ump.update() self.assertFalse(ump.is_volume_muted) self.mock_mp_1._state = STATE_PLAYING self.mock_mp_1.update_ha_state() ump.update() self.assertFalse(ump.is_volume_muted) self.mock_mp_1._is_volume_muted = True self.mock_mp_1.update_ha_state() ump.update() self.assertTrue(ump.is_volume_muted)
def test_service_call_to_command(self): """Test service call to command.""" config = self.config_children_only config['commands'] = { 'turn_off': { 'service': 'test.turn_off', 'data': {} } } universal.validate_config(config) service = mock_service(self.hass, 'test', 'turn_off') ump = universal.UniversalMediaPlayer(self.hass, **config) ump.entity_id = media_player.ENTITY_ID_FORMAT.format(config['name']) ump.update() self.mock_mp_2._state = STATE_PLAYING self.mock_mp_2.update_ha_state() ump.update() ump.turn_off() self.assertEqual(1, len(service))
def test_state_with_children_and_attrs(self): """Test media player with children and master state.""" config = self.config_children_and_attr universal.validate_config(config) ump = universal.UniversalMediaPlayer(self.hass, **config) ump.entity_id = media_player.ENTITY_ID_FORMAT.format(config['name']) ump.update() self.assertEqual(STATE_OFF, ump.state) self.hass.states.set(self.mock_state_switch_id, STATE_ON) ump.update() self.assertEqual(STATE_ON, ump.state) self.mock_mp_1._state = STATE_PLAYING self.mock_mp_1.update_ha_state() ump.update() self.assertEqual(STATE_PLAYING, ump.state) self.hass.states.set(self.mock_state_switch_id, STATE_OFF) ump.update() self.assertEqual(STATE_OFF, ump.state)
def test_supported_media_commands_children_and_cmds(self): """Test supported media commands with children and attrs.""" config = self.config_children_and_attr universal.validate_config(config) config['commands']['turn_on'] = 'test' config['commands']['turn_off'] = 'test' config['commands']['volume_up'] = 'test' config['commands']['volume_down'] = 'test' config['commands']['volume_mute'] = 'test' ump = universal.UniversalMediaPlayer(self.hass, **config) ump.entity_id = media_player.ENTITY_ID_FORMAT.format(config['name']) ump.update() self.mock_mp_1._supported_media_commands = universal.SUPPORT_VOLUME_SET self.mock_mp_1._state = STATE_PLAYING self.mock_mp_1.update_ha_state() ump.update() check_flags = universal.SUPPORT_TURN_ON | universal.SUPPORT_TURN_OFF \ | universal.SUPPORT_VOLUME_STEP | universal.SUPPORT_VOLUME_MUTE self.assertEqual(check_flags, ump.supported_media_commands)
def test_config_no_name(self): """Check config with no Name entry.""" response = universal.validate_config({'platform': 'universal'}) self.assertFalse(response)