CopilotForXcode/Tool/Tests/SuggestionBasicTests/TextExtrationFromCodeTests.swift at 0.41.135 · github/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
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
import Foundation
import XCTest
@testable import SuggestionBasic

final class TextExtrationFromCodeTests: XCTestCase {
    func test_empty_selection() {
        let selection = CursorRange(
            start: CursorPosition(line: 0, character: 0),
            end: CursorPosition(line: 0, character: 0)
        )
        let lines = ["let foo = 1\n", "let bar = 2\n"]
        let result = EditorInformation.code(
            in: lines,
            inside: selection,
            ignoreColumns: false
        )
        XCTAssertEqual(result.code, "")
        XCTAssertEqual(result.lines, ["let foo = 1\n"])
    }

    func test_single_line_selection() {
        let selection = CursorRange(
            start: CursorPosition(line: 0, character: 4),
            end: CursorPosition(line: 0, character: 10)
        )
        let lines = ["let foo = 1\n", "let bar = 2\n"]
        let result = EditorInformation.code(
            in: lines,
            inside: selection,
            ignoreColumns: false
        )
        XCTAssertEqual(result.code, "foo = ")
        XCTAssertEqual(result.lines, ["let foo = 1\n"])
    }

    func test_single_line_selection_with_emoji() {
        let selection = CursorRange(
            start: CursorPosition(line: 0, character: 4),
            end: CursorPosition(line: 0, character: 10)
        )
        let lines = ["let 🎆🎆o = 1\n", "let bar = 2\n"]
        let result = EditorInformation.code(
            in: lines,
            inside: selection,
            ignoreColumns: false
        )
        XCTAssertEqual(result.code, "🎆🎆o ")
        XCTAssertEqual(result.lines, ["let 🎆🎆o = 1\n"])
    }

    func test_single_line_selection_cutting_emoji() {
        // undefined behavior

        let selection = CursorRange(
            start: CursorPosition(line: 0, character: 5),
            end: CursorPosition(line: 0, character: 10)
        )
        let lines = ["let 🎆🎆o = 1\n", "let bar = 2\n"]
        let result = EditorInformation.code(
            in: lines,
            inside: selection,
            ignoreColumns: false
        )
        XCTAssertEqual(result.lines, ["let 🎆🎆o = 1\n"])
    }

    func test_single_line_selection_at_line_end() {
        let selection = CursorRange(
            start: CursorPosition(line: 0, character: 8),
            end: CursorPosition(line: 0, character: 11)
        )
        let lines = ["let foo = 1\n", "let bar = 2\n"]
        let result = EditorInformation.code(
            in: lines,
            inside: selection,
            ignoreColumns: false
        )
        XCTAssertEqual(result.code, "= 1")
        XCTAssertEqual(result.lines, ["let foo = 1\n"])
    }

    func test_multi_line_selection() {
        let selection = CursorRange(
            start: CursorPosition(line: 0, character: 4),
            end: CursorPosition(line: 1, character: 11)
        )
        let lines = ["let foo = 1\n", "let bar = 2\n", "let baz = 3\n"]
        let result = EditorInformation.code(
            in: lines,
            inside: selection,
            ignoreColumns: false
        )
        XCTAssertEqual(result.code, "foo = 1\nlet bar = 2")
        XCTAssertEqual(result.lines, ["let foo = 1\n", "let bar = 2\n"])
    }

    func test_multi_line_selection_with_emoji() {
        let selection = CursorRange(
            start: CursorPosition(line: 0, character: 4),
            end: CursorPosition(line: 1, character: 11)
        )
        let lines = ["🎆🎆 foo = 1\n", "let bar = 2\n", "let baz = 3\n"]
        let result = EditorInformation.code(
            in: lines,
            inside: selection,
            ignoreColumns: false
        )
        XCTAssertEqual(result.code, " foo = 1\nlet bar = 2")
        XCTAssertEqual(result.lines, ["🎆🎆 foo = 1\n", "let bar = 2\n"])
    }

    func test_invalid_selection() {
        let selection = CursorRange(
            start: CursorPosition(line: 1, character: 4),
            end: CursorPosition(line: 0, character: 10)
        )
        let lines = ["let foo = 1", "let bar = 2"]
        let result = EditorInformation.code(
            in: lines,
            inside: selection,
            ignoreColumns: false
        )
        XCTAssertEqual(result.code, "")
        XCTAssertEqual(result.lines, [])
    }

    func test_single_line_selection_ignoring_column() {
        let selection = CursorRange(
            start: CursorPosition(line: 0, character: 4),
            end: CursorPosition(line: 0, character: 10)
        )
        let lines = ["let foo = 1\n", "let bar = 2\n"]
        let result = EditorInformation.code(
            in: lines,
            inside: selection,
            ignoreColumns: true
        )
        XCTAssertEqual(result.code, "let foo = 1\n")
        XCTAssertEqual(result.lines, ["let foo = 1\n"])
    }

    func test_multi_line_selection_ignoring_column() {
        let selection = CursorRange(
            start: CursorPosition(line: 0, character: 4),
            end: CursorPosition(line: 1, character: 11)
        )
        let lines = ["let foo = 1\n", "let bar = 2\n", "let baz = 3\n"]
        let result = EditorInformation.code(
            in: lines,
            inside: selection,
            ignoreColumns: true
        )
        XCTAssertEqual(result.code, "let foo = 1\nlet bar = 2\n")
        XCTAssertEqual(result.lines, ["let foo = 1\n", "let bar = 2\n"])
    }
}