2 # Copyright Michael Orlitzky
4 # http://michael.orlitzky.com/
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.
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.
16 # http://www.fsf.org/licensing/licenses/gpl.html
20 require 'src/websites/youtube'
22 class YoutubeTest
< Test
::Unit::TestCase
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 assert(Youtube
.owns_url
?('http://www.youtube.com/meetlocalbiz#p/u/0/rJVWV4aA6Jk'))
42 def test_doesnt_own_redtube_urls
43 assert(!Youtube
.owns_url
?('http://www.redtube.com/6807'))
44 assert(!Youtube
.owns_url
?('www.redtube.com/6807'))
45 assert(!Youtube
.owns_url
?('http://redtube.com/6807'))
46 assert(!Youtube
.owns_url
?('redtube.com/6807'))
50 def test_parse_video_id
51 yt
= Youtube
.new('http://www.youtube.com/watch?v=SudixyugiX4')
52 expected_result
= 'SudixyugiX4'
53 actual_result
= yt
.send('parse_video_id')
54 assert_equal(expected_result
, actual_result
)
58 def test_parse_video_id_again
59 yt
= Youtube
.new('http://www.youtube.com/watch?v=K9iDMcmm0tE')
60 expected_result
= 'K9iDMcmm0tE'
61 actual_result
= yt
.send('parse_video_id')
62 assert_equal(expected_result
, actual_result
)
66 def test_parse_t_parameter
71 File
.open('test/fixtures/youtube/SudixyugiX4.html') do |f
|
75 expected_result
= 'OEgsToPDskLQUAntWWpzhEMhBMlgqHdo'
76 actual_result
= yt
.send('parse_t_parameter', page_data
)
77 assert_equal(expected_result
, actual_result
)
81 def test_parse_t_parameter_again
82 # This was failing once because my regex missed a hyphen.
83 # I modified the regex to match anything between the pair
84 # of quotes, so it should now catch the hyphen and underscore.
89 File
.open('test/fixtures/youtube/K9iDMcmm0tE.html') do |f
|
93 expected_result
= 'O_EgsToPDskJsXVvAXpAct1zug-lBJBz'
94 actual_result
= yt
.send('parse_t_parameter', page_data
)
95 assert_equal(expected_result
, actual_result
)
99 def test_get_available_formats
100 # Make sure that we can parse the available formats from a page.
101 yt
= Youtube
.new(nil)
105 File
.open('test/fixtures/youtube/BUer8Dv2HW8.html') do |f
|
109 actual_formats
= yt
.send('get_available_formats', page_data
)
110 expected_formats
= [ 37, 22, 35, 18, 34, 5 ]
112 # Make sure that all of the elements of the "expected" set are in
113 # the "actual" set and vice-versa.
114 assert_equal([], expected_formats
- actual_formats
)
115 assert_equal([], actual_formats
- expected_formats
)
119 def test_get_desired_format
120 # The get_desired_format() method should choose the best of the
122 yt
= Youtube
.new(nil)
126 File
.open('test/fixtures/youtube/BUer8Dv2HW8.html') do |f
|
130 available_formats
= yt
.send('get_available_formats', page_data
)
131 actual_result
= yt
.send('get_desired_format', available_formats
)
134 assert_equal(expected_result
, actual_result
)
138 def test_get_video_filename
139 yt
= Youtube
.new('http://www.youtube.com/watch?v=SudixyugiX4')
140 expected_result
= 'SudixyugiX4.flv'
141 actual_result
= yt
.get_video_filename()
142 assert_equal(expected_result
, actual_result
)
146 def test_get_video_filename_again
147 yt
= Youtube
.new('http://www.youtube.com/watch?v=K9iDMcmm0tE')
148 expected_result
= 'K9iDMcmm0tE.flv'
149 actual_result
= yt
.get_video_filename()
150 assert_equal(expected_result
, actual_result
)
154 def test_get_troublesome_video_filename_first_form
155 # This non-alphanumeric video id was causing
156 # get_video_filename to barf.
157 yt
= Youtube
.new('http://www.youtube.com/watch?v=83-hlYMH1XE&feature=dir')
158 expected_result
= '83-hlYMH1XE.flv'
159 actual_result
= yt
.get_video_filename()
160 assert_equal(expected_result
, actual_result
)
164 def test_get_troublesome_video_filename_second_form
165 yt
= Youtube
.new('http://www.youtube.com/v/83-hlYMH1XE&feature=dir')
166 expected_result
= '83-hlYMH1XE.flv'
167 actual_result
= yt
.get_video_filename()
168 assert_equal(expected_result
, actual_result
)