X-Git-Url: http://gitweb.michael.orlitzky.com/?p=dead%2Fwhatever-dl.git;a=blobdiff_plain;f=test%2Fyoutube_test.rb;h=be356aac9716a5e25e2499e8f8cb7820090cce60;hp=ab92f2d115123e6f5e40576ce8a034436012b9fe;hb=90be7049de8f1b9a8da3951a12f922a2e1784d3e;hpb=dbde07faafd93f8a9503c1fedfe317723da7cac5

diff --git a/test/youtube_test.rb b/test/youtube_test.rb
index ab92f2d..be356aa 100644
--- a/test/youtube_test.rb
+++ b/test/youtube_test.rb
@@ -96,6 +96,22 @@ class YoutubeTest < Test::Unit::TestCase
   end
 
 
+  def test_parse_t_parameter_from_url
+    # Tests a different regex than the other 't' parameter tests.
+    yt = Youtube.new(nil)
+
+    page_data = nil
+
+    File.open('test/fixtures/youtube/cpU27Xp8zdM.html') do |f|
+      page_data = f.read
+    end
+
+    expected_result = 'vjVQa1PpcFNyKz8nP3yZ3MHcwOvLXsOWpCJHkgJVShI='
+    actual_result = yt.send('parse_t_parameter', page_data)
+    assert_equal(expected_result, actual_result)
+  end
+
+
   def test_get_available_formats
     # Make sure that we can parse the available formats from a page.
     yt = Youtube.new(nil)
@@ -116,6 +132,26 @@ class YoutubeTest < Test::Unit::TestCase
   end
 
 
+  def test_get_available_formats_from_url
+    # Tests a different regex than the other available formats test.
+    yt = Youtube.new(nil)
+
+    page_data = nil
+
+    File.open('test/fixtures/youtube/cpU27Xp8zdM.html') do |f|
+      page_data = f.read
+    end
+
+    actual_formats = yt.send('get_available_formats', page_data)
+    expected_formats = [ 34, 5 ]
+
+    # Make sure that all of the elements of the "expected" set are in
+    # the "actual" set and vice-versa.
+    assert_equal([], expected_formats - actual_formats)
+    assert_equal([], actual_formats - expected_formats)
+  end
+
+
   def test_get_desired_format
     # The get_desired_format() method should choose the best of the
     # available formats.