3 # Allow direct execution
7 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
9 from test.helper import try_rm
12 from youtube_dl import YoutubeDL
15 def _download_restricted(url, filename, age):
16 """ Returns true if the file has been downloaded """
20 'skip_download': True,
21 'writeinfojson': True,
22 "outtmpl": "%(id)s.%(ext)s",
24 ydl = YoutubeDL(params)
25 ydl.add_default_info_extractors()
26 json_filename = os.path.splitext(filename)[0] + '.info.json'
29 res = os.path.exists(json_filename)
34 class TestAgeRestriction(unittest.TestCase):
35 def _assert_restricted(self, url, filename, age, old_age=None):
36 self.assertTrue(_download_restricted(url, filename, old_age))
37 self.assertFalse(_download_restricted(url, filename, age))
39 def test_youtube(self):
40 self._assert_restricted('07FYdnEawAQ', '07FYdnEawAQ.mp4', 10)
42 def test_youporn(self):
43 self._assert_restricted(
44 'http://www.youporn.com/watch/505835/sex-ed-is-it-safe-to-masturbate-daily/',
45 '505835.mp4', 2, old_age=25)
47 def test_pornotube(self):
48 self._assert_restricted(
49 'http://pornotube.com/c/173/m/1689755/Marilyn-Monroe-Bathing',
53 if __name__ == '__main__':