From e2474da1efa4de0425903500e3dcdec1ec3c6e9e Mon Sep 17 00:00:00 2001
From: Stein Magnus Jodal <stein.magnus@jodal.no>
Date: Wed, 31 Oct 2012 09:42:35 +0100
Subject: [PATCH] Make core.stored_playlists.create() support multibackend
 (#217)

---
 mopidy/core/stored_playlists.py     | 16 +++++++++++++---
 tests/core/stored_playlists_test.py | 22 ++++++++++++++++++++++
 2 files changed, 35 insertions(+), 3 deletions(-)

diff --git a/mopidy/core/stored_playlists.py b/mopidy/core/stored_playlists.py
index a3d52023..55b0bdce 100644
--- a/mopidy/core/stored_playlists.py
+++ b/mopidy/core/stored_playlists.py
@@ -21,16 +21,26 @@ class StoredPlaylistsController(object):
         results = pykka.get_all(futures)
         return list(itertools.chain(*results))
 
-    def create(self, name):
+    def create(self, name, uri_scheme=None):
         """
         Create a new playlist.
 
+        If ``uri_scheme`` matches an URI scheme handled by a current backend,
+        that backend is asked to create the playlist. If ``uri_scheme`` is
+        :class:`None` or doesn't match a current backend, the first backend is
+        asked to create the playlist.
+
         :param name: name of the new playlist
         :type name: string
+        :param uri_scheme: use the backend matching the URI scheme
+        :type uri_scheme: string
         :rtype: :class:`mopidy.models.Playlist`
         """
-        # TODO Support multiple backends
-        return self.backends[0].stored_playlists.create(name).get()
+        if uri_scheme in self.backends.by_uri_scheme:
+            backend = self.backends.by_uri_scheme[uri_scheme]
+        else:
+            backend = self.backends[0]
+        return backend.stored_playlists.create(name).get()
 
     def delete(self, playlist):
         """
diff --git a/tests/core/stored_playlists_test.py b/tests/core/stored_playlists_test.py
index d92b89c0..87c90137 100644
--- a/tests/core/stored_playlists_test.py
+++ b/tests/core/stored_playlists_test.py
@@ -37,5 +37,27 @@ class StoredPlaylistsTest(unittest.TestCase):
         self.assertIn(self.pl2a, result)
         self.assertIn(self.pl2b, result)
 
+    def test_create_without_uri_scheme_uses_first_backend(self):
+        playlist = Playlist()
+        self.sp1.create().get.return_value = playlist
+        self.sp1.reset_mock()
+
+        result = self.core.stored_playlists.create('foo')
+
+        self.assertEqual(playlist, result)
+        self.sp1.create.assert_called_once_with('foo')
+        self.assertFalse(self.sp2.create.called)
+
+    def test_create_with_uri_scheme_selects_the_matching_backend(self):
+        playlist = Playlist()
+        self.sp2.create().get.return_value = playlist
+        self.sp2.reset_mock()
+
+        result = self.core.stored_playlists.create('foo', uri_scheme='dummy2')
+
+        self.assertEqual(playlist, result)
+        self.assertFalse(self.sp1.create.called)
+        self.sp2.create.assert_called_once_with('foo')
+
     # TODO The rest of the stored playlists API is pending redesign before
     # we'll update it to support multiple backends.