62 lines
2.2 KiB
Python
62 lines
2.2 KiB
Python
import pytest
|
|
from unittest.mock import MagicMock, patch
|
|
from fastapi.testclient import TestClient
|
|
from main import app
|
|
from src.services.postgresql_manager import PostgreSQLManager
|
|
from src.models.api_token import APIToken
|
|
from src.models.config import Configuration
|
|
from datetime import datetime, timedelta
|
|
|
|
# client = TestClient(app) # REMOVED
|
|
|
|
# Helper to verify standard API response structure
|
|
def assert_success_response(response):
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
# auth-status returns model directly, not wrapped in {status: success}
|
|
return data
|
|
|
|
def test_get_auth_status(client):
|
|
"""Test GET /api/setup/auth-status endpoint."""
|
|
# Setup mocks in DB
|
|
# We can rely on 'mock_db_session' fixture if it's auto-used or if we patch get_db.
|
|
# But integration tests usually use the real app/client which uses override_get_db in conftest.
|
|
# Assuming conftest.py sets up `override_get_db` or uses a test DB.
|
|
# Ideally for functional test we want to mock the DB data.
|
|
|
|
# If the app uses dependency injection for `get_db`, checking conftest.py helps.
|
|
# Let's assume we can rely on TestClient and if it hits DB, it hits the test DB.
|
|
|
|
response = client.get("/api/setup/auth-status")
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert "garmin" in data
|
|
assert "fitbit" in data
|
|
assert "token_stored" in data["garmin"]
|
|
|
|
@patch('requests.get')
|
|
def test_load_consul_config(mock_get, client):
|
|
"""Test POST /api/setup/load-consul-config."""
|
|
# Mock Consul response
|
|
mock_response = MagicMock()
|
|
mock_response.status_code = 200
|
|
mock_response.json.return_value = [
|
|
# Consul returns list of {Key: ..., Value: b64}
|
|
{
|
|
"Key": "fitbit-garmin-sync/garmin_username",
|
|
"Value": "dGVzdF91c2Vy" # 'test_user' in b64
|
|
},
|
|
{
|
|
"Key": "fitbit-garmin-sync/garmin_password",
|
|
"Value": "dGVzdF9wYXNz" # 'test_pass' in b64
|
|
}
|
|
]
|
|
mock_get.return_value = mock_response
|
|
|
|
response = client.post("/api/setup/load-consul-config")
|
|
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data["status"] == "success"
|
|
assert data["garmin"]["username"] == "test_user"
|