tutanota/test/tests/mail/MailModelTest.ts

import o from "ospec"
import {Notifications} from "../../../src/gui/Notifications.js"
import type {Spy} from "@tutao/tutanota-test-utils"
import {spy} from "@tutao/tutanota-test-utils"
import type {MailboxDetail} from "../../../src/mail/model/MailModel.js"
import {MailModel} from "../../../src/mail/model/MailModel.js"
import {MailFolderType, OperationType} from "../../../src/api/common/TutanotaConstants.js"
import {createMailFolder, MailTypeRef} from "../../../src/api/entities/tutanota/TypeRefs.js"
import type {EntityUpdateData} from "../../../src/api/main/EventController.js"
import {EntityClient} from "../../../src/api/common/EntityClient.js"
import {EntityRestClientMock} from "../api/worker/rest/EntityRestClientMock.js"
import nodemocker from "../nodemocker.js"
import {downcast} from "@tutao/tutanota-utils"
import {WorkerClient} from "../../../src/api/main/WorkerClient.js"
import {MailFacade} from "../../../src/api/worker/facades/MailFacade.js"
import {LoginController} from "../../../src/api/main/LoginController.js"
import {object} from "testdouble"
import {FolderSystem} from "../../../src/mail/model/FolderSystem.js"

o.spec("MailModelTest", function () {
	let notifications: Partial<Notifications>
	let showSpy: Spy
	let model: MailModel
	const inboxFolder = createMailFolder({_id: ["folderListId", "inboxId"]})
	inboxFolder.mails = "instanceListId"
	inboxFolder.folderType = MailFolderType.INBOX
	const anotherFolder = createMailFolder({_id: ["folderListId", "archiveId"]})
	anotherFolder.mails = "anotherListId"
	anotherFolder.folderType = MailFolderType.ARCHIVE
	let mailboxDetails: Partial<MailboxDetail>[]
	let logins: LoginController
	o.beforeEach(function () {
		mailboxDetails = [
			{
				folders: new FolderSystem([inboxFolder]),
			},
		]
		notifications = {}
		showSpy = notifications.showNotification = spy()
		const restClient = new EntityRestClientMock()
		const workerClient = nodemocker.mock<WorkerClient>("worker", {}).set()
		const mailFacade = nodemocker.mock<MailFacade>("mailFacade", {}).set()
		logins = object()
		model = new MailModel(
			downcast(notifications),
			downcast({}),
			workerClient,
			mailFacade,
			new EntityClient(restClient),
			logins,
		)
		// not pretty, but works
		model.mailboxDetails(mailboxDetails as MailboxDetail[])
	})
	// FIXME No way to inject entityRestClient for now
	// o("sends notification on new email in inbox", function () {
	// 	model.entityEventsReceived([
	// 		makeUpdate({
	// 			instanceListId: inboxFolder.mails,
	// 			operation: OperationType.CREATE
	// 		})
	// 	])
	// 	o(showSpy.invocations.length).equals(1)
	// })
	o("doesn't send notification for another folder", async function () {
		await model.entityEventsReceived([
			makeUpdate({
				instanceListId: anotherFolder.mails,
				operation: OperationType.CREATE,
			}),
		])
		o(showSpy.invocations.length).equals(0)
	})
	o("doesn't send notification for move operation", async function () {
		await model.entityEventsReceived([
			makeUpdate({
				instanceListId: anotherFolder.mails,
				operation: OperationType.DELETE,
			}),
			makeUpdate({
				instanceListId: inboxFolder.mails,
				operation: OperationType.CREATE,
			}),
		])
		o(showSpy.invocations.length).equals(0)
	})

	function makeUpdate(arg: {instanceListId: string; operation: OperationType}): EntityUpdateData {
		return Object.assign(
			{},
			{
				type: MailTypeRef.type,
				application: MailTypeRef.app,
				instanceId: "instanceId",
			},
			arg,
		)
	}
})