Create test_supabase.py

tests/test_supabase.py

@@ -0,0 +1,46 @@
+import pytest
+import sys
+import os
+
+# Add src to path
+sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
+
+from src.supabase_integration import SupabaseIntegration
+
+class TestSupabaseIntegration:
+    """Test Supabase integration"""
+    
+    def setup_method(self):
+        """Setup test fixtures"""
+        self.supabase = SupabaseIntegration("test_url", "test_key")
+    
+    def test_get_music_context(self):
+        """Test music context retrieval"""
+        context = self.supabase.get_music_context("test query")
+        
+        assert isinstance(context, dict)
+        assert "songs" in context
+        assert "artists" in context
+        assert "stats" in context
+        assert "summary" in context
+    
+    def test_generate_summary(self):
+        """Test summary generation"""
+        context = {
+            "stats": {"song_count": 100, "artist_count": 10, "user_count": 1000},
+            "songs": [{"title": "Test Song", "artist": "Test Artist"}],
+            "artists": [{"name": "Test Artist"}]
+        }
+        
+        summary = self.supabase.generate_summary(context, "test query")
+        assert isinstance(summary, str)
+        assert len(summary) > 0
+    
+    def test_fallback_context(self):
+        """Test fallback context"""
+        context = self.supabase.get_fallback_context()
+        assert isinstance(context, dict)
+        assert "summary" in context
+
+if __name__ == "__main__":
+    pytest.main([__file__])