google_play_scraper/tests/test_http_utils.py

"""Tests for http_utils.py module."""
import pytest
from google_play_scraper.utils.http_utils import get
from google_play_scraper.exceptions import RateLimitError
from unittest.mock import patch, MagicMock
import requests


@patch('google_play_scraper.utils.http_utils.session')
def test_get_success(mock_session):
    mock_response = MagicMock()
    mock_response.json.return_value = {"data": "test"}
    mock_response.raise_for_status = MagicMock()
    mock_session.get.return_value = mock_response
    
    result = get("https://example.com")
    assert result == {"data": "test"}


@patch('google_play_scraper.utils.http_utils.session')
def test_get_rate_limit(mock_session):
    mock_response = MagicMock()
    mock_response.status_code = 429
    mock_session.get.return_value = mock_response
    mock_session.get.side_effect = requests.exceptions.HTTPError()
    
    with pytest.raises(RateLimitError):
        get("https://example.com")