CB-2365 rp tests

server/bundles/io.cloudbeaver.model/src/io/cloudbeaver/utils/WebTestUtils.java

@@ -29,6 +29,7 @@ import java.net.http.HttpResponse;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.util.Collections;
+import java.util.List;
 import java.util.Map;
 
 public class WebTestUtils {
@@ -42,12 +43,24 @@ public class WebTestUtils {
 
 
     public static Map<String, Object> doPost(String apiUrl, String input, HttpClient client) throws Exception {
-        HttpRequest request = HttpRequest.newBuilder()
+        return doPostWithHeaders(apiUrl, input, client, List.of());
+    }
+
+    public static Map<String, Object> doPostWithHeaders(
+        String apiUrl,
+        String input,
+        HttpClient client,
+        List<String> headers
+    ) throws Exception {
+        HttpRequest.Builder requestBuilder = HttpRequest.newBuilder()
             .uri(URI.create(apiUrl))
             .POST(HttpRequest.BodyPublishers.ofString(input))
-            .header("Content-Type", "application/json")
-            .build();
+            .header("Content-Type", "application/json");
 
+        if (!headers.isEmpty()) {
+            requestBuilder.headers(headers.toArray(String[]::new));
+        }
+        HttpRequest request = requestBuilder.build();
         HttpResponse<String> response = client.send(request,
             HttpResponse.BodyHandlers.ofString());
 

server/test/io.cloudbeaver.test.platform/src/io/cloudbeaver/test/platform/AuthenticationTest.java

@@ -1,15 +1,23 @@
 package io.cloudbeaver.test.platform;
 
+import io.cloudbeaver.auth.provider.rp.RPAuthProvider;
 import io.cloudbeaver.utils.WebTestUtils;
 import org.jkiss.dbeaver.model.auth.SMAuthStatus;
 import org.jkiss.dbeaver.model.data.json.JSONUtils;
 import org.junit.Assert;
 import org.junit.Test;
+import org.mockito.Mockito;
 
 import java.net.http.HttpClient;
+import java.net.http.HttpRequest;
+import java.util.Collections;
+import java.util.List;
 import java.util.Map;
 
 public class AuthenticationTest {
+    public static final String GQL_TEMPLATE_OPEN_SESSION = "openSession.json";
+    public static final String GQL_TEMPLATE_ACTIVE_USER = "activeUser.json";
+    public static final String REVERSE_PROXY_TEST_USER = "reverseProxyTestUser";
 
     @Test
     public void testLoginUser() throws Exception {
@@ -18,4 +26,38 @@ public class AuthenticationTest {
             client, CEServerTestSuite.getScriptsPath(), CEServerTestSuite.GQL_API_URL);
         Assert.assertEquals(SMAuthStatus.SUCCESS.name(), JSONUtils.getString(authInfo, "authStatus"));
     }
+
+    @Test
+    public void testReverseProxyAnonymousModeLogin() throws Exception {
+        HttpClient client = CEServerTestSuite.getClient();
+        Map<String, Object> sessionInfo = openSession(client);
+        Assert.assertTrue(JSONUtils.getBoolean(sessionInfo, "valid"));
+        Map<String, Object> activeUser = getActiveUser(client);
+        Assert.assertEquals(REVERSE_PROXY_TEST_USER, JSONUtils.getString(activeUser, "userId"));
+    }
+
+    private Map<String, Object> openSession(HttpClient client) throws Exception {
+        Map<String, Object> data = doPostQuery(client, GQL_TEMPLATE_OPEN_SESSION);
+        if (data != null) {
+            return JSONUtils.getObject(data, "session");
+        }
+        return Collections.emptyMap();
+    }
+
+    private Map<String, Object> getActiveUser(HttpClient client) throws Exception {
+        Map<String, Object> data = doPostQuery(client, GQL_TEMPLATE_ACTIVE_USER);
+        if (data != null) {
+            return JSONUtils.getObject(data, "user");
+        }
+        return Collections.emptyMap();
+    }
+
+    private Map<String, Object> doPostQuery(HttpClient client, String gqlScript) throws Exception {
+        String input = WebTestUtils.readScriptTemplate(gqlScript, CEServerTestSuite.getScriptsPath());
+        List<String> headers = List.of(RPAuthProvider.X_USER, REVERSE_PROXY_TEST_USER, RPAuthProvider.X_ROLE, "user");
+        Map<String, Object> map = WebTestUtils.doPostWithHeaders(CEServerTestSuite.GQL_API_URL, input, client, headers);
+        return JSONUtils.getObjectOrNull(map, "data");
+    }
+
+
 }

server/test/io.cloudbeaver.test.platform/workspace/conf/cloudbeaver.conf

@@ -1,6 +1,7 @@
 {
     server: {
         serverPort: "${CLOUDBEAVER_TEST_PORT:18978}",
+        serverName: "CloudBeaver CE Test Server",
 
         workspaceLocation: "workspace",
         contentRoot: "workspace/web",
@@ -36,8 +37,10 @@
         anonymousAccessEnabled: true,
         anonymousUserRole: "user",
         supportsCustomConnections: true,
+        enableReverseProxyAuth: true,
         enabledAuthProviders: [
-            "local"
+            "local",
+            "reverseProxy"
         ],
         enabledFeatures: [
             

server/test/io.cloudbeaver.test.platform/workspace/gql_scripts/activeUser.json

@@ -0,0 +1,4 @@
+{
+  "query": "\n    query activeUser {\n  user: activeUser {\n    userId\n  }\n}\n    ",
+  "operationName": "activeUser"
+}

server/test/io.cloudbeaver.test.platform/workspace/gql_scripts/openSession.json

@@ -0,0 +1,7 @@
+{
+  "query": "\n    mutation openSession($defaultLocale: String) {\n  session: openSession(defaultLocale: $defaultLocale) {\n    valid\n  }\n}\n    ",
+  "variables": {
+    "defaultLocale": "en"
+  },
+  "operationName": "openSession"
+}