Fix InlayHintTests

We need to include text edits in the expected inlay hints.
swiftlang · May 24, 2022 · 670a696 · 670a696
1 parent 8c2beb5
commit 670a696
Showing 1 changed file with 24 additions and 13 deletions.
diff --git a/Tests/SourceKitLSPTests/InlayHintTests.swift b/Tests/SourceKitLSPTests/InlayHintTests.swift
@@ -61,6 +61,17 @@ final class InlayHintTests: XCTestCase {
     }
   }
 
+  private func makeInlayHint(position: Position, kind: InlayHintKind, label: String) -> InlayHint {
+    InlayHint(
+      position: position,
+      kind: kind,
+      label: .string(label),
+      textEdits: [
+        TextEdit(range: position..<position, newText: label)
+      ]
+    )
+  }
+
   func testEmpty() throws {
     let text = ""
     let hints = try performInlayHintRequest(text: text)
@@ -74,12 +85,12 @@ final class InlayHintTests: XCTestCase {
     """
     let hints = try performInlayHintRequest(text: text)
     XCTAssertEqual(hints, [
-      InlayHint(
+      makeInlayHint(
         position: Position(line: 0, utf16index: 5),
         kind: .type,
         label: ": Int"
       ),
-      InlayHint(
+      makeInlayHint(
         position: Position(line: 1, utf16index: 5),
         kind: .type,
         label: ": String"
@@ -103,12 +114,12 @@ final class InlayHintTests: XCTestCase {
     let range = Position(line: 6, utf16index: 0)..<Position(line: 9, utf16index: 0)
     let hints = try performInlayHintRequest(text: text, range: range)
     XCTAssertEqual(hints, [
-      InlayHint(
+      makeInlayHint(
         position: Position(line: 6, utf16index: 10),
         kind: .type,
         label: ": Bool"
       ),
-      InlayHint(
+      makeInlayHint(
         position: Position(line: 7, utf16index: 12),
         kind: .type,
         label: ": Int"
@@ -134,27 +145,27 @@ final class InlayHintTests: XCTestCase {
     """
     let hints = try performInlayHintRequest(text: text)
     XCTAssertEqual(hints, [
-      InlayHint(
+      makeInlayHint(
         position: Position(line: 1, utf16index: 20),
         kind: .type,
         label: ": Int"
       ),
-      InlayHint(
+      makeInlayHint(
         position: Position(line: 2, utf16index: 25),
         kind: .type,
         label: ": String"
       ),
-      InlayHint(
+      makeInlayHint(
         position: Position(line: 6, utf16index: 20),
         kind: .type,
         label: ": String"
       ),
-      InlayHint(
+      makeInlayHint(
         position: Position(line: 7, utf16index: 25),
         kind: .type,
         label: ": Int"
       ),
-      InlayHint(
+      makeInlayHint(
         position: Position(line: 11, utf16index: 25),
         kind: .type,
         label: ": Double"
@@ -186,22 +197,22 @@ final class InlayHintTests: XCTestCase {
     """
     let hints = try performInlayHintRequest(text: text)
     XCTAssertEqual(hints, [
-      InlayHint(
+      makeInlayHint(
         position: Position(line: 2, utf16index: 5),
         kind: .type,
         label: ": (Int) -> ()"
       ),
-      InlayHint(
+      makeInlayHint(
         position: Position(line: 3, utf16index: 31),
         kind: .type,
         label: ": String"
       ),
-      InlayHint(
+      makeInlayHint(
         position: Position(line: 4, utf16index: 40),
         kind: .type,
         label: ": Double"
       ),
-      InlayHint(
+      makeInlayHint(
         position: Position(line: 4, utf16index: 43),
         kind: .type,
         label: ": Double"