commit

test_group_profile.py

@@ -0,0 +1,70 @@
+"""
+测试根据环境查询分组和Profile的功能
+"""
+from adspower_client import AdsPowerClient
+from config import Config
+from loguru import logger
+
+def main():
+    logger.info("="*70)
+    logger.info("测试根据环境查询分组和Profile")
+    logger.info("="*70)
+    logger.info(f"当前环境: {Config.ENV}")
+    logger.info("")
+    
+    # 创建客户端
+    client = AdsPowerClient()
+    
+    # 步骤1: 查询所有分组
+    logger.info("步骤1: 查询所有分组")
+    logger.info("-"*70)
+    all_groups = client.list_groups()
+    
+    if all_groups and all_groups.get('code') == 0:
+        groups = all_groups.get('data', {}).get('list', [])
+        logger.info(f"✅ 找到 {len(groups)} 个分组\n")
+    else:
+        logger.error("❌ 查询所有分组失败\n")
+        return
+    
+    # 步骤2: 根据环境查询分组
+    logger.info("步骤2: 根据环境自动获取分组ID")
+    logger.info("-"*70)
+    group_id = client.get_group_by_env()
+    
+    if group_id:
+        logger.info(f"✅ 获取到分组ID: {group_id}\n")
+    else:
+        logger.error("❌ 未找到对应环境的分组\n")
+        logger.info("提示: 请在AdsPower中创建名为 'dev' 的分组（开发环境）或 'prod' 的分组（生产环境）")
+        return
+    
+    # 步骤3: 使用分组ID查询Profile列表
+    logger.info("步骤3: 使用分组ID查询Profile列表")
+    logger.info("-"*70)
+    profiles = client.list_profiles(group_id=group_id)
+    
+    if profiles and profiles.get('code') == 0:
+        profile_list = profiles.get('data', {}).get('list', [])
+        logger.info(f"✅ 在分组 {group_id} 中找到 {len(profile_list)} 个Profile\n")
+        
+        if profile_list:
+            logger.info("前5个Profile详情:")
+            for idx, profile in enumerate(profile_list[:5], 1):
+                profile_id = profile.get('profile_id', 'N/A')
+                profile_name = profile.get('name', 'N/A')
+                group_name = profile.get('group_name', 'N/A')
+                logger.info(f"  {idx}. ID: {profile_id}")
+                logger.info(f"     名称: {profile_name}")
+                logger.info(f"     分组: {group_name}")
+                logger.info("")
+    else:
+        logger.error("❌ 查询Profile列表失败\n")
+        return
+    
+    logger.info("="*70)
+    logger.info("测试完成！")
+    logger.info("="*70)
+
+if __name__ == "__main__":
+    main()