add more tests of webutils.py

Author: Arendelle

physicsLab/web/webutils.py

@@ -357,19 +357,19 @@ class AvatarsIter(_async_tool.AsyncTool):
     ''' 获取头像的迭代器 '''
     def __init__(
             self,
-            search_id: str,
+            user_id: str,
             category: str,
             user: Optional[api.User] = None,
             size_category: str = "full",
             max_retry: Optional[int] = 0,
     ) -> None:
-        ''' @param search_id: 用户id
+        ''' @param user_id: 用户id
             @param category: 只能为 "Experiment" 或 "Discussion" 或 "User"
             @param size_category: 只能为 "small.round" 或 "thumbnail" 或 "full"
             @param user: 查询者, None为匿名用户
             @param max_retry: 最大重试次数(大于等于0), 为None时不限制重试次数
         '''
-        if not isinstance(search_id, str) or \
+        if not isinstance(user_id, str) or \
                 not isinstance(category, str) or \
                 not isinstance(size_category, str) or \
                 not isinstance(user, (api.User, type(None))) or \
@@ -383,18 +383,18 @@ def __init__(
             user = api.User()
 
         if category == "User":
-            self.max_img_counter = user.get_user(search_id)["Data"]["User"]["Avatar"]
+            self.max_img_counter = user.get_user(user_id)["Data"]["User"]["Avatar"]
             category = "users"
         elif category == "Experiment":
-            self.max_img_counter = user.get_summary(search_id, Category.Experiment)["Data"]["Image"]
+            self.max_img_counter = user.get_summary(user_id, Category.Experiment)["Data"]["Image"]
             category = "experiments"
         elif category == "Discussion":
-            self.max_img_counter = user.get_summary(search_id, Category.Discussion)["Data"]["Image"]
+            self.max_img_counter = user.get_summary(user_id, Category.Discussion)["Data"]["Image"]
             category = "experiments"
         else:
             assert False
 
-        self.search_id = search_id
+        self.search_id = user_id
         self.category = category
         self.size_category = size_category
         self.user = user

test_pl/test_pl_web.py

@@ -1,7 +1,4 @@
 # -*- coding: utf-8 -*-
-import inspect
-import asyncio
-import time
 from datetime import datetime
 from .base import *
 
@@ -74,3 +71,24 @@ async def test_banned_msg_iter(self):
         ):
             counter += 1
         self.assertEqual(counter, 1)
+
+    async def test_comments_iter(self):
+        for _ in web.CommentsIter(user=user, id="677d5c6c826568de4e9896c5", category="Discussion"):
+            pass
+
+    # temp user can't get comments on user's profile
+    # async def test_warned_msg_iter(self):
+    #     counter = 0
+    #     for _ in web.WarnedMsgIter(user=user, user_id="63e76ce5fd0015cb932e8b05", start_time=1731155184, end_time=1731155185):
+    #         counter += 1
+    #     self.assertEqual(counter, 1)
+
+    async def test_relations_iter(self):
+        for _ in web.RelationsIter(user=user, user_id="62d3fd092f3a2a60cc8ccc9e", display_type="Following", max_retry=4):
+            pass
+
+    async def test_avatars_iter(self):
+        for _ in web.AvatarsIter(
+            user_id="5ce629e157035932b52f9315", category="User", user=user, size_category="small.round", max_retry=4
+        ):
+            pass