fix(lsp): refactor, fix issues and add benchmark for code lens (#10841)

2 files changed, 98 insertions, 0 deletions

diff --git a/cli/tests/lsp/code_lens_response_changed.json b/cli/tests/lsp/code_lens_response_changed.json
new file mode 100644
index 000000000..b0073a23f
--- /dev/null
+++ b/cli/tests/lsp/code_lens_response_changed.json
@@ -0,0 +1,50 @@
+[
+  {
+    "range": {
+      "start": {
+        "line": 0,
+        "character": 10
+      },
+      "end": {
+        "line": 0,
+        "character": 11
+      }
+    },
+    "data": {
+      "specifier": "file:///a/file.ts",
+      "source": "implementations"
+    }
+  },
+  {
+    "range": {
+      "start": {
+        "line": 0,
+        "character": 10
+      },
+      "end": {
+        "line": 0,
+        "character": 11
+      }
+    },
+    "data": {
+      "specifier": "file:///a/file.ts",
+      "source": "references"
+    }
+  },
+  {
+    "range": {
+      "start": {
+        "line": 4,
+        "character": 6
+      },
+      "end": {
+        "line": 4,
+        "character": 7
+      }
+    },
+    "data": {
+      "specifier": "file:///a/file.ts",
+      "source": "references"
+    }
+  }
+]
diff --git a/cli/tests/lsp/code_lens_response_impl.json b/cli/tests/lsp/code_lens_response_impl.json
index b0073a23f..c6e5bd92d 100644
--- a/cli/tests/lsp/code_lens_response_impl.json
+++ b/cli/tests/lsp/code_lens_response_impl.json
@@ -46,5 +46,53 @@
       "specifier": "file:///a/file.ts",
       "source": "references"
     }
+  },
+  {
+    "range": {
+      "start": {
+        "line": 10,
+        "character": 10
+      },
+      "end": {
+        "line": 10,
+        "character": 11
+      }
+    },
+    "data": {
+      "specifier": "file:///a/file.ts",
+      "source": "implementations"
+    }
+  },
+  {
+    "range": {
+      "start": {
+        "line": 10,
+        "character": 10
+      },
+      "end": {
+        "line": 10,
+        "character": 11
+      }
+    },
+    "data": {
+      "specifier": "file:///a/file.ts",
+      "source": "references"
+    }
+  },
+  {
+    "range": {
+      "start": {
+        "line": 11,
+        "character": 2
+      },
+      "end": {
+        "line": 11,
+        "character": 3
+      }
+    },
+    "data": {
+      "specifier": "file:///a/file.ts",
+      "source": "references"
+    }
   }
 ]