]> gitweb.michael.orlitzky.com - dead/whatever-dl.git/blob - test/youtube_test.rb
Added GetOpt option parsing.
[dead/whatever-dl.git] / test / youtube_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/youtube'
21
22 class YoutubeTest < Test::Unit::TestCase
23
24 def test_owns_youtube_urls
25 assert(Youtube.owns_url?('http://www.youtube.com/watch?v=SudixyugiX4'))
26 assert(Youtube.owns_url?('http://www.youtube.com/watch?v=SudixyugiX4&hl=en'))
27 assert(Youtube.owns_url?('http://youtube.com/watch?v=SudixyugiX4&hl=en'))
28 assert(Youtube.owns_url?('http://www.youtube.com/v/SudixyugiX4'))
29 assert(Youtube.owns_url?('http://www.youtube.com/v/SudixyugiX4&hl=en'))
30 assert(Youtube.owns_url?('http://youtube.com/v/SudixyugiX4&hl=en'))
31 assert(Youtube.owns_url?('http://www.youtube.com/watch?v=K9iDMcmm0tE'))
32 assert(Youtube.owns_url?('http://www.youtube.com/watch?v=K9iDMcmm0tE#'))
33 assert(Youtube.owns_url?('http://www.youtube.com/v/K9iDMcmm0tE'))
34 assert(Youtube.owns_url?('http://www.youtube.com/watch?v=83-hlYMH1XE'))
35 assert(Youtube.owns_url?('http://www.youtube.com/watch?v=83-hlYMH1XE&feature=dir'))
36 assert(Youtube.owns_url?('http://in.youtube.com/watch?v=VcydqSpYN00&feature=channel_page'))
37 assert(Youtube.owns_url?('http://uk.youtube.com/watch?v=LN4Ov6ZLcrI'))
38 end
39
40
41 def test_doesnt_own_redtube_urls
42 assert(!Youtube.owns_url?('http://www.redtube.com/6807'))
43 assert(!Youtube.owns_url?('www.redtube.com/6807'))
44 assert(!Youtube.owns_url?('http://redtube.com/6807'))
45 assert(!Youtube.owns_url?('redtube.com/6807'))
46 end
47
48
49 def test_parse_video_id
50 yt = Youtube.new('http://www.youtube.com/watch?v=SudixyugiX4')
51 expected_result = 'SudixyugiX4'
52 actual_result = yt.send('parse_video_id')
53 assert_equal(expected_result, actual_result)
54 end
55
56
57 def test_parse_video_id_again
58 yt = Youtube.new('http://www.youtube.com/watch?v=K9iDMcmm0tE')
59 expected_result = 'K9iDMcmm0tE'
60 actual_result = yt.send('parse_video_id')
61 assert_equal(expected_result, actual_result)
62 end
63
64
65 def test_parse_t_parameter
66 yt = Youtube.new(nil)
67
68 page_data = nil
69
70 File.open('test/fixtures/youtube/SudixyugiX4.html') do |f|
71 page_data = f.read
72 end
73
74 expected_result = 'OEgsToPDskLQUAntWWpzhEMhBMlgqHdo'
75 actual_result = yt.send('parse_t_parameter', page_data)
76 assert_equal(expected_result, actual_result)
77 end
78
79
80 def test_parse_t_parameter_again
81 # This was failing once because my regex missed a hyphen.
82 # I modified the regex to match anything between the pair
83 # of quotes, so it should now catch the hyphen and underscore.
84 yt = Youtube.new(nil)
85
86 page_data = nil
87
88 File.open('test/fixtures/youtube/K9iDMcmm0tE.html') do |f|
89 page_data = f.read
90 end
91
92 expected_result = 'O_EgsToPDskJsXVvAXpAct1zug-lBJBz'
93 actual_result = yt.send('parse_t_parameter', page_data)
94 assert_equal(expected_result, actual_result)
95 end
96
97
98 def test_get_video_filename
99 yt = Youtube.new('http://www.youtube.com/watch?v=SudixyugiX4')
100 expected_result = 'SudixyugiX4.flv'
101 actual_result = yt.get_video_filename()
102 assert_equal(expected_result, actual_result)
103 end
104
105
106 def test_get_video_filename_again
107 yt = Youtube.new('http://www.youtube.com/watch?v=K9iDMcmm0tE')
108 expected_result = 'K9iDMcmm0tE.flv'
109 actual_result = yt.get_video_filename()
110 assert_equal(expected_result, actual_result)
111 end
112
113
114 def test_get_troublesome_video_filename_first_form
115 # This non-alphanumeric video id was causing
116 # get_video_filename to barf.
117 yt = Youtube.new('http://www.youtube.com/watch?v=83-hlYMH1XE&feature=dir')
118 expected_result = '83-hlYMH1XE.flv'
119 actual_result = yt.get_video_filename()
120 assert_equal(expected_result, actual_result)
121 end
122
123
124 def test_get_troublesome_video_filename_second_form
125 yt = Youtube.new('http://www.youtube.com/v/83-hlYMH1XE&feature=dir')
126 expected_result = '83-hlYMH1XE.flv'
127 actual_result = yt.get_video_filename()
128 assert_equal(expected_result, actual_result)
129 end
130
131 end