]> Raphaƫl G. Git Repositories - youtubedl/blob - test/test_YoutubeDLCookieJar.py
Update README.md
[youtubedl] / test / test_YoutubeDLCookieJar.py
1 #!/usr/bin/env python
2 # coding: utf-8
3
4 from __future__ import unicode_literals
5
6 import os
7 import re
8 import sys
9 import tempfile
10 import unittest
11 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
12
13 from youtube_dl.utils import YoutubeDLCookieJar
14
15
16 class TestYoutubeDLCookieJar(unittest.TestCase):
17 def test_keep_session_cookies(self):
18 cookiejar = YoutubeDLCookieJar('./test/testdata/cookies/session_cookies.txt')
19 cookiejar.load(ignore_discard=True, ignore_expires=True)
20 tf = tempfile.NamedTemporaryFile(delete=False)
21 try:
22 cookiejar.save(filename=tf.name, ignore_discard=True, ignore_expires=True)
23 temp = tf.read().decode('utf-8')
24 self.assertTrue(re.search(
25 r'www\.foobar\.foobar\s+FALSE\s+/\s+TRUE\s+0\s+YoutubeDLExpiresEmpty\s+YoutubeDLExpiresEmptyValue', temp))
26 self.assertTrue(re.search(
27 r'www\.foobar\.foobar\s+FALSE\s+/\s+TRUE\s+0\s+YoutubeDLExpires0\s+YoutubeDLExpires0Value', temp))
28 finally:
29 tf.close()
30 os.remove(tf.name)
31
32 def test_strip_httponly_prefix(self):
33 cookiejar = YoutubeDLCookieJar('./test/testdata/cookies/httponly_cookies.txt')
34 cookiejar.load(ignore_discard=True, ignore_expires=True)
35
36 def assert_cookie_has_value(key):
37 self.assertEqual(cookiejar._cookies['www.foobar.foobar']['/'][key].value, key + '_VALUE')
38
39 assert_cookie_has_value('HTTPONLY_COOKIE')
40 assert_cookie_has_value('JS_ACCESSIBLE_COOKIE')
41
42 def test_malformed_cookies(self):
43 cookiejar = YoutubeDLCookieJar('./test/testdata/cookies/malformed_cookies.txt')
44 cookiejar.load(ignore_discard=True, ignore_expires=True)
45 # Cookies should be empty since all malformed cookie file entries
46 # will be ignored
47 self.assertFalse(cookiejar._cookies)
48
49
50 if __name__ == '__main__':
51 unittest.main()