| """ |
| Comprehensive Test Suite for Drug Repurposing API |
| Tests both with pytest and standalone requests |
| """ |
| import pytest |
| import requests |
| import json |
| import time |
| from fastapi.testclient import TestClient |
| from app.main import app |
|
|
| client = TestClient(app) |
|
|
| |
| |
| |
|
|
| class TestHealthCheck: |
| """Test health check endpoint""" |
| |
| def test_health_check(self): |
| """Test health endpoint returns 200""" |
| response = client.get("/health") |
| assert response.status_code == 200 |
| data = response.json() |
| assert data["status"] == "healthy" |
| assert "version" in data |
| assert "service" in data |
|
|
|
|
| class TestRootEndpoint: |
| """Test root endpoint""" |
| |
| def test_root_endpoint(self): |
| """Test root endpoint returns API info""" |
| response = client.get("/") |
| assert response.status_code == 200 |
| data = response.json() |
| assert "name" in data |
| assert "version" in data |
| assert "docs" in data |
|
|
|
|
| class TestModelStatus: |
| """Test model status endpoint""" |
| |
| def test_model_status(self): |
| """Test model status endpoint""" |
| response = client.get("/api/v1/model-status") |
| assert response.status_code == 200 |
| data = response.json() |
| assert "model" in data |
| assert "device" in data |
| assert "gpu_available" in data |
|
|
|
|
| class TestDiseaseTargets: |
| """Test disease target endpoint""" |
| |
| def test_disease_targets_valid_disease(self): |
| """Test with valid disease name""" |
| payload = { |
| "disease_name": "Type 2 Diabetes", |
| "top_n": 5 |
| } |
| response = client.post("/api/v1/disease-targets", json=payload) |
| |
| assert response.status_code in [200, 404, 500] |
| """Test with invalid disease name""" |
| payload = { |
| "disease_name": "NonexistentDisease12345XYZ", |
| "top_n": 5 |
| } |
| response = client.post("/api/v1/disease-targets", json=payload) |
| assert response.status_code in [404, 500] |
|
|
|
|
| class TestProteinSequences: |
| """Test protein sequence endpoint""" |
| |
| def test_protein_sequences_empty_list(self): |
| """Test with empty target list""" |
| response = client.post("/api/v1/protein-sequences", json=[]) |
| assert response.status_code == 200 |
| data = response.json() |
| assert data["total_requested"] == 0 |
|
|
|
|
| class TestDrugLibrary: |
| """Test drug library endpoint""" |
| |
| def test_drug_library(self): |
| """Test drug library endpoint""" |
| response = client.get("/api/v1/drug-library") |
| assert response.status_code == 200 |
| data = response.json() |
| assert "total_drugs" in data |
| assert "drugs" in data |
| assert isinstance(data["drugs"], list) |
|
|
|
|
| class TestRequestValidation: |
| """Test request validation""" |
| |
| def test_invalid_top_n_too_large(self): |
| """Test validation of top_n parameter""" |
| payload = { |
| "disease_name": "Type 2 Diabetes", |
| "top_n": 101 |
| } |
| response = client.post("/api/v1/disease-targets", json=payload) |
| |
| |
| assert response.status_code in [200, 422] |
| |
| def test_missing_required_field(self): |
| """Test missing required field""" |
| payload = { |
| "top_n": 10 |
| |
| } |
| response = client.post("/api/v1/disease-targets", json=payload) |
| assert response.status_code == 422 |
|
|
|
|
| class TestScreeningPipeline: |
| """Test complete screening pipeline""" |
| |
| def test_screening_with_mock_data(self): |
| """Test screening endpoint with mock data""" |
| payload = { |
| "disease_name": "Type 2 Diabetes", |
| "min_score": 0.5, |
| "top_n_targets": 3, |
| "known_drugs": ["Metformin"] |
| } |
| response = client.post("/api/v1/screen", json=payload) |
| |
| assert response.status_code in [200, 404, 500] |
| |
| if response.status_code == 200: |
| data = response.json() |
| assert "disease_name" in data |
| assert "total_targets_found" in data |
| assert "total_drugs_screened" in data |
| assert "top_candidates" in data |
| assert "warnings" in data |
|
|
|
|
| if __name__ == "__main__": |
| |
| |
| pytest.main([__file__, "-v"]) |
| pytest.main([__file__, "-v"]) |
|
|
