X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=test%2Fredtube_test.rb;h=9d6c709120fc0a3f46b5a948741629dffa2d1ee7;hb=e756e0b650774f2503702512ccdc02e86eee1788;hp=a929f612dad16583220ca97fd2faad60a72dddfc;hpb=168ef3b2ccf5b97d561a3c542a18e8e7587de291;p=dead%2Fwhatever-dl.git

diff --git a/test/redtube_test.rb b/test/redtube_test.rb
index a929f61..9d6c709 100644
--- 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
@@ -29,10 +47,18 @@ class RedtubeTest < Test::Unit::TestCase
 
   
   def test_get_video_url
-    rt = Redtube.new()
-
-    test_result = rt.get_video_url('http://www.redtube.com/6807')
-    assert_equal("http://dl.redtube.com/_videos_t4vn23s9jc5498tgj49icfj4678/0000006/X57OBH08G.flv", test_result)
+    rt = Redtube.new('http://www.redtube.com/6807')
+    expected_result = 'http://dl.redtube.com/_videos_t4vn23s9jc5498tgj49icfj4678/0000006/X57OBH08G.flv'
+    actual_result = rt.get_video_url()
+    assert_equal(expected_result, actual_result)
   end
 
+  
+  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
 end