Skip to content

Commit

Permalink
fix: list_users should accept FgaObject type
Browse files Browse the repository at this point in the history
  • Loading branch information
evansims committed Jun 13, 2024
1 parent 664b0b9 commit 1f74653
Show file tree
Hide file tree
Showing 3 changed files with 18 additions and 15 deletions.
23 changes: 12 additions & 11 deletions openfga_sdk/client/models/list_users_request.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@
"""

from openfga_sdk.client.models.tuple import ClientTuple
from openfga_sdk.models.fga_object import FgaObject
from openfga_sdk.models.user_type_filter import UserTypeFilter


Expand All @@ -21,7 +22,7 @@ class ClientListUsersRequest:

def __init__(
self,
object: str = None,
object: FgaObject = None,
relation: str = None,
user_filters: list[UserTypeFilter] = None,
contextual_tuples: list[ClientTuple] = None,
Expand All @@ -34,7 +35,7 @@ def __init__(
self._context = context

@property
def object(self):
def object(self) -> FgaObject:
"""Gets the object of this ClientListUsersRequest.
Expand All @@ -44,7 +45,7 @@ def object(self):
return self._object

@object.setter
def object(self, object):
def object(self, object: FgaObject):
"""Sets the object of this ClientListUsersRequest.
Expand All @@ -55,7 +56,7 @@ def object(self, object):
self._object = object

@property
def relation(self):
def relation(self) -> str:
"""Gets the relation of this ClientListUsersRequest.
Expand All @@ -65,7 +66,7 @@ def relation(self):
return self._relation

@relation.setter
def relation(self, relation):
def relation(self, relation: str):
"""Sets the relation of this ClientListUsersRequest.
Expand All @@ -76,7 +77,7 @@ def relation(self, relation):
self._relation = relation

@property
def user_filters(self):
def user_filters(self) -> list[UserTypeFilter]:
"""Gets the user_filters of this ClientListUsersRequest.
Expand All @@ -86,7 +87,7 @@ def user_filters(self):
return self._user_filters

@user_filters.setter
def user_filters(self, user_filters):
def user_filters(self, user_filters: list[UserTypeFilter]):
"""Sets the user_filters of this ClientListUsersRequest.
Expand All @@ -97,7 +98,7 @@ def user_filters(self, user_filters):
self._user_filters = user_filters

@property
def contextual_tuples(self):
def contextual_tuples(self) -> list[ClientTuple]:
"""Gets the contextual_tuples of this ClientListUsersRequest.
Expand All @@ -107,7 +108,7 @@ def contextual_tuples(self):
return self._contextual_tuples

@contextual_tuples.setter
def contextual_tuples(self, contextual_tuples):
def contextual_tuples(self, contextual_tuples: list[ClientTuple]):
"""Sets the contextual_tuples of this ClientListUsersRequest.
Expand All @@ -118,7 +119,7 @@ def contextual_tuples(self, contextual_tuples):
self._contextual_tuples = contextual_tuples

@property
def context(self):
def context(self) -> object:
"""Gets the context of this ClientListUsersRequest.
Additional request context that will be used to evaluate any ABAC conditions encountered in the query evaluation.
Expand All @@ -129,7 +130,7 @@ def context(self):
return self._context

@context.setter
def context(self, context):
def context(self, context: object):
"""Sets the context of this ClientListUsersRequest.
Additional request context that will be used to evaluate any ABAC conditions encountered in the query evaluation.
Expand Down
5 changes: 3 additions & 2 deletions test/test_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,7 @@
from openfga_sdk.models.create_store_request import CreateStoreRequest
from openfga_sdk.models.create_store_response import CreateStoreResponse
from openfga_sdk.models.expand_response import ExpandResponse
from openfga_sdk.models.fga_object import FgaObject
from openfga_sdk.models.get_store_response import GetStoreResponse
from openfga_sdk.models.leaf import Leaf
from openfga_sdk.models.list_objects_response import ListObjectsResponse
Expand Down Expand Up @@ -2508,7 +2509,7 @@ async def test_list_users(self, mock_request):

async with OpenFgaClient(configuration) as api_client:
body = ClientListUsersRequest(
object="document:2021-budget",
object=FgaObject(type="document", id="2021-budget"),
relation="can_read",
user_filters=[
UserTypeFilter(type="user"),
Expand Down Expand Up @@ -2565,7 +2566,7 @@ async def test_list_users(self, mock_request):
post_params=[],
body={
"authorization_model_id": "01G5JAVJ41T49E9TT3SKVS7X1J",
"object": "document:2021-budget",
"object": {"id": "2021-budget", "type": "document"},
"relation": "can_read",
"user_filters": [
{"type": "user"},
Expand Down
5 changes: 3 additions & 2 deletions test/test_client_sync.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@
from openfga_sdk.models.create_store_request import CreateStoreRequest
from openfga_sdk.models.create_store_response import CreateStoreResponse
from openfga_sdk.models.expand_response import ExpandResponse
from openfga_sdk.models.fga_object import FgaObject
from openfga_sdk.models.get_store_response import GetStoreResponse
from openfga_sdk.models.leaf import Leaf
from openfga_sdk.models.list_objects_response import ListObjectsResponse
Expand Down Expand Up @@ -2508,7 +2509,7 @@ def test_list_users(self, mock_request):

with OpenFgaClient(configuration) as api_client:
body = ClientListUsersRequest()
body.object = "document:2021-budget"
body.object = FgaObject(type="document", id="2021-budget")
body.relation = "can_read"
body.user_filters = [
UserTypeFilter(type="user"),
Expand Down Expand Up @@ -2564,7 +2565,7 @@ def test_list_users(self, mock_request):
post_params=[],
body={
"authorization_model_id": "01G5JAVJ41T49E9TT3SKVS7X1J",
"object": "document:2021-budget",
"object": {"id": "2021-budget", "type": "document"},
"relation": "can_read",
"user_filters": [
{"type": "user"},
Expand Down

0 comments on commit 1f74653

Please sign in to comment.