summaryrefslogtreecommitdiffstats
path: root/test/test_cookies.py
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-08-05 09:06:11 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-08-05 09:06:11 +0000
commitfd5a06560caab95c71a2e2e805efa8d0f3a696a0 (patch)
treee1c600b8612bc4b301e2f51b875fcd835c5008cc /test/test_cookies.py
parentReleasing progress-linux version 2024.05.27-1~progress7.99u1. (diff)
downloadyt-dlp-fd5a06560caab95c71a2e2e805efa8d0f3a696a0.tar.xz
yt-dlp-fd5a06560caab95c71a2e2e805efa8d0f3a696a0.zip
Merging upstream version 2024.07.01.
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'test/test_cookies.py')
-rw-r--r--test/test_cookies.py133
1 files changed, 67 insertions, 66 deletions
diff --git a/test/test_cookies.py b/test/test_cookies.py
index bd61f30..e1271f6 100644
--- a/test/test_cookies.py
+++ b/test/test_cookies.py
@@ -67,6 +67,7 @@ class TestCookies(unittest.TestCase):
({'XDG_CURRENT_DESKTOP': 'GNOME'}, _LinuxDesktopEnvironment.GNOME),
({'XDG_CURRENT_DESKTOP': 'GNOME:GNOME-Classic'}, _LinuxDesktopEnvironment.GNOME),
({'XDG_CURRENT_DESKTOP': 'GNOME : GNOME-Classic'}, _LinuxDesktopEnvironment.GNOME),
+ ({'XDG_CURRENT_DESKTOP': 'ubuntu:GNOME'}, _LinuxDesktopEnvironment.GNOME),
({'XDG_CURRENT_DESKTOP': 'Unity', 'DESKTOP_SESSION': 'gnome-fallback'}, _LinuxDesktopEnvironment.GNOME),
({'XDG_CURRENT_DESKTOP': 'KDE', 'KDE_SESSION_VERSION': '5'}, _LinuxDesktopEnvironment.KDE5),
@@ -106,7 +107,7 @@ class TestCookies(unittest.TestCase):
def test_chrome_cookie_decryptor_windows_v10(self):
with MonkeyPatch(cookies, {
- '_get_windows_v10_key': lambda *args, **kwargs: b'Y\xef\xad\xad\xeerp\xf0Y\xe6\x9b\x12\xc2<z\x16]\n\xbb\xb8\xcb\xd7\x9bA\xc3\x14e\x99{\xd6\xf4&'
+ '_get_windows_v10_key': lambda *args, **kwargs: b'Y\xef\xad\xad\xeerp\xf0Y\xe6\x9b\x12\xc2<z\x16]\n\xbb\xb8\xcb\xd7\x9bA\xc3\x14e\x99{\xd6\xf4&',
}):
encrypted_value = b'v10T\xb8\xf3\xb8\x01\xa7TtcV\xfc\x88\xb8\xb8\xef\x05\xb5\xfd\x18\xc90\x009\xab\xb1\x893\x85)\x87\xe1\xa9-\xa3\xad='
value = '32101439'
@@ -121,17 +122,17 @@ class TestCookies(unittest.TestCase):
self.assertEqual(decryptor.decrypt(encrypted_value), value)
def test_safari_cookie_parsing(self):
- cookies = \
- b'cook\x00\x00\x00\x01\x00\x00\x00i\x00\x00\x01\x00\x01\x00\x00\x00\x10\x00\x00\x00\x00\x00\x00\x00Y' \
- b'\x00\x00\x00\x00\x00\x00\x00 \x00\x00\x00\x00\x00\x00\x008\x00\x00\x00B\x00\x00\x00F\x00\x00\x00H' \
- b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x80\x03\xa5>\xc3A\x00\x00\x80\xc3\x07:\xc3A' \
- b'localhost\x00foo\x00/\x00test%20%3Bcookie\x00\x00\x00\x054\x07\x17 \x05\x00\x00\x00Kbplist00\xd1\x01' \
- b'\x02_\x10\x18NSHTTPCookieAcceptPolicy\x10\x02\x08\x0b&\x00\x00\x00\x00\x00\x00\x01\x01\x00\x00\x00' \
- b'\x00\x00\x00\x00\x03\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00('
+ cookies = (
+ b'cook\x00\x00\x00\x01\x00\x00\x00i\x00\x00\x01\x00\x01\x00\x00\x00\x10\x00\x00\x00\x00\x00\x00\x00Y'
+ b'\x00\x00\x00\x00\x00\x00\x00 \x00\x00\x00\x00\x00\x00\x008\x00\x00\x00B\x00\x00\x00F\x00\x00\x00H'
+ b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x80\x03\xa5>\xc3A\x00\x00\x80\xc3\x07:\xc3A'
+ b'localhost\x00foo\x00/\x00test%20%3Bcookie\x00\x00\x00\x054\x07\x17 \x05\x00\x00\x00Kbplist00\xd1\x01'
+ b'\x02_\x10\x18NSHTTPCookieAcceptPolicy\x10\x02\x08\x0b&\x00\x00\x00\x00\x00\x00\x01\x01\x00\x00\x00'
+ b'\x00\x00\x00\x00\x03\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00(')
jar = parse_safari_cookies(cookies)
self.assertEqual(len(jar), 1)
- cookie = list(jar)[0]
+ cookie = next(iter(jar))
self.assertEqual(cookie.domain, 'localhost')
self.assertEqual(cookie.port, None)
self.assertEqual(cookie.path, '/')
@@ -164,7 +165,7 @@ class TestLenientSimpleCookie(unittest.TestCase):
attributes = {
key: value
for key, value in dict(morsel).items()
- if value != ""
+ if value != ''
}
self.assertEqual(attributes, expected_attributes, message)
@@ -174,133 +175,133 @@ class TestLenientSimpleCookie(unittest.TestCase):
self._run_tests(
# Copied from https://github.com/python/cpython/blob/v3.10.7/Lib/test/test_http_cookies.py
(
- "Test basic cookie",
- "chips=ahoy; vienna=finger",
- {"chips": "ahoy", "vienna": "finger"},
+ 'Test basic cookie',
+ 'chips=ahoy; vienna=finger',
+ {'chips': 'ahoy', 'vienna': 'finger'},
),
(
- "Test quoted cookie",
+ 'Test quoted cookie',
'keebler="E=mc2; L=\\"Loves\\"; fudge=\\012;"',
- {"keebler": 'E=mc2; L="Loves"; fudge=\012;'},
+ {'keebler': 'E=mc2; L="Loves"; fudge=\012;'},
),
(
"Allow '=' in an unquoted value",
- "keebler=E=mc2",
- {"keebler": "E=mc2"},
+ 'keebler=E=mc2',
+ {'keebler': 'E=mc2'},
),
(
"Allow cookies with ':' in their name",
- "key:term=value:term",
- {"key:term": "value:term"},
+ 'key:term=value:term',
+ {'key:term': 'value:term'},
),
(
"Allow '[' and ']' in cookie values",
- "a=b; c=[; d=r; f=h",
- {"a": "b", "c": "[", "d": "r", "f": "h"},
+ 'a=b; c=[; d=r; f=h',
+ {'a': 'b', 'c': '[', 'd': 'r', 'f': 'h'},
),
(
- "Test basic cookie attributes",
+ 'Test basic cookie attributes',
'Customer="WILE_E_COYOTE"; Version=1; Path=/acme',
- {"Customer": ("WILE_E_COYOTE", {"version": "1", "path": "/acme"})},
+ {'Customer': ('WILE_E_COYOTE', {'version': '1', 'path': '/acme'})},
),
(
- "Test flag only cookie attributes",
+ 'Test flag only cookie attributes',
'Customer="WILE_E_COYOTE"; HttpOnly; Secure',
- {"Customer": ("WILE_E_COYOTE", {"httponly": True, "secure": True})},
+ {'Customer': ('WILE_E_COYOTE', {'httponly': True, 'secure': True})},
),
(
- "Test flag only attribute with values",
- "eggs=scrambled; httponly=foo; secure=bar; Path=/bacon",
- {"eggs": ("scrambled", {"httponly": "foo", "secure": "bar", "path": "/bacon"})},
+ 'Test flag only attribute with values',
+ 'eggs=scrambled; httponly=foo; secure=bar; Path=/bacon',
+ {'eggs': ('scrambled', {'httponly': 'foo', 'secure': 'bar', 'path': '/bacon'})},
),
(
"Test special case for 'expires' attribute, 4 digit year",
'Customer="W"; expires=Wed, 01 Jan 2010 00:00:00 GMT',
- {"Customer": ("W", {"expires": "Wed, 01 Jan 2010 00:00:00 GMT"})},
+ {'Customer': ('W', {'expires': 'Wed, 01 Jan 2010 00:00:00 GMT'})},
),
(
"Test special case for 'expires' attribute, 2 digit year",
'Customer="W"; expires=Wed, 01 Jan 98 00:00:00 GMT',
- {"Customer": ("W", {"expires": "Wed, 01 Jan 98 00:00:00 GMT"})},
+ {'Customer': ('W', {'expires': 'Wed, 01 Jan 98 00:00:00 GMT'})},
),
(
- "Test extra spaces in keys and values",
- "eggs = scrambled ; secure ; path = bar ; foo=foo ",
- {"eggs": ("scrambled", {"secure": True, "path": "bar"}), "foo": "foo"},
+ 'Test extra spaces in keys and values',
+ 'eggs = scrambled ; secure ; path = bar ; foo=foo ',
+ {'eggs': ('scrambled', {'secure': True, 'path': 'bar'}), 'foo': 'foo'},
),
(
- "Test quoted attributes",
+ 'Test quoted attributes',
'Customer="WILE_E_COYOTE"; Version="1"; Path="/acme"',
- {"Customer": ("WILE_E_COYOTE", {"version": "1", "path": "/acme"})}
+ {'Customer': ('WILE_E_COYOTE', {'version': '1', 'path': '/acme'})},
),
# Our own tests that CPython passes
(
"Allow ';' in quoted value",
'chips="a;hoy"; vienna=finger',
- {"chips": "a;hoy", "vienna": "finger"},
+ {'chips': 'a;hoy', 'vienna': 'finger'},
),
(
- "Keep only the last set value",
- "a=c; a=b",
- {"a": "b"},
+ 'Keep only the last set value',
+ 'a=c; a=b',
+ {'a': 'b'},
),
)
def test_lenient_parsing(self):
self._run_tests(
(
- "Ignore and try to skip invalid cookies",
+ 'Ignore and try to skip invalid cookies',
'chips={"ahoy;": 1}; vienna="finger;"',
- {"vienna": "finger;"},
+ {'vienna': 'finger;'},
),
(
- "Ignore cookies without a name",
- "a=b; unnamed; c=d",
- {"a": "b", "c": "d"},
+ 'Ignore cookies without a name',
+ 'a=b; unnamed; c=d',
+ {'a': 'b', 'c': 'd'},
),
(
"Ignore '\"' cookie without name",
'a=b; "; c=d',
- {"a": "b", "c": "d"},
+ {'a': 'b', 'c': 'd'},
),
(
- "Skip all space separated values",
- "x a=b c=d x; e=f",
- {"a": "b", "c": "d", "e": "f"},
+ 'Skip all space separated values',
+ 'x a=b c=d x; e=f',
+ {'a': 'b', 'c': 'd', 'e': 'f'},
),
(
- "Skip all space separated values",
+ 'Skip all space separated values',
'x a=b; data={"complex": "json", "with": "key=value"}; x c=d x',
- {"a": "b", "c": "d"},
+ {'a': 'b', 'c': 'd'},
),
(
- "Expect quote mending",
+ 'Expect quote mending',
'a=b; invalid="; c=d',
- {"a": "b", "c": "d"},
+ {'a': 'b', 'c': 'd'},
),
(
- "Reset morsel after invalid to not capture attributes",
- "a=b; invalid; Version=1; c=d",
- {"a": "b", "c": "d"},
+ 'Reset morsel after invalid to not capture attributes',
+ 'a=b; invalid; Version=1; c=d',
+ {'a': 'b', 'c': 'd'},
),
(
- "Reset morsel after invalid to not capture attributes",
- "a=b; $invalid; $Version=1; c=d",
- {"a": "b", "c": "d"},
+ 'Reset morsel after invalid to not capture attributes',
+ 'a=b; $invalid; $Version=1; c=d',
+ {'a': 'b', 'c': 'd'},
),
(
- "Continue after non-flag attribute without value",
- "a=b; path; Version=1; c=d",
- {"a": "b", "c": "d"},
+ 'Continue after non-flag attribute without value',
+ 'a=b; path; Version=1; c=d',
+ {'a': 'b', 'c': 'd'},
),
(
- "Allow cookie attributes with `$` prefix",
+ 'Allow cookie attributes with `$` prefix',
'Customer="WILE_E_COYOTE"; $Version=1; $Secure; $Path=/acme',
- {"Customer": ("WILE_E_COYOTE", {"version": "1", "secure": True, "path": "/acme"})},
+ {'Customer': ('WILE_E_COYOTE', {'version': '1', 'secure': True, 'path': '/acme'})},
),
(
- "Invalid Morsel keys should not result in an error",
- "Key=Value; [Invalid]=Value; Another=Value",
- {"Key": "Value", "Another": "Value"},
+ 'Invalid Morsel keys should not result in an error',
+ 'Key=Value; [Invalid]=Value; Another=Value',
+ {'Key': 'Value', 'Another': 'Value'},
),
)