CopilotForXcode/Tool/Tests/GitHubCopilotServiceTests/FetchSuggestionsTests.swift at main · GoodOpenRepo/CopilotForXcode · 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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
import LanguageServerProtocol
import XCTest

@testable import GitHubCopilotService

final class FetchSuggestionTests: XCTestCase {
    func test_process_suggestions_from_server() async throws {
        struct TestServer: GitHubCopilotLSP {
            func sendNotification(_ notif: LanguageServerProtocol.ClientNotification) async throws {
                fatalError()
            }

            func sendRequest<E>(_: E) async throws -> E.Response where E: GitHubCopilotRequestType {
                return GitHubCopilotRequest.GetCompletionsCycling.Response(completions: [
                    .init(
                        text: "Hello World\n",
                        position: .init((0, 0)),
                        uuid: "uuid",
                        range: .init(start: .init((0, 0)), end: .init((0, 4))),
                        displayText: "Hello"
                    ),
                    .init(
                        text: " ",
                        position: .init((0, 0)),
                        uuid: "uuid",
                        range: .init(start: .init((0, 0)), end: .init((0, 1))),
                        displayText: " "
                    ),
                    .init(
                        text: " \n",
                        position: .init((0, 0)),
                        uuid: "uuid",
                        range: .init(start: .init((0, 0)), end: .init((0, 2))),
                        displayText: " \n"
                    ),
                ]) as! E.Response
            }
        }
        let service = GitHubCopilotSuggestionService(designatedServer: TestServer())
        let completions = try await service.getCompletions(
            fileURL: .init(fileURLWithPath: "/file.swift"),
            content: "",
            cursorPosition: .outOfScope,
            tabSize: 4,
            indentSize: 4,
            usesTabsForIndentation: false,
            ignoreSpaceOnlySuggestions: false,
            ignoreTrailingNewLinesAndSpaces: false
        )
        XCTAssertEqual(completions.count, 3)
    }

    func test_ignore_empty_suggestions() async throws {
        struct TestServer: GitHubCopilotLSP {
            func sendNotification(_ notif: LanguageServerProtocol.ClientNotification) async throws {
                fatalError()
            }

            func sendRequest<E>(_: E) async throws -> E.Response where E: GitHubCopilotRequestType {
                return GitHubCopilotRequest.GetCompletionsCycling.Response(completions: [
                    .init(
                        text: "Hello World\n",
                        position: .init((0, 0)),
                        uuid: "uuid",
                        range: .init(start: .init((0, 0)), end: .init((0, 4))),
                        displayText: "Hello"
                    ),
                    .init(
                        text: " ",
                        position: .init((0, 0)),
                        uuid: "uuid",
                        range: .init(start: .init((0, 0)), end: .init((0, 1))),
                        displayText: " "
                    ),
                    .init(
                        text: " \n",
                        position: .init((0, 0)),
                        uuid: "uuid",
                        range: .init(start: .init((0, 0)), end: .init((0, 2))),
                        displayText: " \n"
                    ),
                ]) as! E.Response
            }
        }
        let service = GitHubCopilotSuggestionService(designatedServer: TestServer())
        let completions = try await service.getCompletions(
            fileURL: .init(fileURLWithPath: "/file.swift"),
            content: "",
            cursorPosition: .outOfScope,
            tabSize: 4,
            indentSize: 4,
            usesTabsForIndentation: false,
            ignoreSpaceOnlySuggestions: true,
            ignoreTrailingNewLinesAndSpaces: false
        )
        XCTAssertEqual(completions.count, 1)
        XCTAssertEqual(completions.first?.text, "Hello World\n")
    }

    func test_if_language_identifier_is_unknown_returns_correctly() async throws {
        struct Err: Error, LocalizedError {
            var errorDescription: String? {
                "sendRequest Should not be falled"
            }
        }

        class TestServer: GitHubCopilotLSP {
            func sendNotification(_ notif: LanguageServerProtocol.ClientNotification) async throws {
                // unimplemented
            }

            func sendRequest<E>(_ r: E) async throws -> E.Response where E: GitHubCopilotRequestType {
                return GitHubCopilotRequest.GetCompletionsCycling.Response(completions: [
                    .init(
                        text: "Hello World\n",
                        position: .init((0, 0)),
                        uuid: "uuid",
                        range: .init(start: .init((0, 0)), end: .init((0, 4))),
                        displayText: "Hello"
                    ),
                ]) as! E.Response
            }
        }
        let testServer = TestServer()
        let service = GitHubCopilotSuggestionService(designatedServer: testServer)
        let completions = try await service.getCompletions(
            fileURL: .init(fileURLWithPath: "/"),
            content: "",
            cursorPosition: .outOfScope,
            tabSize: 4,
            indentSize: 4,
            usesTabsForIndentation: false,
            ignoreSpaceOnlySuggestions: false,
            ignoreTrailingNewLinesAndSpaces: true
        )
        XCTAssertEqual(completions.count, 1)
        XCTAssertEqual(completions.first?.text, "Hello World")
    }
}