Updated Redtube to work with the site's new algorithm.

[dead/whatever-dl.git] / test / redtube_test.rb

diff --git a/test/redtube_test.rb b/test/redtube_test.rb
index a929f612dad16583220ca97fd2faad60a72dddfc..e226f815b9550a1a928bb5b82193e852aa1a8963 100644 (file)
--- a/test/redtube_test.rb
+++ b/test/redtube_test.rb
@@ -1,9 +1,27 @@
-# Unit tests for the Redtube class. Basically just checking
-# the results of get_video_url for known ids.
+#
+# Copyright Michael Orlitzky
+#
+# http://michael.orlitzky.com/
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# http://www.fsf.org/licensing/licenses/gpl.html
+#
 
 require 'test/unit'
 require 'src/websites/redtube'
 
+# Unit tests for the Redtube class. Basically just checking
+# the results of get_video_url for known ids.
+
 class RedtubeTest < Test::Unit::TestCase
 
   def test_owns_redtube_urls
@@ -28,11 +46,27 @@ class RedtubeTest < Test::Unit::TestCase
   end
 
   
-  def test_get_video_url
-    rt = Redtube.new()
+  def test_get_video_filename
+    rt = Redtube.new('http://www.redtube.com/6807')
+    # I don't know where they get these filenames from, but whatever.
+    expected_result = 'X57OBH08G.flv'
+    actual_result = rt.get_video_filename()
+    assert_equal(expected_result, actual_result)
+  end
+
+
+
+  def test_parse_hash_flv
+    rt = Redtube.new(nil)
+
+    page_data = nil
+
+    File.open('test/fixtures/redtube/6807.html') do |f|
+      page_data = f.read
+    end
 
-    test_result = rt.get_video_url('http://www.redtube.com/6807')
-    assert_equal("http://dl.redtube.com/_videos_t4vn23s9jc5498tgj49icfj4678/0000006/X57OBH08G.flv", test_result)
+    test_result = rt.send('parse_hash_flv', page_data)
+    assert_equal('/be03bc2d810017a8689d96c61a626303/4ac6529f', test_result)
   end
 
 end