@@ -41,6 +41,15 @@ describe('SalesforceApexContentProvider', () => {
41
41
provider = new SalesforceApexContentProvider ( mockConnectionProvider , mockLogger ) ;
42
42
} ) ;
43
43
44
+ // Mock timers to control async behavior
45
+ beforeAll ( ( ) => {
46
+ jest . useFakeTimers ( ) ;
47
+ } ) ;
48
+
49
+ afterAll ( ( ) => {
50
+ jest . useRealTimers ( ) ;
51
+ } ) ;
52
+
44
53
describe ( 'provideTextDocumentContent' , ( ) => {
45
54
it ( 'should return class body for valid class name URI' , async ( ) => {
46
55
// Arrange
@@ -55,6 +64,7 @@ describe('SalesforceApexContentProvider', () => {
55
64
56
65
// Act
57
66
const result = await provider . provideTextDocumentContent ( uri ) ;
67
+ jest . runAllTimers ( ) ; // Ensure all promises are resolved
58
68
59
69
// Assert
60
70
expect ( result ) . toBe ( expectedBody ) ;
@@ -78,6 +88,7 @@ describe('SalesforceApexContentProvider', () => {
78
88
79
89
// Act
80
90
const result = await provider . provideTextDocumentContent ( uri ) ;
91
+ jest . runAllTimers ( ) ; // Ensure all promises are resolved
81
92
82
93
// Assert
83
94
expect ( result ) . toBe ( expectedBody ) ;
@@ -95,6 +106,7 @@ describe('SalesforceApexContentProvider', () => {
95
106
96
107
// Act
97
108
const result = await provider . provideTextDocumentContent ( uri ) ;
109
+ jest . runAllTimers ( ) ; // Ensure all promises are resolved
98
110
99
111
// Assert
100
112
expect ( result ) . toContain ( '// Error fetching Apex class' ) ;
@@ -113,6 +125,7 @@ describe('SalesforceApexContentProvider', () => {
113
125
114
126
// Act
115
127
const result = await provider . provideTextDocumentContent ( uri ) ;
128
+ jest . runAllTimers ( ) ; // Ensure all promises are resolved
116
129
117
130
// Assert
118
131
expect ( result ) . toContain ( '// Error fetching Apex class' ) ;
@@ -129,6 +142,7 @@ describe('SalesforceApexContentProvider', () => {
129
142
130
143
// Act
131
144
const result = await provider . provideTextDocumentContent ( uri , cancellationToken ) ;
145
+ jest . runAllTimers ( ) ; // Ensure all promises are resolved
132
146
133
147
// Assert
134
148
expect ( result ) . toContain ( '// Error fetching Apex class' ) ;
@@ -150,6 +164,7 @@ describe('SalesforceApexContentProvider', () => {
150
164
// Act
151
165
const result1 = await provider . provideTextDocumentContent ( uri ) ;
152
166
const result2 = await provider . provideTextDocumentContent ( uri ) ;
167
+ jest . runAllTimers ( ) ; // Ensure all promises are resolved
153
168
154
169
// Assert
155
170
expect ( result1 ) . toBe ( expectedBody ) ;
0 commit comments