|
1 | | -from http.cookiejar import Cookie |
| 1 | +from http.cookiejar import Cookie, CookieJar |
2 | 2 | from unittest.mock import MagicMock, patch |
3 | 3 |
|
4 | 4 | import pytest |
5 | | -from yt_dlp.cookies import SUPPORTED_BROWSERS |
6 | | -from yt_dlp.utils import YoutubeDLCookieJar |
7 | 5 |
|
8 | 6 | from revChatGPTAuth.openai_cookie_parser import OpenAICookieParser |
| 7 | +from revChatGPTAuth.supported_browser import SupportedBrowser |
9 | 8 |
|
10 | 9 |
|
11 | 10 | class TestOpenAICookieParser: |
12 | | - |
13 | | - def test_init_with_unsupported_browser(self): |
14 | | - with pytest.raises(ValueError): |
15 | | - OpenAICookieParser(browser_name='Internet Explorer Holy Shit') |
| 11 | + SUPPORTED_BROWSERS = [SupportedBrowser(browser_name) for browser_name in OpenAICookieParser.SUPPORTED_BROWSERS] |
16 | 12 |
|
17 | 13 | @pytest.mark.parametrize('supported_browser', SUPPORTED_BROWSERS) |
18 | | - def test_init_with_supported_browser(self, supported_browser: str): |
| 14 | + def test_init_with_supported_browser(self, supported_browser: SupportedBrowser): |
19 | 15 | parser = OpenAICookieParser(browser_name=supported_browser) |
20 | 16 | assert parser.BROWSER_NAME == supported_browser |
21 | 17 |
|
22 | 18 | @pytest.mark.parametrize('supported_browser', SUPPORTED_BROWSERS) |
23 | | - @patch('revChatGPTAuth.openai_cookie_parser.extract_cookies_from_browser') |
| 19 | + @patch('revChatGPTAuth.openai_cookie_parser.load_cookies') |
24 | 20 | def test_parse_cookie_with_supported_browser( |
25 | 21 | self, |
26 | | - mock_extract_cookies: MagicMock, |
27 | | - supported_browser: str, |
| 22 | + mock_load_cookies: MagicMock, |
| 23 | + supported_browser: SupportedBrowser, |
28 | 24 | ): |
29 | 25 | # given |
30 | | - cookie_jar = YoutubeDLCookieJar() |
| 26 | + cookie_jar = CookieJar() |
31 | 27 | cookie_jar.set_cookie( |
32 | 28 | Cookie( |
33 | 29 | version=0, |
@@ -85,7 +81,7 @@ def test_parse_cookie_with_supported_browser( |
85 | 81 | comment_url=None, |
86 | 82 | rest={}, |
87 | 83 | )) |
88 | | - mock_extract_cookies.return_value = cookie_jar |
| 84 | + mock_load_cookies.return_value = cookie_jar |
89 | 85 | parser = OpenAICookieParser(browser_name=supported_browser) |
90 | 86 |
|
91 | 87 | # when |
|
0 commit comments