from fastapi.testclient import TestClient
from api.main import app
import pytest

client = TestClient(app)

def test_get_status():
    response = client.get("/status")
    assert response.status_code == 200
    assert response.json()["status"] == "operational"

def test_get_current_price_no_data(mocker):
    # Mock database connection to return no data
    mock_db = mocker.patch("utils.database.get_connection")
    mock_con = mock_db.return_value.__enter__.return_value
    mock_con.execute.return_value.fetchone.return_value = None
    
    response = client.get("/current-price")
    assert response.status_code == 404
    assert response.json()["detail"] == "No data available"