Add GitHub Actions CI workflow

Author: RobuRishabh

.github/workflows/test.yml

@@ -0,0 +1,45 @@
+name: CI Tests
+
+# Trigger the workflow on pushes and pull requests to the main branch
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+    branches:
+      - main
+
+jobs:
+  test:
+    runs-on: ubuntu-latest  # Use the latest Ubuntu runner
+
+    steps:
+      # Step 1: Checkout the repository code
+      - name: Checkout code
+        uses: actions/checkout@v4
+
+      # Step 2: Set up Python environment
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.11'  # Specify your Python version
+
+      # Step 3: Install dependencies
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -r requirements.txt
+
+      # Step 4: Run unit tests
+      - name: Run tests
+        run: |
+          python -m unittest discover -s tests -p "test_*.py"
+
+      # Optional: Cache dependencies to speed up future runs
+      - name: Cache dependencies
+        uses: actions/cache@v3
+        with:
+          path: ~/.cache/pip
+          key: ${{ runner.os }}-pip-${{ hashFiles('**/requirements.txt') }}
+          restore-keys: |
+            ${{ runner.os }}-pip-

requirements.txt

@@ -31,6 +31,7 @@ pillow
 pydantic
 python-dateutil
 python-dotenv
+pytest
 PyYAML
 requests
 scikit-learn

tests/test_app.py

@@ -0,0 +1,14 @@
+import unittest
+from app import app  # Assuming Flask or similar backend
+
+class TestApp(unittest.TestCase):
+    def setUp(self):
+        self.app = app.test_client()
+        self.app.testing = True
+
+    def test_homepage_loads(self):
+        response = self.app.get("/")
+        self.assertEqual(response.status_code, 200)
+
+if __name__ == "__main__":
+    unittest.main()

tests/test_mock_databse.py

@@ -0,0 +1,14 @@
+import unittest
+from mock_database import flights
+
+class TestMockDatabase(unittest.TestCase):
+    def test_flight_data_exists(self):
+        self.assertGreater(len(flights), 0, "Flight database should not be empty")
+    
+    def test_flight_structure(self):
+        flight = flights[0]
+        expected_keys = {"flight_number", "origin", "destination", "time"}
+        self.assertEqual(set(flight.keys()), expected_keys, "Flight data has incorrect structure")
+
+if __name__ == "__main__":
+    unittest.main()

tests/test_ollama_api.py

@@ -0,0 +1,24 @@
+import unittest
+from unittest.mock import patch
+from ollama_api import call_ollama_api
+
+class TestOllamaAPI(unittest.TestCase):
+    @patch("ollama_api.requests.post")
+    def test_call_ollama_api_success(self, mock_post):
+        mock_post.return_value.status_code = 200
+        mock_post.return_value.json.return_value = {"response": "Flight NY100 departs at 08:00."}
+        query = "What are flights from New York to London?"
+        flight_data = [{"flight_number": "NY100", "origin": "New York", "destination": "London", "time": "2025-05-01 08:00"}]
+        response = call_ollama_api(query, flight_data)
+        self.assertEqual(response, "Flight NY100 departs at 08:00.")
+
+    @patch("ollama_api.requests.post")
+    def test_call_ollama_api_failure(self, mock_post):
+        mock_post.return_value.status_code = 500
+        query = "What are flights from New York to London?"
+        flight_data = [{"flight_number": "NY100", "origin": "New York", "destination": "London", "time": "2025-05-01 08:00"}]
+        with self.assertRaises(Exception):
+            call_ollama_api(query, flight_data)
+
+if __name__ == "__main__":
+    unittest.main()

tests/test_query_handler.py

@@ -0,0 +1,23 @@
+import unittest
+from query_handler import parse_query, search_flights
+
+class TestQueryHandler(unittest.TestCase):
+    def test_parse_query(self):
+        query = "What flights are from New York to London?"
+        result = parse_query(query)
+        self.assertEqual(result["origin"], "New York")
+        self.assertEqual(result["destination"], "London")
+
+    def test_search_flights(self):
+        query = {"origin": "New York", "destination": "London"}
+        results = search_flights(query)
+        self.assertGreater(len(results), 0, "Should find at least one matching flight")
+        self.assertIn("NY100", [f["flight_number"] for f in results])
+
+    def test_no_results(self):
+        query = {"origin": "Mars", "destination": "Jupiter"}
+        results = search_flights(query)
+        self.assertEqual(len(results), 0, "Should return no results for invalid query")
+
+if __name__ == "__main__":
+    unittest.main()