copilot-for-eclipse/com.microsoft.copilot.eclipse.ui.test/src/com/microsoft/copilot/eclipse/ui/completion/CompletionManagerTests.java at main · fvsgit/copilot-for-eclipse · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
// Copyright (c) Microsoft Corporation.
// Licensed under the MIT license.

package com.microsoft.copilot.eclipse.ui.completion;

import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IResource;
import org.eclipse.jface.text.IDocument;
import org.eclipse.lsp4e.LSPEclipseUtils;
import org.eclipse.lsp4j.Position;
import org.eclipse.lsp4j.Range;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.contexts.IContextService;
import org.eclipse.ui.texteditor.ITextEditor;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;

import com.microsoft.copilot.eclipse.core.completion.AcceptSuggestionType;
import com.microsoft.copilot.eclipse.core.completion.CompletionProvider;
import com.microsoft.copilot.eclipse.core.lsp.CopilotLanguageServerConnection;
import com.microsoft.copilot.eclipse.core.lsp.protocol.CompletionItem;
import com.microsoft.copilot.eclipse.core.lsp.protocol.CopilotLanguageServerSettings;
import com.microsoft.copilot.eclipse.ui.preferences.LanguageServerSettingManager;

@ExtendWith(MockitoExtension.class)
class CompletionManagerTests extends CompletionBaseTests {

  @Mock
  private CopilotLanguageServerConnection mockLsConnection;

  @Test
  void testReplaceCompletion1() throws Exception {
    IFile file = project.getFile("Test.java");
    String content = """
          public class App {

          public void hi() {
            System.out.println("");
          }

        }
        """;
    file.create(content.getBytes(), IResource.FORCE, null);
    int documentVersion = 1;

    IEditorPart editorPart = getEditorPartFor(file);
    assertTrue(editorPart instanceof ITextEditor);

    ITextEditor textEditor = (ITextEditor) editorPart;

    when(mockLsConnection.getDocumentVersion(any())).thenReturn(documentVersion);
    CopilotLanguageServerSettings settings = new CopilotLanguageServerSettings();
    LanguageServerSettingManager languageServerSettingManager = mock(LanguageServerSettingManager.class);
    when(languageServerSettingManager.getSettings()).thenReturn(settings);
    CompletionManager manager = new CompletionManager(mockLsConnection, mock(CompletionProvider.class), textEditor,
        languageServerSettingManager);

    List<CompletionItem> completions = List.of(new CompletionItem("uuid", "    System.out.println(\"hi\");",
        new Range(new Position(3, 0), new Position(3, 27)), "hi\");", new Position(3, 24), documentVersion));

    manager.onCompletionResolved(LSPEclipseUtils.toUri(file.getLocation().toFile()).toASCIIString(), completions);
    manager.acceptSuggestion(AcceptSuggestionType.FULL);

    IDocument document = textEditor.getDocumentProvider().getDocument(textEditor.getEditorInput());
    assertTrue(document.get().contains("  System.out.println(\"hi\");\n"));
  }

  @Test
  void testEnableCompletionContext() throws Exception {
    IFile file = project.getFile("Test2.java");
    String content = """
        public class Test2 {
          public void hi() {
            System.out.println("");
          }
        }
        """;
    file.create(content.getBytes(), IResource.FORCE, null);
    int documentVersion = 1;

    IEditorPart editorPart = getEditorPartFor(file);
    assertTrue(editorPart instanceof ITextEditor);

    ITextEditor textEditor = (ITextEditor) editorPart;

    when(mockLsConnection.getDocumentVersion(any())).thenReturn(documentVersion);
    CopilotLanguageServerSettings settings = new CopilotLanguageServerSettings();
    LanguageServerSettingManager languageServerSettingManager = mock(LanguageServerSettingManager.class);
    when(languageServerSettingManager.getSettings()).thenReturn(settings);
    CompletionManager manager = new CompletionManager(mockLsConnection, mock(CompletionProvider.class), textEditor,
        languageServerSettingManager);

    List<CompletionItem> completions = List.of(new CompletionItem("uuid", "    System.out.println(\"hi\");",
        new Range(new Position(2, 0), new Position(2, 27)), "hi\");", new Position(3, 24), documentVersion));

    manager.onCompletionResolved(LSPEclipseUtils.toUri(file.getLocation().toFile()).toASCIIString(), completions);

    IContextService contextService = PlatformUI.getWorkbench().getService(IContextService.class);
    Set<Object> activeContexts = new HashSet<>(contextService.getActiveContextIds());
    assertTrue(activeContexts.contains("com.microsoft.copilot.eclipse.completionAvailableContext"));
  }

}