Add support for Profile feature in REST Identity store (Fix #91)

2018-12-21 19:21:15 +01:00
parent ad1b91f370
commit 92cf5c6b21
7 changed files with 462 additions and 49 deletions
--- a/src/test/java/io/kamax/mxisd/backend/rest/RestProfileProviderTest.java
+++ b/src/test/java/io/kamax/mxisd/backend/rest/RestProfileProviderTest.java
@@ -0,0 +1,146 @@
+/*
+ * mxisd - Matrix Identity Server Daemon
+ * Copyright (C) 2018 Kamax Sarl
+ *
+ * https://www.kamax.io/
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package io.kamax.mxisd.backend.rest;
+
+import com.github.tomakehurst.wiremock.junit.WireMockRule;
+import io.kamax.matrix.MatrixID;
+import io.kamax.matrix._MatrixID;
+import io.kamax.matrix.json.GsonUtil;
+import io.kamax.mxisd.config.rest.RestBackendConfig;
+import io.kamax.mxisd.exception.InternalServerError;
+import io.kamax.mxisd.profile.JsonProfileRequest;
+import io.kamax.mxisd.profile.JsonProfileResult;
+import org.apache.http.entity.ContentType;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+
+import java.util.Optional;
+
+import static com.github.tomakehurst.wiremock.client.WireMock.*;
+import static io.kamax.mxisd.config.rest.RestBackendConfig.ProfileEndpoints;
+import static org.junit.Assert.*;
+
+public class RestProfileProviderTest {
+
+    @Rule
+    public WireMockRule wireMockRule = new WireMockRule(65000);
+
+    private final String displayNameEndpoint = "/displayName";
+
+    private final _MatrixID userId = MatrixID.from("john", "matrix.localhost").valid();
+
+    private RestProfileProvider p;
+
+    @Before
+    public void before() {
+        ProfileEndpoints endpoints = new ProfileEndpoints();
+        endpoints.setDisplayName(displayNameEndpoint);
+
+        RestBackendConfig cfg = new RestBackendConfig();
+        cfg.setEnabled(true);
+        cfg.setHost("http://localhost:65000");
+        cfg.getEndpoints().setProfile(endpoints);
+        cfg.build();
+
+        p = new RestProfileProvider(cfg);
+    }
+
+    @Test
+    public void forNameFound() {
+        String value = "This is my display name";
+
+        JsonProfileResult r = new JsonProfileResult();
+        r.setDisplayName(value);
+        stubFor(post(urlEqualTo(displayNameEndpoint))
+                .willReturn(aResponse()
+                        .withHeader("Content-Type", ContentType.APPLICATION_JSON.getMimeType())
+                        .withBody(GsonUtil.get().toJson(GsonUtil.makeObj("profile", r)))
+                )
+        );
+
+        Optional<String> v = p.getDisplayName(userId);
+
+        verify(postRequestedFor(urlMatching(displayNameEndpoint))
+                .withHeader("Content-Type", containing(ContentType.APPLICATION_JSON.getMimeType()))
+                .withRequestBody(equalTo(GsonUtil.get().toJson(new JsonProfileRequest(userId))))
+        );
+
+        assertTrue(v.isPresent());
+        assertEquals(value, v.get());
+    }
+
+    @Test
+    public void forNameNotFound() {
+        stubFor(post(urlEqualTo(displayNameEndpoint))
+                .willReturn(aResponse()
+                        .withStatus(404)
+                )
+        );
+
+        Optional<String> v = p.getDisplayName(userId);
+
+        verify(postRequestedFor(urlMatching(displayNameEndpoint))
+                .withHeader("Content-Type", containing(ContentType.APPLICATION_JSON.getMimeType()))
+                .withRequestBody(equalTo(GsonUtil.get().toJson(new JsonProfileRequest(userId))))
+        );
+
+        assertFalse(v.isPresent());
+    }
+
+    @Test
+    public void forNameEmptyBody() {
+        stubFor(post(urlEqualTo(displayNameEndpoint))
+                .willReturn(aResponse()
+                        .withHeader("Content-Type", ContentType.APPLICATION_JSON.getMimeType())
+                )
+        );
+
+        Optional<String> v = p.getDisplayName(userId);
+
+        verify(postRequestedFor(urlMatching(displayNameEndpoint))
+                .withHeader("Content-Type", containing(ContentType.APPLICATION_JSON.getMimeType()))
+                .withRequestBody(equalTo(GsonUtil.get().toJson(new JsonProfileRequest(userId))))
+        );
+
+        assertFalse(v.isPresent());
+    }
+
+    @Test(expected = InternalServerError.class)
+    public void forNameInvalidBody() {
+        stubFor(post(urlEqualTo(displayNameEndpoint))
+                .willReturn(aResponse()
+                        .withHeader("Content-Type", ContentType.APPLICATION_JSON.getMimeType())
+                        .withBody("This is not a valid JSON object")
+                )
+        );
+
+        try {
+            p.getDisplayName(userId);
+        } finally {
+            verify(postRequestedFor(urlMatching(displayNameEndpoint))
+                    .withHeader("Content-Type", containing(ContentType.APPLICATION_JSON.getMimeType()))
+                    .withRequestBody(equalTo(GsonUtil.get().toJson(new JsonProfileRequest(userId))))
+            );
+        }
+    }
+
+}