def setUp(self):
data = {
'id': 'test',
'name': '',
'stream_id': 'test stream',
'muted': False,
'clients': [
{'id': 'a'},
{'id': 'b'}
]
}
server = MagicMock()
stream = MagicMock()
stream.friendly_name = 'test stream'
stream.status = 'playing'
client = MagicMock()
client.volume = 50
server.streams = [stream]
server.stream = MagicMock(return_value=stream)
server.client = MagicMock(return_value=client)
self.group = Snapgroup(server, data)
def synchronize(self, status):
"""Synchronize snapserver."""
self._version = status.get('server').get('version')
self._groups = {}
self._clients = {}
self._streams = {}
for stream in status.get('server').get('streams'):
self._streams[stream.get('id')] = Snapstream(stream)
_LOGGER.debug('stream found: %s', self._streams[stream.get('id')])
for group in status.get('server').get('groups'):
self._groups[group.get('id')] = Snapgroup(self, group)
_LOGGER.debug('group found: %s', self._groups[group.get('id')])
for client in group.get('clients'):
self._clients[client.get('id')] = Snapclient(self, client)
_LOGGER.debug('client found: %s', self._clients[client.get('id')])
class TestSnapgroup(unittest.TestCase):
def setUp(self):
data = {
'id': 'test',
'name': '',
'stream_id': 'test stream',
'muted': False,
'clients': [
{'id': 'a'},
{'id': 'b'}
]
}
server = MagicMock()
stream = MagicMock()
stream.friendly_name = 'test stream'
stream.status = 'playing'
client = MagicMock()
client.volume = 50
server.streams = [stream]
server.stream = MagicMock(return_value=stream)
server.client = MagicMock(return_value=client)
self.group = Snapgroup(server, data)
def test_init(self):
self.assertEqual(self.group.identifier, 'test')
self.assertEqual(self.group.name, '')
self.assertEqual(self.group.friendly_name, 'test stream')
self.assertEqual(self.group.stream, 'test stream')
self.assertEqual(self.group.muted, False)
self.assertEqual(self.group.volume, 50)
self.assertEqual(self.group.clients, ['a', 'b'])
self.assertEqual(self.group.stream_status, 'playing')
def test_update(self):
self.group.update({
'stream_id': 'other stream'
})
self.assertEqual(self.group.stream, 'other stream')
def test_set_muted(self):
async_run(self.group.set_muted(True))
self.assertEqual(self.group.muted, True)
def test_set_volume(self):
async_run(self.group.set_volume(75))
def test_set_stream(self):
async_run(self.group.set_stream('new stream'))
self.assertEqual(self.group.stream, 'new stream')
def test_add_client(self):
self.group.add_client('c')
# TODO: add assert
def test_remove_client(self):
self.group.remove_client('a')
# TODO: add assert
def test_streams_by_name(self):
self.assertEqual(self.group.streams_by_name().keys(), set(['test stream']))
def test_update_mute(self):
self.group.update_mute({'mute': True})
self.assertEqual(self.group.muted, True)
def test_update_stream(self):
self.group.update_stream({'stream_id': 'other stream'})
self.assertEqual(self.group.stream, 'other stream')
def test_snapshot_restore(self):
async_run(self.group.set_muted(False))
self.group.snapshot()
async_run(self.group.set_muted(True))
self.assertEqual(self.group.muted, True)
async_run(self.group.restore())
self.assertEqual(self.group.muted, False)
def test_set_callback(self):
cb = MagicMock()
self.group.set_callback(cb)
self.group.update_mute({'mute': True})
cb.assert_called_with(self.group)