1 /**
2 * Copyright (c) 2025 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include <cstddef>
17 #include <string>
18 #include <vector>
19 #include "lsp_api_test.h"
20 #include "lsp/include/internal_api.h"
21 #include "public/es2panda_lib.h"
22
23 using ark::es2panda::lsp::Initializer;
24
25 class LSPGetFileReferencesTests : public LSPAPITests {
26 public:
27 LSPGetFileReferencesTests() = default;
28 ~LSPGetFileReferencesTests() override = default;
29
30 NO_COPY_SEMANTIC(LSPGetFileReferencesTests);
31 NO_MOVE_SEMANTIC(LSPGetFileReferencesTests);
32
MockGetFileReferences(char const * fileName,const std::vector<std::string> & filePaths)33 References MockGetFileReferences(char const *fileName, const std::vector<std::string> &filePaths)
34 {
35 Initializer initializer = Initializer();
36 auto context = initializer.CreateContext(fileName, ES2PANDA_STATE_CHECKED);
37 bool isPackageModule =
38 reinterpret_cast<ark::es2panda::public_lib::Context *>(context)->parserProgram->IsPackage();
39 initializer.DestroyContext(context);
40 References result {};
41 for (auto const &file : filePaths) {
42 auto referenceContext = initializer.CreateContext(file.c_str(), ES2PANDA_STATE_CHECKED);
43 auto refInfo = ark::es2panda::lsp::GetFileReferencesImpl(referenceContext, fileName, isPackageModule);
44 result.referenceInfos.insert(result.referenceInfos.end(), refInfo.referenceInfos.begin(),
45 refInfo.referenceInfos.end());
46 initializer.DestroyContext(referenceContext);
47 }
48 return result;
49 }
50 };
51
TEST_F(LSPGetFileReferencesTests,GetFileReferences1)52 TEST_F(LSPGetFileReferencesTests, GetFileReferences1)
53 {
54 std::vector<std::string> files = {"refer-1.ets", "file-references1.ets"};
55 std::vector<std::string> texts = {
56 R"(export function A(a:number, b:number): number {
57 return a + b;
58 }
59 export function B(a:number, b:number): number {
60 return a + b;
61 })",
62 R"(import {A} from "./refer-1";
63 import {B} from "./refer-1.ets";
64 A(1, 2);
65 B(1, 2);)"};
66 auto filePaths = CreateTempFile(files, texts);
67 int const expectedFileCount = 2;
68 ASSERT_EQ(filePaths.size(), expectedFileCount);
69
70 References result = MockGetFileReferences(filePaths[0].c_str(), filePaths);
71 // NOLINTBEGIN(readability-magic-numbers)
72 std::vector<ReferenceInfo> expectedResult {{filePaths[1], 16, 11}, {filePaths[1], 45, 15}};
73 // NOLINTEND(readability-magic-numbers)
74 ASSERT_EQ(result.referenceInfos.size(), expectedResult.size());
75 for (size_t i = 0; i < expectedResult.size(); i++) {
76 ASSERT_EQ(result.referenceInfos[i].fileName, expectedResult[i].fileName);
77 ASSERT_EQ(result.referenceInfos[i].start, expectedResult[i].start);
78 ASSERT_EQ(result.referenceInfos[i].length, expectedResult[i].length);
79 }
80 }
81
TEST_F(LSPGetFileReferencesTests,GetFileReferences2)82 TEST_F(LSPGetFileReferencesTests, GetFileReferences2)
83 {
84 std::vector<std::string> files = {"refer-2.ts", "file-references2.ets"};
85 std::vector<std::string> texts = {
86 R"(export function A(a:number, b:number): number {
87 return a + b;
88 }
89 export function B(a:number, b:number): number {
90 return a + b;
91 })",
92 R"(import {A} from "./refer-2";
93 import {B} from "./refer-2.ts";
94 A(1, 2);
95 B(1, 2);)"};
96 auto filePaths = CreateTempFile(files, texts);
97 int const expectedFileCount = 2;
98 ASSERT_EQ(filePaths.size(), expectedFileCount);
99
100 References result = MockGetFileReferences(filePaths[0].c_str(), filePaths);
101 // NOLINTBEGIN(readability-magic-numbers)
102 std::vector<ReferenceInfo> expectedResult {{filePaths[1], 16, 11}, {filePaths[1], 45, 14}};
103 // NOLINTEND(readability-magic-numbers)
104 ASSERT_EQ(result.referenceInfos.size(), expectedResult.size());
105 for (size_t i = 0; i < expectedResult.size(); i++) {
106 ASSERT_EQ(result.referenceInfos[i].fileName, expectedResult[i].fileName);
107 ASSERT_EQ(result.referenceInfos[i].start, expectedResult[i].start);
108 ASSERT_EQ(result.referenceInfos[i].length, expectedResult[i].length);
109 }
110 }
111