user tests

SebRut · Sep 10, 2021 · f4dbcfc · f4dbcfc
1 parent 9c2a2cb
commit f4dbcfc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pygrocy/grocy_api_client.py b/pygrocy/grocy_api_client.py
@@ -509,12 +509,12 @@ def get_meal_plan_section(self, meal_plan_section_id) -> MealPlanSectionResponse
     def get_users(self) -> List[UserDto]:
         parsed_json = self._do_get_request("users")
         if parsed_json:
-            return [UserDto(user) for user in parsed_json]
+            return [UserDto(**user) for user in parsed_json]
 
     def get_user(self, id: int) -> UserDto:
         query_params = []
         if id:
             query_params.append(f"id={id}")
         parsed_json = self._do_get_request("users")
         if parsed_json:
-            return UserDto(parsed_json[0])
+            return UserDto(**parsed_json[0])
diff --git a/test/test_users.py b/test/test_users.py
@@ -9,6 +9,8 @@ def test_get_user_by_id_valid(self, grocy):
         user = grocy.user(id=1)
         assert isinstance(user, User)
         assert user.id == 1
+        assert user.display_name == "Demo User"
+        assert user.username == "Demo User"
 
     @pytest.mark.vcr
     def test_get_users_valid(self, grocy):