Create test_ai_system.py

tests/test_ai_system.py

@@ -0,0 +1,51 @@
+import pytest
+import sys
+import os
+
+# Add src to path
+sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
+
+from src.ai_system import SaemsTunesAISystem
+from src.supabase_integration import SupabaseIntegration
+from src.security_system import SecuritySystem
+from src.monitoring_system import ComprehensiveMonitor
+
+class TestAISystem:
+    """Test AI system functionality"""
+    
+    def setup_method(self):
+        """Setup test fixtures"""
+        self.supabase = SupabaseIntegration("test_url", "test_key")
+        self.security = SecuritySystem()
+        self.monitor = ComprehensiveMonitor()
+        
+        # Mock the model loading to avoid downloading
+        self.ai_system = SaemsTunesAISystem(self.supabase, self.security, self.monitor)
+        self.ai_system.model_loaded = True  # Simulate loaded model
+    
+    def test_process_query_basic(self):
+        """Test basic query processing"""
+        response = self.ai_system.process_query("Hello", "test_user")
+        assert isinstance(response, str)
+        assert len(response) > 0
+    
+    def test_build_prompt(self):
+        """Test prompt building"""
+        context = {
+            "stats": {"song_count": 100, "artist_count": 10, "user_count": 1000},
+            "summary": "Test context"
+        }
+        
+        prompt = self.ai_system.build_prompt("test question", context)
+        
+        assert "test question" in prompt
+        assert "Test context" in prompt
+        assert "<|system|>" in prompt
+        assert "<|user|>" in prompt
+    
+    def test_system_health(self):
+        """Test system health check"""
+        assert self.ai_system.is_healthy() == True
+
+if __name__ == "__main__":
+    pytest.main([__file__])