tests for translate_uri

mopidy_emby/backend.py

@@ -34,15 +34,19 @@ def __init__(self, config, audio):
 class EmbyPlaybackProvider(backend.PlaybackProvider):
 
     def translate_uri(self, uri):
-        id = uri.split(':')[-1]
+        if uri.startswith('emby:track:') and len(uri.split(':')) == 3:
+            id = uri.split(':')[-1]
 
-        track_url = self.backend.remote.api_url(
-            '/Audio/{}/stream.mp3'.format(id)
-        )
+            track_url = self.backend.remote.api_url(
+                '/Audio/{}/stream.mp3'.format(id)
+            )
 
-        logger.debug('Emby track streaming url: {}'.format(track_url))
+            logger.debug('Emby track streaming url: {}'.format(track_url))
 
-        return track_url
+            return track_url
+
+        else:
+            return None
 
 
 class EmbyLibraryProvider(backend.LibraryProvider):

tests/conftest.py

@@ -34,6 +34,12 @@ def emby_client(config, mocker):
 @pytest.fixture
 def backend_mock():
     backend_mock = mock.Mock(autospec=mopidy_emby.backend.EmbyBackend)
+
+    return backend_mock
+
+
+@pytest.fixture
+def libraryprovider(backend_mock):
     backend_mock.remote(autospec=mopidy_emby.backend.EmbyHandler)
     backend_mock.remote.get_artists.return_value = ['Artistlist']
     backend_mock.remote.get_albums.return_value = ['Albumlist']
@@ -43,9 +49,12 @@ def backend_mock():
         'Id': 123
     }
 
-    return backend_mock
+    return mopidy_emby.backend.EmbyLibraryProvider(backend_mock)
 
 
 @pytest.fixture
-def provider(backend_mock):
-    return mopidy_emby.backend.EmbyLibraryProvider(backend_mock)
+def playbackprovider(backend_mock, emby_client):
+    backend_mock.remote = emby_client
+
+    return mopidy_emby.backend.EmbyPlaybackProvider(audio=mock.Mock(),
+                                                    backend=backend_mock)

tests/test_extension.py

@@ -213,21 +213,35 @@ def test_create_artists(data, expected, emby_client):
     ('emby:artist:123', ['Albumlist']),
     ('emby:album:123', ['Tracklist']),
 ])
-def test_browse(uri, expected, provider):
-    assert provider.browse(uri) == expected
+def test_browse(uri, expected, libraryprovider):
+    assert libraryprovider.browse(uri) == expected
 
 
 @pytest.mark.parametrize('uri,expected', [
     ('emby:track:123', [{'Name': 'Foo', 'Id': 123}]),
     ('emby:track', [])
 ])
-def test_lookup_uri(uri, expected, provider):
-    assert provider.lookup(uri=uri) == expected
+def test_lookup_uri(uri, expected, libraryprovider):
+    assert libraryprovider.lookup(uri=uri) == expected
 
 
 @pytest.mark.parametrize('uri,expected', [
     (['emby:track:123'], {'emby:track:123': [{'Name': 'Foo', 'Id': 123}]}),
     (['emby:track'], {u'emby:track': []})
 ])
-def test_lookup_uris(uri, expected, provider):
-    assert provider.lookup(uris=uri) == expected
+def test_lookup_uris(uri, expected, libraryprovider):
+    assert libraryprovider.lookup(uris=uri) == expected
+
+
+@pytest.mark.parametrize('uri,expected', [
+    (
+        'emby:track:123',
+        'https://foo.bar:443/Audio/123/stream.mp3?format=json'
+    ),
+    (
+        'emby:foobar',
+        None
+    )
+])
+def test_translate_uri(playbackprovider, uri, expected):
+    assert playbackprovider.translate_uri(uri) == expected