优化结果

InfoGenie-backend/test/test_mongo.py

@@ -0,0 +1,70 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+测试MongoDB连接
+"""
+
+import os
+from pymongo import MongoClient
+from dotenv import load_dotenv
+
+# 加载环境变量
+load_dotenv()
+
+def test_mongodb_connection():
+    """测试MongoDB连接"""
+    try:
+        # 获取连接字符串
+        mongo_uri = os.environ.get('MONGO_URI')
+        print(f"连接字符串: {mongo_uri}")
+        
+        # 创建连接
+        client = MongoClient(mongo_uri)
+        
+        # 测试连接
+        client.admin.command('ping')
+        print("✅ MongoDB连接成功！")
+        
+        # 获取数据库
+        db = client.InfoGenie
+        print(f"数据库: {db.name}")
+        
+        # 测试集合访问
+        userdata_collection = db.userdata
+        print(f"用户集合: {userdata_collection.name}")
+        
+        # 测试查询（计算文档数量）
+        count = userdata_collection.count_documents({})
+        print(f"用户数据集合中有 {count} 个文档")
+        
+        # 关闭连接
+        client.close()
+        
+    except Exception as e:
+        print(f"❌ MongoDB连接失败: {str(e)}")
+        
+        # 尝试其他认证数据库
+        print("\n尝试使用不同的认证配置...")
+        try:
+            # 尝试不指定认证数据库
+            uri_without_auth = "mongodb://shumengya:tyh%4019900420@192.168.1.233:27017/InfoGenie"
+            client2 = MongoClient(uri_without_auth)
+            client2.admin.command('ping')
+            print("✅ 不使用authSource连接成功！")
+            client2.close()
+        except Exception as e2:
+            print(f"❌ 无authSource也失败: {str(e2)}")
+            
+        # 尝试使用InfoGenie作为认证数据库
+        try:
+            uri_with_infogenie_auth = "mongodb://shumengya:tyh%4019900420@192.168.1.233:27017/InfoGenie?authSource=InfoGenie"
+            client3 = MongoClient(uri_with_infogenie_auth)
+            client3.admin.command('ping')
+            print("✅ 使用InfoGenie作为authSource连接成功！")
+            client3.close()
+        except Exception as e3:
+            print(f"❌ InfoGenie authSource也失败: {str(e3)}")
+
+if __name__ == "__main__":
+    print("🔧 测试MongoDB连接...")
+    test_mongodb_connection()