tutanota/test/tests/api/common/mail/CommonMailUtilsTest.ts

import o from "@tutao/otest"
import { createEncryptedMailAddress, createMail, createMailAddress, Mail, MailTypeRef } from "../../../../../src/api/entities/tutanota/TypeRefs.js"
import { EncryptionAuthStatus, MailAuthenticationStatus, MailState } from "../../../../../src/api/common/TutanotaConstants.js"
import { downcast } from "@tutao/tutanota-utils"
import { getDisplayedSender, isTutanotaTeamAddress, MailAddressAndName } from "../../../../../src/api/common/mail/CommonMailUtils.js"
import { createTestEntity } from "../../../TestUtils.js"
import { getConfidentialIcon } from "../../../../../src/mail/model/MailUtils.js"
import { Icons } from "../../../../../src/gui/base/icons/Icons.js"
import { ProgrammingError } from "../../../../../src/api/common/error/ProgrammingError.js"

o.spec("MailUtilsTest", function () {
	function createSystemMail(realSender: MailAddressAndName): Mail {
		return createMail({
			sender: createMailAddress({
				address: "system@tutanota.de",
				name: "System",
				_id: "",
				_ownerGroup: "",
				contact: null,
			}),
			replyTos: [createEncryptedMailAddress(realSender)],
			state: MailState.RECEIVED,
			authStatus: MailAuthenticationStatus.AUTHENTICATED,
			_errors: {},
			_id: ["", ""],
			_ownerEncSessionKey: null,
			_ownerGroup: "",
			_permissions: "",
			attachments: [],
			bccRecipients: [],
			body: null,
			bucketKey: null,
			ccRecipients: [],
			confidential: false,
			conversationEntry: ["", ""],
			differentEnvelopeSender: null,
			encryptionAuthStatus: null,
			firstRecipient: null,
			headers: null,
			listUnsubscribe: false,
			mailDetails: null,
			mailDetailsDraft: null,
			method: "",
			movedTime: null,
			phishingStatus: "",
			receivedDate: new Date(),
			recipientCount: "",
			replyType: "",
			sentDate: null,
			subject: "",
			toRecipients: [],
			unread: false,
		})
	}

	o("getDisplayedSender", function () {
		let mail: Mail = downcast("placeholder that won't get used")

		const createSalesMail = () => (mail = createSystemMail({ address: "sales@tutao.de", name: "Sales" }))
		const assertDisplayedSender = (address: string, name: string) => o(getDisplayedSender(mail)).deepEquals({ address, name })
		const assertIsSales = () => assertDisplayedSender("sales@tutao.de", "Sales")
		const assertIsNoReply = () => assertDisplayedSender("no-reply@tutao.de", "Do not reply to this!!!")
		const assertIsSystem = () => assertDisplayedSender("system@tutanota.de", "System")

		// Should be good
		createSalesMail()
		assertIsSales()
		createSalesMail().replyTos = [createEncryptedMailAddress({ address: "no-reply@tutao.de", name: "Do not reply to this!!!" })]
		assertIsNoReply()

		// Not authenticated
		createSalesMail().authStatus = MailAuthenticationStatus.HARD_FAIL
		assertIsSystem()

		// Not received
		createSalesMail().state = MailState.DRAFT
		assertIsSystem()

		// Multiple reply-tos, can't determine which
		createSalesMail().replyTos = [mail.replyTos[0], mail.replyTos[1]]
		assertIsSystem()

		// Not a Tutao address
		createSalesMail().replyTos = [createEncryptedMailAddress({ address: "bed-free@tutanota.de", name: "Bernd das Brot" })]
		assertIsSystem()
	})

	o("isTutanotaTeamAddress", function () {
		o(isTutanotaTeamAddress("system@tutanota.de")).deepEquals(false)
		o(isTutanotaTeamAddress("bed-free@tutanota.de")).deepEquals(false)
		o(isTutanotaTeamAddress("admin@somesite.com")).deepEquals(false)
		o(isTutanotaTeamAddress("no-reply@tutao.de")).deepEquals(true)
		o(isTutanotaTeamAddress("sales@tutao.de")).deepEquals(true)
		o(isTutanotaTeamAddress("no-reply@tutanota.de")).deepEquals(true)
	})

	o("getConfidentialIcon", function () {
		const mail: Mail = createTestEntity(MailTypeRef, { confidential: true, encryptionAuthStatus: EncryptionAuthStatus.PQ_AUTHENTICATION_SUCCEEDED })
		o(getConfidentialIcon(mail)).equals(Icons.PQLock)

		mail.encryptionAuthStatus = EncryptionAuthStatus.PQ_AUTHENTICATION_FAILED
		o(getConfidentialIcon(mail)).equals(Icons.PQLock)

		mail.encryptionAuthStatus = EncryptionAuthStatus.AES_NO_AUTHENTICATION
		o(getConfidentialIcon(mail)).equals(Icons.Lock)

		mail.encryptionAuthStatus = null
		o(getConfidentialIcon(mail)).equals(Icons.Lock)

		mail.encryptionAuthStatus = EncryptionAuthStatus.RSA_NO_AUTHENTICATION
		o(getConfidentialIcon(mail)).equals(Icons.Lock)

		mail.confidential = false
		o(() => getConfidentialIcon(mail)).throws(ProgrammingError)
	})
})