add missing test

aantix · Dec 13, 2011 · 719cc49 · 719cc49
1 parent dafa108
commit 719cc49
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/lib/youtube_search.rb b/lib/youtube_search.rb
@@ -11,8 +11,7 @@ def self.search(query, options={})
   end
 
   def self.playlist_videos(playlist_id)
-    playlist_id.gsub!(/^PL/, "")
-
+    playlist_id = playlist_id.sub(/^PL/, "")
     xml = open("https://gdata.youtube.com/feeds/api/playlists/#{playlist_id}?v=2").read
     parse(xml, :type => :playlist)
   end

diff --git a/spec/youtube_search_spec.rb b/spec/youtube_search_spec.rb
@@ -68,6 +68,12 @@
     it "can retrieve videos from a playlist" do
       YoutubeSearch.playlist_videos('5F23DAF4BFE3D14C').size.should == 6
     end
+
+    it "can retrieve videos from a playlist with new-style ids" do
+      id = 'PL5F23DAF4BFE3D14C'
+      YoutubeSearch.playlist_videos(id).size.should == 6
+      id.should == 'PL5F23DAF4BFE3D14C'
+    end
   end
 
   describe 'options_with_per_page_and_page' do