test: Add tests on OpenFgaClient writeTuples() deleteTuples() and rea…

…dLatestAuthorizationModel()
openfga · Sep 26, 2023 · 916f89e · 916f89e
1 parent 62176c2
commit 916f89e
Showing 1 changed file with 63 additions and 0 deletions.
diff --git a/src/test/java/dev/openfga/sdk/api/client/OpenFgaClientTest.java b/src/test/java/dev/openfga/sdk/api/client/OpenFgaClientTest.java
@@ -557,6 +557,27 @@ public void readAuthorizationModels_500() throws Exception {
                 "{\"code\":\"internal_error\",\"message\":\"Internal Server Error\"}", exception.getResponseBody());
     }
 
+    @Test
+    public void readLatestAuthorizationModelTest() throws Exception {
+        // Given
+        String getUrl = String.format("https://localhost/stores/%s/authorization-models?page_size=1", DEFAULT_STORE_ID);
+        String responseBody = String.format(
+                "{\"authorization_models\":[{\"id\":\"%s\",\"schema_version\":\"%s\"}]}",
+                DEFAULT_AUTH_MODEL_ID, DEFAULT_SCHEMA_VERSION);
+        mockHttpClient.onGet(getUrl).doReturn(200, responseBody);
+
+        // When
+        ReadAuthorizationModelResponse response =
+                fga.readLatestAuthorizationModel().get();
+
+        // Then
+        mockHttpClient.verify().get(getUrl).called(1);
+        assertNotNull(response.getAuthorizationModel());
+        AuthorizationModel authModel = response.getAuthorizationModel();
+        assertEquals(DEFAULT_AUTH_MODEL_ID, authModel.getId());
+        assertEquals(DEFAULT_SCHEMA_VERSION, authModel.getSchemaVersion());
+    }
+
     /**
      * Create a new authorization model.
      */
@@ -961,6 +982,48 @@ public void writeTest_deletes() throws Exception {
         mockHttpClient.verify().post(postPath).withBody(is(expectedBody)).called(1);
     }
 
+    @Test
+    public void writeTuplesTest() throws Exception {
+        // Given
+        String postPath = String.format("https://localhost/stores/%s/write", DEFAULT_STORE_ID);
+        String expectedBody = String.format(
+                "{\"writes\":{\"tuple_keys\":[{\"object\":\"%s\",\"relation\":\"%s\",\"user\":\"%s\"}]},\"deletes\":null,\"authorization_model_id\":\"%s\"}",
+                DEFAULT_OBJECT, DEFAULT_RELATION, DEFAULT_USER, DEFAULT_AUTH_MODEL_ID);
+        mockHttpClient.onPost(postPath).withBody(is(expectedBody)).doReturn(200, EMPTY_RESPONSE_BODY);
+        TupleKeys tuples = new TupleKeys()
+                .tupleKeys(List.of(new TupleKey()
+                        ._object(DEFAULT_OBJECT)
+                        .relation(DEFAULT_RELATION)
+                        .user(DEFAULT_USER)));
+
+        // When
+        fga.writeTuples(tuples);
+
+        // Then
+        mockHttpClient.verify().post(postPath).withBody(is(expectedBody)).called(1);
+    }
+
+    @Test
+    public void deleteTuplesTest() throws Exception {
+        // Given
+        String postPath = String.format("https://localhost/stores/%s/write", DEFAULT_STORE_ID);
+        String expectedBody = String.format(
+                "{\"writes\":null,\"deletes\":{\"tuple_keys\":[{\"object\":\"%s\",\"relation\":\"%s\",\"user\":\"%s\"}]},\"authorization_model_id\":\"%s\"}",
+                DEFAULT_OBJECT, DEFAULT_RELATION, DEFAULT_USER, DEFAULT_AUTH_MODEL_ID);
+        mockHttpClient.onPost(postPath).withBody(is(expectedBody)).doReturn(200, EMPTY_RESPONSE_BODY);
+        TupleKeys tuples = new TupleKeys()
+                .tupleKeys(List.of(new TupleKey()
+                        ._object(DEFAULT_OBJECT)
+                        .relation(DEFAULT_RELATION)
+                        .user(DEFAULT_USER)));
+
+        // When
+        fga.deleteTuples(tuples);
+
+        // Then
+        mockHttpClient.verify().post(postPath).withBody(is(expectedBody)).called(1);
+    }
+
     @Test
     public void write_storeIdRequired() {
         // Given