Stowage/tutanota
2
0
Fork 0
mirror of https://github.com/tutao/tutanota.git synced 2025-10-19 07:53:47 +00:00
Code Issues Projects Releases Packages Wiki Activity

wip: Add connectivity model to spam handler

Browse source
This commit is contained in:
sug 2025-10-14 18:23:57 +02:00 • committed by abp
parent 3f09b0f5db
commit 5b8b49d316
No known key found for this signature in database
GPG key ID: 791D4EC38A7AA7C2
5 changed files with 22 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

9
test/tests/mail/MailModelTest.ts
View file

@ -27,6 +27,7 @@ import { ClientModelInfo } from "../../../src/common/api/common/EntityFunctions"
import { InboxRuleHandler } from "../../../src/mail-app/mail/model/InboxRuleHandler"
import { SpamClassificationHandler } from "../../../src/mail-app/mail/model/SpamClassificationHandler"
import { SpamClassifier } from "../../../src/mail-app/workerUtils/spamClassification/SpamClassifier"
import { WebsocketConnectivityModel } from "../../../src/common/misc/WebsocketConnectivityModel"
const { anything } = matchers
@ -40,6 +41,7 @@ o.spec("MailModelTest", function () {
anotherFolder.folderType = MailSetKind.ARCHIVE
let logins: LoginController
let mailFacade: MailFacade
let connectivityModel: WebsocketConnectivityModel
const restClient: EntityRestClientMock = new EntityRestClientMock()
o.beforeEach(function () {
@ -53,6 +55,9 @@ o.spec("MailModelTest", function () {
when(userController.isUpdateForLoggedInUserInstance(matchers.anything(), matchers.anything())).thenReturn(false)
when(logins.getUserController()).thenReturn(userController)
connectivityModel = object<WebsocketConnectivityModel>()
when(connectivityModel.isLeader()).thenReturn(true)
model = new MailModel(
downcast({}),
mailboxModel,
@ -60,7 +65,7 @@ o.spec("MailModelTest", function () {
new EntityClient(restClient, ClientModelInfo.getNewInstanceForTestsOnly()),
logins,
mailFacade,
null,
connectivityModel,
() => object(),
() => null,
)
@ -114,7 +119,7 @@ o.spec("MailModelTest", function () {
mailboxModel = instance(MailboxModel)
inboxRuleHandler = object<InboxRuleHandler>()
spamClassifier = object<SpamClassifier>()
spamClassificationHandler = new SpamClassificationHandler(object(), spamClassifier, entityClient, object())
spamClassificationHandler = new SpamClassificationHandler(object(), spamClassifier, entityClient, object(), connectivityModel)
model = new MailModel(
downcast({}),