From 1c22a425bdc20f2abd56f3ea8cfb6387e375bd58 Mon Sep 17 00:00:00 2001
From: yamashush <38120991+yamashush@users.noreply.github.com>
Date: Fri, 19 Jan 2024 12:21:27 +0900
Subject: [PATCH] test: complete todo

Signed-off-by: yamashush <38120991+yamashush@users.noreply.github.com>
---
 backend/src/history/history.service.spec.ts | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/backend/src/history/history.service.spec.ts b/backend/src/history/history.service.spec.ts
index 9988de0b6..ff67f580b 100644
--- a/backend/src/history/history.service.spec.ts
+++ b/backend/src/history/history.service.spec.ts
@@ -393,15 +393,18 @@ describe('HistoryService', () => {
         updatedAt: historyEntryImport.lastVisitedAt,
       };
 
-      const createQueryBuilder = mockSelectQueryBuilderInRepo(noteRepo, note);
+      mockSelectQueryBuilderInRepo(noteRepo, note);
+      const createQueryBuilderForEntityManager = {
+        where: () => createQueryBuilderForEntityManager,
+        getMany: () => [historyEntry],
+      };
+
       const mockedManager = Mock.of<EntityManager>({
-        find: jest.fn().mockResolvedValueOnce([historyEntry]),
-        createQueryBuilder: () => createQueryBuilder,
-        remove: jest.fn().mockImplementationOnce(async (_: HistoryEntry) => {
-          // TODO: reimplement checks below
-          //expect(await (await entry.note).aliases).toHaveLength(1);
-          //expect((await (await entry.note).aliases)[0].name).toEqual(alias);
-          //expect(entry.pinStatus).toEqual(false);
+        createQueryBuilder: jest.fn().mockImplementation(() => createQueryBuilderForEntityManager),
+        remove: jest.fn().mockImplementationOnce(async (entry: HistoryEntry) => {
+          expect(await (await entry.note).aliases).toHaveLength(1);
+          expect((await (await entry.note).aliases)[0].name).toEqual(alias);
+          expect(entry.pinStatus).toEqual(false);
         }),
         save: jest.fn().mockImplementationOnce(async (entry: HistoryEntry) => {
           expect((await entry.note).aliases).toEqual(