test/redtube_test.rb

   1 #
   2 # Copyright Michael Orlitzky
   3 #
   4 # http://michael.orlitzky.com/
   5 #
   6 # This program is free software: you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation, either version 3 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14 # GNU General Public License for more details.
  15 #
  16 # http://www.fsf.org/licensing/licenses/gpl.html
  17 #
  18
  19 require 'test/unit'
  20 require 'src/websites/redtube'
  21
  22 # Unit tests for the Redtube class. Basically just checking
  23 # the results of get_video_url for known ids.
  24
  25 class RedtubeTest < Test::Unit::TestCase
  26
  27   def test_owns_redtube_urls
  28     assert(Redtube.owns_url?('http://www.redtube.com/6807'))
  29     assert(Redtube.owns_url?('www.redtube.com/6807'))
  30     assert(Redtube.owns_url?('http://redtube.com/6807'))
  31     assert(Redtube.owns_url?('redtube.com/6807'))
  32   end
  33
  34
  35   def test_doesnt_own_howcast_urls
  36     assert(!Redtube.owns_url?('http://www.howcast.com/6807'))
  37     assert(!Redtube.owns_url?('www.howcast.com/6807'))
  38     assert(!Redtube.owns_url?('http://howcast.com/6807'))
  39     assert(!Redtube.owns_url?('howcast.com/6807'))
  40   end
  41
  42
  43   def test_doesnt_own_misc_urls
  44     assert(!Redtube.owns_url?('http://redtube/123'))
  45     assert(!Redtube.owns_url?('www.redtube.com/abc'))
  46   end
  47
  48
  49   def test_get_video_filename
  50     rt = Redtube.new('http://www.redtube.com/6807')
  51     # I don't know where they get these filenames from, but whatever.
  52     expected_result = 'X57OBH08G.flv'
  53     actual_result = rt.get_video_filename()
  54     assert_equal(expected_result, actual_result)
  55   end
  56
  57
  58
  59   def test_parse_hash_flv
  60     rt = Redtube.new(nil)
  61
  62     page_data = nil
  63
  64     File.open('test/fixtures/redtube/6807.html') do |f|
  65       page_data = f.read
  66     end
  67
  68     test_result = rt.send('parse_hash_flv', page_data)
  69     assert_equal('/be03bc2d810017a8689d96c61a626303/4ac6529f', test_result)
  70   end
  71
  72 end