import pytest
from unittest.mock import patch
import httpx
import client
@pytest.fixture
def mock_env_vars(monkeypatch):
monkeypatch.setenv("IOL_USER", "test_user")
monkeypatch.setenv("IOL_PASS", "test_pass")
@pytest.fixture
def mock_token_response():
return {"access_token": "test_token", "token_type": "bearer", "expires_in": 3600}
@pytest.fixture
def mock_profile_response():
return {
"nombre": "Test",
"apellido": "User",
"numeroCuenta": "123456",
"dni": "12345678",
"cuitCuil": "20123456789",
"perfilInversor": "Agresivo",
"email": "test@example.com",
}
@pytest.fixture
def mock_portfolio_response():
return {
"pais": "Argentina",
"activos": [
{
"titulo": {
"simbolo": "GGAL",
"tipo": "ACCIONES",
"descripcion": "GRUPO FINANCIERO GALICIA",
},
"cantidad": 100,
"ultimoPrecio": 250.5,
"variacionDiaria": 1.5,
"puntosVariacion": 2.0,
"gananciaPorcentaje": 5.0,
"gananciaDinero": 1000.0,
"comprometido": 0,
"ppc": 200.0,
"valorizado": 25050.0,
}
],
}
@pytest.fixture
def mock_account_status_response():
return {
"cuentas": [
{
"numero": 123,
"tipo": "Inversion_Argentina_Pesos",
"moneda": "Peso_Argentino",
"disponible": 1000.0,
"comprometido": 0.0,
"saldo": 1000.0,
"titulosValorizados": 5000.0,
"total": 6000.0,
"margenDescubierto": 0.0,
"estado": "Operable",
}
],
"estadisticas": [
{"descripcion": "Operaciones del mes", "cantidad": 5, "volumen": 10000.0}
],
"totalEnPesos": 6000.0,
}
@pytest.fixture
def mock_quote_response():
return {
"simbolo": "GGAL",
"puntas": {},
"ultimoPrecio": 755.55,
"variacionPorcentual": 1.25,
"apertura": 750.0,
"maximo": 760.0,
"minimo": 745.0,
"fechaHora": "2024-01-25T17:00:00",
"tendencia": "Sube",
"cierreAnterior": 746.2,
"montoOperado": 150000000.0,
"volumenNominal": 200000,
"cantidadOperaciones": 2500,
"mercado": "BCBA",
"moneda": "ARS",
}
@pytest.fixture
def mock_historical_data_response():
return [
{"fecha": "2025-05-02", "apertura": 100.0, "cierre": 102.0, "volumen": 1000},
{"fecha": "2025-05-05", "apertura": 102.5, "cierre": 101.0, "volumen": 1200},
{"fecha": "2025-05-06", "apertura": 101.0, "cierre": 103.0, "volumen": 1100},
{"fecha": "2025-05-07", "apertura": 103.0, "cierre": 102.5, "volumen": 1050},
]
def test_get_auth_token(mock_env_vars, mock_token_response):
with patch("httpx.post") as mock_post:
mock_post.return_value.json.return_value = mock_token_response
mock_post.return_value.raise_for_status = lambda: None
token = client.get_auth_token()
assert token == "test_token"
mock_post.assert_called_once()
def test_get_profile_data(mock_env_vars, mock_token_response, mock_profile_response):
with patch("httpx.post") as mock_post, patch("httpx.get") as mock_get:
mock_post.return_value.json.return_value = mock_token_response
mock_post.return_value.raise_for_status = lambda: None
mock_get.return_value.json.return_value = mock_profile_response
mock_get.return_value.raise_for_status = lambda: None
profile = client.get_profile_data()
assert profile["nombre"] == "Test"
assert profile["apellido"] == "User"
mock_get.assert_called_once()
def test_get_user_portfolio(
mock_env_vars, mock_token_response, mock_portfolio_response
):
with patch("httpx.post") as mock_post, patch("httpx.get") as mock_get:
mock_post.return_value.json.return_value = mock_token_response
mock_post.return_value.raise_for_status = lambda: None
mock_get.return_value.json.return_value = mock_portfolio_response
mock_get.return_value.raise_for_status = lambda: None
portfolio = client.get_user_portfolio()
assert portfolio["pais"] == "Argentina"
assert len(portfolio["activos"]) == 1
assert portfolio["activos"][0]["titulo"]["simbolo"] == "GGAL"
mock_get.assert_called_once()
def test_get_operation_details(mock_env_vars, mock_token_response):
mock_operation = {
"numero": 123456,
"tipo": "Compra",
"fecha": "2024-01-01",
"estado": "Terminada",
"simbolo": "GGAL",
"cantidad": 100,
"precio": 250.5,
"monto": 25050.0,
}
with patch("httpx.post") as mock_post, patch("httpx.get") as mock_get:
mock_post.return_value.json.return_value = mock_token_response
mock_post.return_value.raise_for_status = lambda: None
mock_get.return_value.json.return_value = mock_operation
mock_get.return_value.raise_for_status = lambda: None
operation = client.get_operation_details(123456)
assert operation["numero"] == 123456
assert operation["tipo"] == "Compra"
mock_get.assert_called_once()
def test_api_error_handling(mock_env_vars, mock_token_response):
with patch("httpx.post") as mock_post, patch("httpx.get") as mock_get:
mock_post.return_value.json.return_value = mock_token_response
mock_post.return_value.raise_for_status = lambda: None
mock_get.side_effect = httpx.HTTPError("API Error")
with pytest.raises(httpx.HTTPError):
client.get_profile_data()
def test_get_account_status(
mock_env_vars, mock_token_response, mock_account_status_response
):
with patch("httpx.post") as mock_post, patch("httpx.get") as mock_get:
mock_post.return_value.json.return_value = mock_token_response
mock_post.return_value.raise_for_status = lambda: None
mock_get.return_value.json.return_value = mock_account_status_response
mock_get.return_value.raise_for_status = lambda: None
status = client.get_account_status()
assert len(status["cuentas"]) == 1
assert status["cuentas"][0]["numero"] == 123
assert status["cuentas"][0]["tipo"] == "Inversion_Argentina_Pesos"
assert status["totalEnPesos"] == 6000.0
mock_get.assert_called_once()
def test_get_stock_quote(mock_env_vars, mock_token_response, mock_quote_response):
with patch("httpx.post") as mock_post, patch("httpx.get") as mock_get:
mock_post.return_value.json.return_value = mock_token_response
mock_post.return_value.raise_for_status = lambda: None
mock_get.return_value.json.return_value = mock_quote_response
mock_get.return_value.raise_for_status = lambda: None
quote = client.get_stock_quote("GGAL")
assert quote["simbolo"] == "GGAL"
assert quote["ultimoPrecio"] == 755.55
assert quote["mercado"] == "BCBA"
mock_get.assert_called_once()
def test_get_historical_data(mock_env_vars, mock_token_response, mock_historical_data_response):
with patch("httpx.post") as mock_post, patch("httpx.get") as mock_get:
mock_post.return_value.json.return_value = mock_token_response
mock_post.return_value.raise_for_status = lambda: None
mock_get.return_value.json.return_value = mock_historical_data_response
mock_get.return_value.raise_for_status = lambda: None
start_date = "2025-05-02"
end_date = "2025-05-09"
historical_data = client.get_historical_data(
symbol="AL30", start_date=start_date, end_date=end_date
)
assert len(historical_data) == 4
mock_get.assert_called_once()
