dbeaver/pro#7618 added lsp tests to cb

server/features/io.cloudbeaver.server.feature/feature.xml

@@ -49,5 +49,6 @@
     <plugin id="org.jkiss.dbeaver.net.ssh" version="0.0.0"/>
     <plugin id="org.jkiss.dbeaver.net.ssh.sshj" version="0.0.0"/>
     <plugin id="org.jkiss.bundle.graphql" version="0.0.0"/>
+    <plugin id="org.jkiss.dbeaver.model.lsp" version="0.0.0"/>
 
 </feature>

server/test/io.cloudbeaver.test.platform/META-INF/MANIFEST.MF

@@ -14,6 +14,8 @@ Require-Bundle: org.eclipse.core.runtime,
  org.jkiss.dbeaver.model,
  org.jkiss.dbeaver.osgi.test.runner;visibility:=reexport,
  org.jkiss.dbeaver.model.sql,
+ org.jkiss.dbeaver.model.lsp,
+ org.jkiss.dbeaver.model.jdbc,
  org.jkiss.dbeaver.registry,
  org.jkiss.dbeaver.ext.generic,
  org.jkiss.dbeaver.ext.h2,
@@ -30,5 +32,9 @@ Require-Bundle: org.eclipse.core.runtime,
  org.jkiss.dbeaver.ext.mysql,
  org.jkiss.dbeaver.ext.postgresql,
  org.jkiss.dbeaver.ext.oracle,
- org.jkiss.dbeaver.ext.mssql
+ org.jkiss.dbeaver.ext.mssql,
+ org.jkiss.dbeaver.ext.h2,
+ org.jkiss.dbeaver.ext.generic,
+ org.eclipse.lsp4j,
+ org.eclipse.lsp4j.jsonrpc
 Export-Package: io.cloudbeaver.test.platform

server/test/io.cloudbeaver.test.platform/src/io/cloudbeaver/model/lsp/DBLTextDocumentServiceContextTest.java

@@ -0,0 +1,131 @@
+/*
+ * DBeaver - Universal Database Manager
+ * Copyright (C) 2010-2025 DBeaver Corp and others
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package io.cloudbeaver.model.lsp;
+
+import org.eclipse.lsp4j.*;
+import org.jkiss.dbeaver.ext.h2.model.H2SQLDialect;
+import org.jkiss.dbeaver.model.lsp.context.ContextAwareDocument;
+import org.junit.Assert;
+import org.junit.Test;
+
+import java.util.List;
+import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.ExecutionException;
+
+public class DBLTextDocumentServiceContextTest extends H2DataSourceTest {
+
+    @Test
+    public void shouldInitH2Context() {
+        TextDocumentItem document = DocumentServiceTestUtils.createAndSaveDocument(
+            service, "select * from table", project.getId(), DocumentServiceTestUtils.BASIC_RESOURCE_PATH
+        );
+
+        ContextAwareDocument contextedDocument = DocumentServiceTestUtils.getDocument(service, document.getUri());
+        Assert.assertNotNull(contextedDocument);
+        Assert.assertEquals(dataSourceDescriptor.getDataSource(), contextedDocument.getDataSource());
+        Assert.assertNotNull(contextedDocument.getExecutionContext());
+        Assert.assertEquals(dataSourceDescriptor.getDataSource(), contextedDocument.getExecutionContext().getDataSource());
+        Assert.assertTrue(contextedDocument.getSyntaxManager().getDialect() instanceof H2SQLDialect);
+        Assert.assertNotNull(contextedDocument.getRuleManager());
+    }
+
+    @Test
+    public void shouldFormatQuery() throws ExecutionException, InterruptedException {
+        String query = """
+            INSERT INTO users (id, profile) VALUES (1,'{"name": "JohnDoe"}'::jsonb) ON CONFLICT (id)
+            DO UPDATE SET profile = users.profile || EXCLUDED.profile RETURNING id, profile->>'name' AS name;
+            """.trim();
+        DocumentFormattingParams formattingParams = DocumentServiceTestUtils.setupDocumentAndBuildFormattingParams(service, query);
+
+        CompletableFuture<List<? extends TextEdit>> future = service.formatting(formattingParams);
+
+        TextEdit edit = future.get().getFirst();
+        String expectedQuery = """
+            INSERT
+                INTO
+                users (id,
+                profile)
+            VALUES (1,
+            '{"name": "JohnDoe"}'::jsonb) ON
+            CONFLICT (id)
+            DO
+            UPDATE
+            SET
+                profile = users.profile || EXCLUDED.profile RETURNING id,
+                profile->>'name' AS name;
+            """.trim();
+
+        Assert.assertEquals(expectedQuery.trim(), edit.getNewText());
+        Position start = edit.getRange().getStart();
+        Assert.assertEquals(0, start.getLine());
+        Assert.assertEquals(0, start.getCharacter());
+
+        Position end = edit.getRange().getEnd();
+        Assert.assertEquals(1, end.getLine());
+        Assert.assertEquals(97, end.getCharacter());
+    }
+
+    @Test
+    public void shouldReturnEmptyCompletionsForInvalidPosition() throws ExecutionException, InterruptedException {
+        String query = "SEL";
+        ContextAwareDocument document = DocumentServiceTestUtils.createAndSaveDocument(
+            service, query, project.getId(), DocumentServiceTestUtils.BASIC_RESOURCE_PATH
+        );
+        TextDocumentIdentifier documentId = new TextDocumentIdentifier(document.getUri());
+        CompletionParams completionParams = new CompletionParams(documentId, new Position(1, 42));
+
+        CompletionList completions = service.completion(completionParams).get().getRight();
+
+        Assert.assertNotNull(completions);
+        Assert.assertTrue(completions.getItems().isEmpty());
+    }
+
+    @Test
+    public void shouldSuggestKeywordCompletion() throws ExecutionException, InterruptedException {
+        String query = "SEL";
+        ContextAwareDocument document = DocumentServiceTestUtils.createAndSaveDocument(
+            service, query, project.getId(), DocumentServiceTestUtils.BASIC_RESOURCE_PATH
+        );
+        TextDocumentIdentifier documentId = new TextDocumentIdentifier(document.getUri());
+        CompletionParams completionParams = new CompletionParams(documentId, new Position(0, 3));
+
+        CompletionList completions = service.completion(completionParams).get().getRight();
+
+        Assert.assertNotNull(completions);
+        Assert.assertFalse(completions.getItems().isEmpty());
+        Assert.assertEquals("SELECT", completions.getItems().getFirst().getLabel());
+    }
+
+    @Test
+    public void shouldSuggestMultilineKeywordCompletion() throws ExecutionException, InterruptedException {
+        String query = """
+            SELECT *
+                FR
+            """;
+        ContextAwareDocument document = DocumentServiceTestUtils.createAndSaveDocument(
+            service, query, project.getId(), DocumentServiceTestUtils.BASIC_RESOURCE_PATH
+        );
+        TextDocumentIdentifier documentId = new TextDocumentIdentifier(document.getUri());
+        CompletionParams completionParams = new CompletionParams(documentId, new Position(1, 6));
+
+        CompletionList completions = service.completion(completionParams).get().getRight();
+
+        Assert.assertNotNull(completions);
+        Assert.assertEquals(1, completions.getItems().size());
+        Assert.assertEquals("FROM", completions.getItems().getFirst().getLabel());
+    }
+}