create_station returns a Station model

related: #57

2 files changed, 7 insertions, 4 deletions

diff --git a/pandora/client.py b/pandora/client.py
index 0eaa977..01b7654 100644
--- a/pandora/client.py
+++ b/pandora/client.py
@@ -199,6 +199,8 @@ class APIClient(BaseAPIClient):
 
     def create_station(self, search_token=None, artist_token=None,
                        track_token=None):
+        from .models.pandora import Station
+
         kwargs = {}
 
         if search_token:
@@ -210,7 +212,8 @@ class APIClient(BaseAPIClient):
         else:
             raise KeyError("Must pass a type of token")
 
-        return self("station.createStation", **kwargs)
+        return Station.from_json(self,
+                                 self("station.createStation", **kwargs))
 
     def delete_feedback(self, feedback_id):  # pragma: no cover
         return self("station.deleteFeedback",
diff --git a/tests/test_pandora/test_client.py b/tests/test_pandora/test_client.py
index f88b755..6c61355 100644
--- a/tests/test_pandora/test_client.py
+++ b/tests/test_pandora/test_client.py
@@ -147,19 +147,19 @@ class TestGettingAds(TestCase):
 class TestCreatingStation(TestCase):
 
     def test_using_search_token(self):
-        client = APIClient(Mock(), None, None, None, None)
+        client = APIClient(Mock(return_value={}), None, None, None, None)
         client.create_station(search_token="foo")
         client.transport.assert_called_with(
             "station.createStation", musicToken="foo")
 
     def test_using_artist_token(self):
-        client = APIClient(Mock(), None, None, None, None)
+        client = APIClient(Mock(return_value={}), None, None, None, None)
         client.create_station(artist_token="foo")
         client.transport.assert_called_with(
             "station.createStation", trackToken="foo", musicType="artist")
 
     def test_using_track_token(self):
-        client = APIClient(Mock(), None, None, None, None)
+        client = APIClient(Mock(return_value={}), None, None, None, None)
         client.create_station(track_token="foo")
         client.transport.assert_called_with(
             "station.createStation", trackToken="foo", musicType="song")