diff --git a/rocketchat_API/APISections/custom_user_status.py b/rocketchat_API/APISections/custom_user_status.py index 2965687..431a8a2 100644 --- a/rocketchat_API/APISections/custom_user_status.py +++ b/rocketchat_API/APISections/custom_user_status.py @@ -1,7 +1,8 @@ -from rocketchat_API.APISections.base import RocketChatBase +from rocketchat_API.APISections.base import RocketChatBase, paginated class RocketChatCustomUserStatus(RocketChatBase): + @paginated("statuses") def custom_user_status_list(self, **kwargs): """Lists all available custom user's status.""" return self.call_api_get("custom-user-status.list", kwargs=kwargs) diff --git a/tests/test_custom_user_status.py b/tests/test_custom_user_status.py index c2369ab..68ebd93 100644 --- a/tests/test_custom_user_status.py +++ b/tests/test_custom_user_status.py @@ -22,10 +22,8 @@ def created_status(logged_rocket, status_name): def test_custom_user_status_list(logged_rocket, created_status): - result = logged_rocket.custom_user_status_list() - assert "statuses" in result - found = any(s["_id"] == created_status["_id"] for s in result["statuses"]) - assert found + statuses = list(logged_rocket.custom_user_status_list()) + assert any(s["_id"] == created_status["_id"] for s in statuses) def test_custom_user_status_create_delete(logged_rocket): @@ -38,7 +36,7 @@ def test_custom_user_status_create_delete(logged_rocket): logged_rocket.custom_user_status_delete(custom_user_status_id=status["_id"]) - statuses = logged_rocket.custom_user_status_list().get("statuses", []) + statuses = list(logged_rocket.custom_user_status_list()) assert not any(s["_id"] == status["_id"] for s in statuses)