Spaces:
Running
Running
| """Unit tests for ticker_validator module.""" | |
| import sys | |
| import os | |
| import unittest | |
| from unittest.mock import patch | |
| try: | |
| import pytest | |
| _slow = pytest.mark.slow | |
| except ImportError: | |
| # pytest not installed – define a no-op decorator so the file loads cleanly | |
| def _slow(cls): | |
| return cls | |
| sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..")) | |
| from ticker_validator import validate_ticker, validate_ticker_format, TickerValidationResult | |
| # --------------------------------------------------------------------------- | |
| # Format-only tests (no network, no DB) | |
| # --------------------------------------------------------------------------- | |
| class TestFormatValidation(unittest.TestCase): | |
| def test_invalid_format_numbers(self): | |
| result = validate_ticker_format("123ABC") | |
| self.assertFalse(result.valid) | |
| self.assertIn("not a valid ticker format", result.error) | |
| def test_invalid_format_too_long(self): | |
| result = validate_ticker_format("ABCDEF") | |
| self.assertFalse(result.valid) | |
| self.assertIn("not a valid ticker format", result.error) | |
| def test_invalid_format_empty(self): | |
| for value in ("", " "): | |
| result = validate_ticker_format(value) | |
| self.assertFalse(result.valid) | |
| self.assertIn("Please enter", result.error) | |
| def test_reserved_word(self): | |
| for word in ("TEST", "NULL"): | |
| result = validate_ticker_format(word) | |
| self.assertFalse(result.valid) | |
| self.assertIn("reserved word", result.error) | |
| def test_ticker_normalization(self): | |
| """Lowercase and padded tickers should normalise cleanly.""" | |
| for raw in ("aapl", " AAPL ", "Aapl"): | |
| result = validate_ticker_format(raw) | |
| self.assertTrue(result.valid) | |
| self.assertEqual(result.ticker, "AAPL") | |
| # --------------------------------------------------------------------------- | |
| # Full-stack tests (hit yfinance – mark slow for CI skipping) | |
| # --------------------------------------------------------------------------- | |
| class TestFullValidation(unittest.TestCase): | |
| def test_valid_ticker_aapl(self): | |
| result = validate_ticker("AAPL") | |
| self.assertTrue(result.valid) | |
| self.assertIn("Apple", result.company_name) | |
| self.assertIn(result.source, ("api", "local_db", "cache")) | |
| def test_nonexistent_ticker(self): | |
| result = validate_ticker("XYZQW") | |
| self.assertFalse(result.valid) | |
| self.assertNotEqual(result.error, "") | |
| def test_result_has_suggestions(self): | |
| """A close-but-wrong ticker should surface suggestions.""" | |
| result = validate_ticker("AAPLL") | |
| # Either invalid with suggestions or (edge case) valid – just check structure | |
| self.assertIsInstance(result.suggestions, list) | |
| if not result.valid: | |
| self.assertTrue(len(result.suggestions) > 0 or result.error != "") | |
| class TestLocalDbFastPath(unittest.TestCase): | |
| def test_known_ticker_skips_yfinance_lookup(self): | |
| with patch("ticker_validator.is_known_ticker", return_value=True), \ | |
| patch("ticker_validator.get_company_name", return_value="Apple Inc."), \ | |
| patch("ticker_validator.yf.Ticker") as mock_ticker: | |
| result = validate_ticker("AAPL") | |
| self.assertTrue(result.valid) | |
| self.assertEqual(result.company_name, "Apple Inc.") | |
| self.assertEqual(result.source, "local_db") | |
| self.assertFalse(result.warning) | |
| mock_ticker.assert_not_called() | |
| if __name__ == "__main__": | |
| unittest.main() | |