Adds extra event to tests

neo4j · Dec 21, 2024 · 961e357 · 961e357
1 parent 9d76d04
commit 961e357
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/packages/vscode-extension/tests/specs/unit/connectionService.spec.ts b/packages/vscode-extension/tests/specs/unit/connectionService.spec.ts
@@ -193,7 +193,7 @@ suite('Connection service spec', () => {
       assert.strictEqual(returnedPassword2, 'mock-password-2');
     });
 
-    test('Should notify language server when a Connection is deleted', async () => {
+    test.only('Should notify language server when a Connection is deleted', async () => {
       const sendNotificationSpy = sandbox.spy(
         mockLanguageClient,
         'sendNotification',
@@ -203,16 +203,22 @@ suite('Connection service spec', () => {
         mockConnection,
         'mock-password',
       );
-
+      sendNotificationSpy.resetHistory();
       await connection.deleteConnectionAndUpdateDatabaseConnection(
         mockConnection.key,
       );
 
+      sandbox.assert.calledTwice(sendNotificationSpy);
       sandbox.assert.calledWithExactly(
-        sendNotificationSpy,
+        sendNotificationSpy.getCall(0),
         'connectionDisconnected',
         undefined,
       );
+      sandbox.assert.calledWith(
+        sendNotificationSpy.getCall(1),
+        'relintDocuments',
+        undefined,
+      );
     });
   });