1 files changed, 3 insertions, 1 deletions
diff --git a/tst/email.spec.ts b/tst/email.spec.ts
index 7315816..e966030 100644
--- a/tst/email.spec.ts
+++ b/tst/email.spec.ts
@@ -31,7 +31,7 @@ const getMocks = () => {
     connect: mock(() => Promise.resolve()),
     getMailboxLock: mock(() => Promise.resolve(lock)),
     messageDelete: mock(() => Promise.resolve(true)),
-    close: mock(() => constVoid()),
+    logout: mock(() => Promise.resolve()),
   };
 
   const mockDependencies: Partial<EmailJobDependencies> = {
@@ -116,6 +116,7 @@ test("releases lock on left", async () => {
     TE.mapLeft(() => {
       expect(imap.getMailboxLock).toHaveBeenCalledTimes(1);
       expect(lock.release).toHaveBeenCalledTimes(1);
+      expect(imap.logout).toHaveBeenCalledTimes(1);
     }),
   )();
 });
@@ -131,6 +132,7 @@ test("releases lock on right", async () => {
     TE.map(() => {
       expect(imap.getMailboxLock).toHaveBeenCalledTimes(1);
       expect(lock.release).toHaveBeenCalledTimes(1);
+      expect(imap.logout).toHaveBeenCalledTimes(1);
     }),
     TE.mapLeft(() => expect(false).toBeTruthy()),
   )();