feat: client to update auth model id and store id

Author: adriantam

openfga_sdk/api_client.py

@@ -705,3 +705,15 @@ def _get_store_id(self):
                 'store_id is required but not configured'
             )
         return configuration.store_id
+
+    def set_store_id(self, value):
+        """
+        Update the store ID in the configuration
+        """
+        self.configuration.store_id = value
+
+    def get_store_id(self):
+        """
+        Return the store id (if any) store in the configuration
+        """
+        return self.configuration.store_id

openfga_sdk/client/openfga_client.py

@@ -128,6 +128,30 @@ def _get_authorization_model_id(self, options):
         else:
             return self._client_configuration.authorization_model_id
 
+    def set_store_id(self, value):
+        """
+        Update the store ID in the configuration
+        """
+        self._api_client.set_store_id(value)
+
+    def get_store_id(self):
+        """
+        Return the store id (if any) store in the configuration
+        """
+        return self._api_client.get_store_id()
+
+    def set_authorization_model_id(self, value):
+        """
+        Update the authorizaiton model id in the configuration
+        """
+        self._client_configuration.authorization_model_id = value
+
+    def get_authorization_model_id(self):
+        """
+        Return the authorizaiton model id
+        """
+        return self._client_configuration.authorization_model_id
+
     #################
     # Stores
     #################

test/test_openfga_client.py

@@ -177,7 +177,6 @@ async def test_create_store(self, mock_request):
             '''
         mock_request.return_value = mock_response(response_body, 201)
         configuration = self.configuration
-        configuration.store_id = store_id
         async with OpenFgaClient(configuration) as api_client:
             api_response = await api_client.create_store(
                 CreateStoreRequest(name="test-store"),
@@ -1911,3 +1910,99 @@ async def test_write_assertions(self, mock_request):
                 _preload_content=ANY,
                 _request_timeout=None
             )
+
+    @patch.object(rest.RESTClientObject, 'request')
+    async def test_set_store_id(self, mock_request):
+        """Test case for write assertions
+
+        Get all stores  # noqa: E501
+        """
+        mock_request.return_value = mock_response('', 204)
+        configuration = self.configuration
+        configuration.store_id = store_id
+        async with OpenFgaClient(configuration) as api_client:
+            api_client.set_store_id("01YCP46JKYM8FJCQ37NMBYHE5Y")
+            await api_client.write_assertions(
+                [Assertion(
+                    tuple_key=TupleKey(object="document:2021-budget", relation="reader",
+                                       user="user:anne"),
+                    expectation=True,
+                )],
+                options={"authorization_model_id": "01G5JAVJ41T49E9TT3SKVS7X1J"}
+            )
+            self.assertEqual(api_client.get_store_id(), "01YCP46JKYM8FJCQ37NMBYHE5Y")
+            mock_request.assert_called_once_with(
+                'PUT',
+                'http://api.fga.example/stores/01YCP46JKYM8FJCQ37NMBYHE5Y/assertions/01G5JAVJ41T49E9TT3SKVS7X1J',
+                headers=ANY,
+                body={"assertions": [{"tuple_key": {"object": "document:2021-budget",
+                                                    "relation": "reader", "user": "user:anne"}, "expectation": True}]},
+                query_params=[],
+                post_params=[],
+                _preload_content=ANY,
+                _request_timeout=None
+            )
+
+    @patch.object(rest.RESTClientObject, 'request')
+    async def test_config_auth_model(self, mock_request):
+        """Test case for write assertions
+
+        Get all stores  # noqa: E501
+        """
+        mock_request.return_value = mock_response('', 204)
+        configuration = self.configuration
+        configuration.store_id = store_id
+        configuration.authorization_model_id = "01G5JAVJ41T49E9TT3SKVS7X1J"
+        async with OpenFgaClient(configuration) as api_client:
+            await api_client.write_assertions(
+                [Assertion(
+                    tuple_key=TupleKey(object="document:2021-budget", relation="reader",
+                                       user="user:anne"),
+                    expectation=True,
+                )],
+                options={}
+            )
+            self.assertEqual(api_client.get_authorization_model_id(), "01G5JAVJ41T49E9TT3SKVS7X1J")
+            mock_request.assert_called_once_with(
+                'PUT',
+                'http://api.fga.example/stores/01YCP46JKYM8FJCQ37NMBYHE5X/assertions/01G5JAVJ41T49E9TT3SKVS7X1J',
+                headers=ANY,
+                body={"assertions": [{"tuple_key": {"object": "document:2021-budget",
+                                                    "relation": "reader", "user": "user:anne"}, "expectation": True}]},
+                query_params=[],
+                post_params=[],
+                _preload_content=ANY,
+                _request_timeout=None
+            )
+
+    @patch.object(rest.RESTClientObject, 'request')
+    async def test_update_auth_model(self, mock_request):
+        """Test case for write assertions
+
+        Get all stores  # noqa: E501
+        """
+        mock_request.return_value = mock_response('', 204)
+        configuration = self.configuration
+        configuration.store_id = store_id
+        configuration.authorization_model_id = "01G5JAVJ41T49E9TT3SKVS7X1J"
+        async with OpenFgaClient(configuration) as api_client:
+            api_client.set_authorization_model_id("01G5JAVJ41T49E9TT3SKVS7X2J")
+            await api_client.write_assertions(
+                [Assertion(
+                    tuple_key=TupleKey(object="document:2021-budget", relation="reader",
+                                       user="user:anne"),
+                    expectation=True,
+                )],
+                options={}
+            )
+            mock_request.assert_called_once_with(
+                'PUT',
+                'http://api.fga.example/stores/01YCP46JKYM8FJCQ37NMBYHE5X/assertions/01G5JAVJ41T49E9TT3SKVS7X2J',
+                headers=ANY,
+                body={"assertions": [{"tuple_key": {"object": "document:2021-budget",
+                                                    "relation": "reader", "user": "user:anne"}, "expectation": True}]},
+                query_params=[],
+                post_params=[],
+                _preload_content=ANY,
+                _request_timeout=None
+            )