From: Michael Orlitzky <michael@orlitzky.com>
Date: Thu, 14 Jan 2010 04:13:43 +0000 (-0500)
Subject: Added two new Youtube tests for the available/desired video formats.
X-Git-Url: https://gitweb.michael.orlitzky.com/?a=commitdiff_plain;h=60c71dd77a54a8e2e482e9b43c0c917d174ffe03;p=dead%2Fwhatever-dl.git

Added two new Youtube tests for the available/desired video formats.
---

diff --git a/test/youtube_test.rb b/test/youtube_test.rb
index fa61cb2..ec91d35 100644
--- a/test/youtube_test.rb
+++ b/test/youtube_test.rb
@@ -95,6 +95,45 @@ class YoutubeTest < Test::Unit::TestCase
   end
 
 
+  def test_get_available_formats
+    # Make sure that we can parse the available formats from a page.
+    yt = Youtube.new(nil)
+
+    page_data = nil
+
+    File.open('test/fixtures/youtube/BUer8Dv2HW8.html') do |f|
+      page_data = f.read
+    end
+
+    actual_formats = yt.send('get_available_formats', page_data)
+    expected_formats = [ 37, 22, 35, 18, 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.
+    yt = Youtube.new(nil)
+
+    page_data = nil
+
+    File.open('test/fixtures/youtube/BUer8Dv2HW8.html') do |f|
+      page_data = f.read
+    end
+
+    available_formats = yt.send('get_available_formats', page_data)
+    actual_result = yt.send('get_desired_format', available_formats)
+    expected_result = 37
+
+    assert_equal(expected_result, actual_result)
+  end
+
+
   def test_get_video_filename
     yt = Youtube.new('http://www.youtube.com/watch?v=SudixyugiX4')
     expected_result = 'SudixyugiX4.flv'