Merge pull request #15 from lzwjava/feat/integration-tests

Feat: Add Integration Tests

Author: lzwjava

.github/workflows/test.yaml

@@ -39,3 +39,11 @@ jobs:
           export PYTHONPATH=$PYTHONPATH:.
           python3 -m coverage run -m unittest discover tests
           python3 -m coverage report -m mini_copilot/*.py mini_copilot/commands/*.py
+
+      - name: Run integration tests
+        if: env.GITHUB_TOKEN_INTEGRATION != ''
+        env:
+          GITHUB_TOKEN_INTEGRATION: ${{ secrets.GITHUB_TOKEN_INTEGRATION }}
+        run: |
+          export PYTHONPATH=$PYTHONPATH:.
+          python3 -m unittest discover integration_tests

integration_tests/__init__.py

@@ -0,0 +1,44 @@
+import unittest
+import os
+import requests
+from mini_copilot import github_api
+
+class TestCopilotIntegration(unittest.TestCase):
+    def setUp(self):
+        self.github_token = os.environ.get("GITHUB_TOKEN_INTEGRATION")
+        if not self.github_token:
+            self.skipTest("GITHUB_TOKEN_INTEGRATION environment variable not set")
+
+    def test_copilot_token_and_models(self):
+        # Step 1: Exchange GitHub token for Copilot token
+        try:
+            copilot_token = github_api.get_copilot_token(self.github_token)
+            self.assertTrue(len(copilot_token) > 0, "Copilot token should not be empty")
+            print("\n✅ Successfully obtained Copilot token.")
+        except Exception as e:
+            self.fail(f"Failed to get Copilot token: {e}")
+
+        # Step 2: Fetch available models
+        try:
+            models = github_api.get_models(copilot_token)
+            self.assertIsInstance(models, list, "Models should be a list")
+            self.assertTrue(len(models) > 0, "Should find at least one model")
+            model_ids = [m['id'] for m in models]
+            print(f"✅ Found {len(models)} models. Available: {', '.join(model_ids[:5])}...")
+        except Exception as e:
+            self.fail(f"Failed to fetch models: {e}")
+
+    def test_copilot_chat(self):
+        copilot_token = github_api.get_copilot_token(self.github_token)
+        messages = [{"role": "user", "content": "Say 'hello world' and nothing else."}]
+        
+        try:
+            response = github_api.chat(messages, copilot_token, model="gpt-4o")
+            content = response.get("content", "").lower()
+            self.assertIn("hello world", content)
+            print(f"✅ Copilot Chat responded correctly: {content}")
+        except Exception as e:
+            self.fail(f"Chat integration failed: {e}")
+
+if __name__ == "__main__":
+    unittest.main()