mirror of
https://github.com/bohd4nx/FragmentAPI.git
synced 2026-07-25 14:24:31 +00:00
bohd4nx
dc9661134d
- Removed the old CookieResult model from pyfragment.core.models and created a new one in pyfragment.services.cookies.models. - Implemented get_cookies_from_browser function in pyfragment.services.cookies.service to extract cookies directly from supported browsers. - Updated imports across various modules to reflect the new structure and removed unused imports. - Cleaned up models in ads, anonymous_numbers, giveaways, marketplace, and purchases domains by removing unnecessary __all__ declarations. - Added new example scripts for cookie extraction and various purchase functionalities. - Consolidated constants into a new pyfragment.core.constants module for better organization.
132 lines
4.1 KiB
Python
132 lines
4.1 KiB
Python
"""Extract Fragment cookies from browser stores and validate required keys."""
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from pyfragment import CookieError, get_cookies_from_browser
|
|
from pyfragment.core.constants import REQUIRED_COOKIE_KEYS
|
|
|
|
FAKE_JAR = [
|
|
{"name": "stel_ssid", "value": "abc123", "domain": "fragment.com", "expires": "2027-04-03T20:52:16.375Z"},
|
|
{"name": "stel_dt", "value": "-120", "domain": "fragment.com"},
|
|
{"name": "stel_token", "value": "tok_xyz", "domain": "fragment.com"},
|
|
{"name": "stel_ton_token", "value": "ton_xyz", "domain": "fragment.com"},
|
|
{"name": "unrelated", "value": "noise", "domain": "fragment.com"},
|
|
]
|
|
|
|
|
|
def _mock_rookiepy(jar: list[dict[str, str]] | None = None) -> MagicMock:
|
|
mock = MagicMock()
|
|
mock.chrome.return_value = jar if jar is not None else FAKE_JAR
|
|
return mock
|
|
|
|
|
|
PATCH = "pyfragment.services.cookies.service.rookiepy"
|
|
|
|
|
|
# unsupported browser tests
|
|
|
|
|
|
def test_unsupported_browser_raises() -> None:
|
|
with pytest.raises(CookieError, match="Unsupported browser"):
|
|
get_cookies_from_browser("internet_explorer")
|
|
|
|
|
|
# successful extraction tests
|
|
|
|
|
|
def test_returns_required_keys_only() -> None:
|
|
with patch(PATCH, _mock_rookiepy()):
|
|
result = get_cookies_from_browser("chrome")
|
|
|
|
assert set(result.cookies.keys()) == set(REQUIRED_COOKIE_KEYS)
|
|
assert result.cookies["stel_ssid"] == "abc123"
|
|
assert result.cookies["stel_dt"] == "-120"
|
|
assert result.cookies["stel_token"] == "tok_xyz"
|
|
assert result.cookies["stel_ton_token"] == "ton_xyz"
|
|
assert "unrelated" not in result.cookies
|
|
|
|
|
|
def test_returns_expiry() -> None:
|
|
with patch(PATCH, _mock_rookiepy()):
|
|
result = get_cookies_from_browser("chrome")
|
|
|
|
assert result.expires == "2027-04-03T20:52:16.375000+00:00"
|
|
|
|
|
|
def test_returns_none_expiry_when_missing() -> None:
|
|
jar = [{"name": k, "value": "v"} for k in REQUIRED_COOKIE_KEYS]
|
|
with patch(PATCH, _mock_rookiepy(jar)):
|
|
result = get_cookies_from_browser("chrome")
|
|
|
|
assert result.expires is None
|
|
|
|
|
|
def test_default_browser_is_chrome() -> None:
|
|
mock_rp = _mock_rookiepy()
|
|
with patch(PATCH, mock_rp):
|
|
get_cookies_from_browser()
|
|
|
|
mock_rp.chrome.assert_called_once_with(["fragment.com"])
|
|
|
|
|
|
def test_browser_name_is_case_insensitive() -> None:
|
|
mock_rp = _mock_rookiepy()
|
|
with patch(PATCH, mock_rp):
|
|
result = get_cookies_from_browser("Chrome")
|
|
|
|
assert result.cookies["stel_ssid"] == "abc123"
|
|
|
|
|
|
# missing cookies tests
|
|
|
|
|
|
def test_missing_cookies_raises() -> None:
|
|
partial_jar = [
|
|
{"name": "stel_ssid", "value": "abc123"},
|
|
{"name": "stel_dt", "value": "-120"},
|
|
# stel_token and stel_ton_token missing
|
|
]
|
|
with patch(PATCH, _mock_rookiepy(partial_jar)):
|
|
with pytest.raises(CookieError, match="Fragment cookies not found in chrome"):
|
|
get_cookies_from_browser("chrome")
|
|
|
|
|
|
def test_empty_cookie_value_treated_as_missing() -> None:
|
|
jar_with_empty = [
|
|
{"name": "stel_ssid", "value": "abc123"},
|
|
{"name": "stel_dt", "value": ""}, # empty — should be treated as missing
|
|
{"name": "stel_token", "value": "tok_xyz"},
|
|
{"name": "stel_ton_token", "value": "ton_xyz"},
|
|
]
|
|
with patch(PATCH, _mock_rookiepy(jar_with_empty)):
|
|
with pytest.raises(CookieError, match="Fragment cookies not found in chrome"):
|
|
get_cookies_from_browser("chrome")
|
|
|
|
|
|
# expired cookie tests
|
|
|
|
|
|
def test_expired_cookie_raises() -> None:
|
|
expired_jar = [
|
|
{"name": "stel_ssid", "value": "abc123", "expires": "2020-01-01T00:00:00.000Z"},
|
|
{"name": "stel_dt", "value": "-120"},
|
|
{"name": "stel_token", "value": "tok_xyz"},
|
|
{"name": "stel_ton_token", "value": "ton_xyz"},
|
|
]
|
|
with patch(PATCH, _mock_rookiepy(expired_jar)):
|
|
with pytest.raises(CookieError, match="expired"):
|
|
get_cookies_from_browser("chrome")
|
|
|
|
|
|
# read failure tests
|
|
|
|
|
|
def test_browser_read_error_raises() -> None:
|
|
mock_rp = MagicMock()
|
|
mock_rp.chrome.side_effect = PermissionError("locked")
|
|
with patch(PATCH, mock_rp):
|
|
with pytest.raises(CookieError, match="Failed to read chrome cookies"):
|
|
get_cookies_from_browser("chrome")
|