Spaces:
Running
Running
File size: 4,021 Bytes
95c7e7f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 | """Integration tests for the Flask API validation endpoints.
Run with: python -m unittest tests.test_api -v
Slow tests (live yfinance) are marked; skip them with:
python -m unittest tests.test_api.TestAPIFast -v
"""
import sys
import os
import json
import unittest
sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
# Import the Flask app (IRIS_System init may fail in CI — that's fine)
from app import app as flask_app
class TestAPIFast(unittest.TestCase):
"""Tests that don't require live network access."""
def setUp(self):
flask_app.config["TESTING"] = True
self.client = flask_app.test_client()
def test_health_endpoint(self):
"""GET /api/health should return 200 with ticker_db_loaded=true."""
resp = self.client.get("/api/health")
self.assertEqual(resp.status_code, 200)
data = json.loads(resp.data)
self.assertIn("status", data)
self.assertEqual(data["status"], "ok")
self.assertTrue(data.get("ticker_db_loaded"), "Ticker DB should be loaded")
self.assertGreater(data.get("ticker_count", 0), 0)
def test_validate_missing_body(self):
"""POST /api/validate-ticker with no body should return valid=false."""
resp = self.client.post("/api/validate-ticker",
content_type="application/json",
data=json.dumps({}))
self.assertEqual(resp.status_code, 200)
data = json.loads(resp.data)
self.assertFalse(data.get("valid"))
def test_validate_invalid_format(self):
"""POST /api/validate-ticker with bad format should return valid=false immediately."""
resp = self.client.post("/api/validate-ticker",
content_type="application/json",
data=json.dumps({"ticker": "123ABC"}))
self.assertEqual(resp.status_code, 200)
data = json.loads(resp.data)
self.assertFalse(data.get("valid"))
self.assertIn("error", data)
def test_analyze_rejects_invalid_ticker(self):
"""GET /api/analyze with a clearly invalid ticker should return 422."""
resp = self.client.get("/api/analyze?ticker=XYZQW")
self.assertEqual(resp.status_code, 422)
data = json.loads(resp.data)
self.assertFalse(data.get("valid"))
self.assertIn("error", data)
class TestAPISlow(unittest.TestCase):
"""Tests that hit live yfinance — skip in CI with -k 'not slow'."""
def setUp(self):
flask_app.config["TESTING"] = True
self.client = flask_app.test_client()
def test_validate_valid_ticker(self):
"""POST /api/validate-ticker with AAPL should return valid=true."""
resp = self.client.post("/api/validate-ticker",
content_type="application/json",
data=json.dumps({"ticker": "AAPL"}))
self.assertEqual(resp.status_code, 200)
data = json.loads(resp.data)
self.assertTrue(data.get("valid"))
self.assertEqual(data.get("ticker"), "AAPL")
self.assertIn("company_name", data)
def test_validate_invalid_ticker(self):
"""POST /api/validate-ticker with XYZQW should return valid=false with error."""
resp = self.client.post("/api/validate-ticker",
content_type="application/json",
data=json.dumps({"ticker": "XYZQW"}))
self.assertEqual(resp.status_code, 200)
data = json.loads(resp.data)
self.assertFalse(data.get("valid"))
self.assertIn("error", data)
def test_analyze_accepts_valid_ticker(self):
"""GET /api/analyze with AAPL should NOT return 422 (validation must pass)."""
resp = self.client.get("/api/analyze?ticker=AAPL")
self.assertNotEqual(resp.status_code, 422,
"Validation gate should not reject AAPL")
if __name__ == "__main__":
unittest.main()
|