diff --git a/src/galaxy/api/types.py b/src/galaxy/api/types.py index 4ba4c76..72c0d31 100644 --- a/src/galaxy/api/types.py +++ b/src/galaxy/api/types.py @@ -61,9 +61,11 @@ class NextStep: if not stored_credentials: return NextStep("web_session", PARAMS, cookies=COOKIES, js=JS) - :param auth_params: configuration options: {"window_title": :class:`str`, "window_width": :class:`str`, "window_height": :class:`int`, "start_uri": :class:`int`, "end_uri_regex": :class:`str`} + :param auth_params: configuration options: {"window_title": :class:`str`, "window_width": :class:`str`, + "window_height": :class:`int`, "start_uri": :class:`int`, "end_uri_regex": :class:`str`} :param cookies: browser initial set of cookies - :param js: a map of the url regex patterns into the list of *js* scripts that should be executed on every document at given step of internal browser authentication. + :param js: a map of the url regex patterns into the list of *js* scripts that should be executed + on every document at given step of internal browser authentication. """ next_step: str auth_params: Dict[str, str] @@ -158,9 +160,14 @@ class UserInfo: :param user_id: id of the user :param user_name: username of the user + :param avatar_url: the URL of the user avatar + :param profile_url: the URL of the user profile """ user_id: str user_name: str + avatar_url: Optional[str] + profile_url: Optional[str] + @dataclass class GameTime: diff --git a/tests/test_friends.py b/tests/test_friends.py index 4222e38..c52ced1 100644 --- a/tests/test_friends.py +++ b/tests/test_friends.py @@ -17,8 +17,8 @@ async def test_get_friends_success(plugin, read, write): read.side_effect = [async_return_value(create_message(request)), async_return_value(b"", 10)] plugin.get_friends.return_value = async_return_value([ - UserInfo("3", "Jan"), - UserInfo("5", "Ola") + UserInfo("3", "Jan", "https://avatar.url/u3", None), + UserInfo("5", "Ola", None, "https://profile.url/u5") ]) await plugin.run() plugin.get_friends.assert_called_with() @@ -29,8 +29,8 @@ async def test_get_friends_success(plugin, read, write): "id": "3", "result": { "friend_info_list": [ - {"user_id": "3", "user_name": "Jan"}, - {"user_id": "5", "user_name": "Ola"} + {"user_id": "3", "user_name": "Jan", "avatar_url": "https://avatar.url/u3"}, + {"user_id": "5", "user_name": "Ola", "profile_url": "https://profile.url/u5"} ] } } @@ -64,7 +64,7 @@ async def test_get_friends_failure(plugin, read, write): @pytest.mark.asyncio async def test_add_friend(plugin, write): - friend = UserInfo("7", "Kuba") + friend = UserInfo("7", "Kuba", avatar_url="https://avatar.url/kuba.jpg", profile_url="https://profile.url/kuba") plugin.add_friend(friend) await skip_loop() @@ -74,7 +74,12 @@ async def test_add_friend(plugin, write): "jsonrpc": "2.0", "method": "friend_added", "params": { - "friend_info": {"user_id": "7", "user_name": "Kuba"} + "friend_info": { + "user_id": "7", + "user_name": "Kuba", + "avatar_url": "https://avatar.url/kuba.jpg", + "profile_url": "https://profile.url/kuba" + } } } ]