mattermost-mobile-upstream/app/actions/local/systems.test.ts
Carlos Garcia a5ff8a0d95
Mm 68767 zero persistence (#9803)
* implements auto cache cleanup when zero persistence mode is enabled (0 value)

* uses memory flag with sqlite instead of loki when running in zero persistence mode

* renames ephemeral mode manager and improves readability

* zero persistence refactors and restore values during transitions

* fixes ephemeral mode tests

* use initializeClient instead of just initialize to wait for doReconnect to finish

that way teams loading is awaited while WS reconnects during ZPM transitions

* adds inital sync as a mount hook to prevent undesired load error states

during app cold start in ZPM mode, we need a way to load team and channels
without showing loading error during the transition. The nre is initial sync
hook allows for it

* reset started sync subject on wipe

* removes extra log lines

* resets tracked server during ephemeral mode transitions

* improvements to apply persistence mode changes when something fails

* address coderabbit PR suggestions

* add batching time logging

* throws ephemeral mode change error as is

* make deleteServerDatabaseFiles private

* inline derive persistence flag function

* return current user id when refetching current user

* support zero persistence mode server for push notifications

* adds support for zero persistence mode when sharing to mattermost

* adds clarifying comment

* catch and log error deleting files

* reorder translations to correct position

* adds some new tests to improve coverage

* ios push notification native fix

* logout when fetchMe fails with 401 status code during reconnection

* properly surface errors

* return proper false value when checking for zero persistence flag

* change comments and properly await on some operations

* prevent not set persistence mode to report as an error

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

* include servers with no persistence flag set in share list

* disable button for accessibility

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

* improves when to store and remove signing key from global area

* review push notification reception running in zero persistence mode

* add server to ephemeral mode manager right after login

---------

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
2026-07-16 13:55:45 +02:00

689 lines
27 KiB
TypeScript

// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.
import Database from '@nozbe/watermelondb/Database';
import {removePushSigningKey, storePushSigningKey} from '@actions/app/global';
import {getPosts} from '@actions/local/post';
import {ActionType} from '@constants';
import {SYSTEM_IDENTIFIERS} from '@constants/database';
import {PostTypes} from '@constants/post';
import DatabaseManager from '@database/manager';
import TestHelper from '@test/test_helper';
import {logError} from '@utils/log';
jest.mock('@actions/app/global', () => ({
removePushSigningKey: jest.fn(),
storePushSigningKey: jest.fn(),
}));
import {
storeConfig,
storeConfigAndLicense,
storeDataRetentionPolicies,
updateLastDataRetentionRun,
dataRetentionCleanup,
setLastServerVersionCheck,
setGlobalThreadsTab,
dismissAnnouncement,
expiredBoRPostCleanup,
setDisconnectedSince,
} from './systems';
import type {DataRetentionPoliciesRequest} from '@actions/remote/systems';
import type ServerDataOperator from '@database/operator/server_data_operator';
import type SystemModel from '@typings/database/models/servers/system';
const serverUrl = 'baseHandler.test.com';
let operator: ServerDataOperator;
jest.mock('@init/credentials', () => {
const original = jest.requireActual('@init/credentials');
return {
...original,
getServerCredentials: jest.fn(async (url: string) => ({serverUrl: url})),
};
});
beforeEach(async () => {
await DatabaseManager.init([serverUrl]);
operator = DatabaseManager.serverDatabases[serverUrl]!.operator;
});
afterEach(async () => {
await DatabaseManager.destroyServerDatabase(serverUrl);
});
describe('storeConfigAndLicense', () => {
it('handle not found database - storeConfig', async () => {
const models = await storeConfig('foo', {} as ClientConfig);
expect(models).toBeDefined();
expect(models.length).toBe(0);
});
it('handle undefined config - storeConfig', async () => {
const models = await storeConfig(serverUrl, undefined);
expect(models).toBeDefined();
expect(models.length).toBe(0);
});
it('base case - storeConfig', async () => {
await operator.handleConfigs({
configs: [
{id: 'DataRetentionEnableMessageDeletion', value: 'true'},
{id: 'AboutLink', value: 'link'},
],
configsToDelete: [],
prepareRecordsOnly: false,
});
const models = await storeConfig(serverUrl, {AboutLink: 'link'} as ClientConfig);
expect(models).toBeDefined();
expect(models.length).toBe(1); // data retention removed
});
it('nothing to update - storeConfig', async () => {
await operator.handleConfigs({
configs: [
{id: 'AboutLink', value: 'link'},
],
configsToDelete: [],
prepareRecordsOnly: false,
});
const models = await storeConfig(serverUrl, {AboutLink: 'link'} as ClientConfig);
expect(models).toBeDefined();
expect(models.length).toBe(0);
});
it('handle not found database', async () => {
const models = await storeConfigAndLicense('foo', {} as ClientConfig, {} as ClientLicense);
expect(models).toBeDefined();
expect(models.length).toBe(0);
});
it('base case', async () => {
const models = await storeConfigAndLicense(serverUrl, {AboutLink: 'link'} as ClientConfig, {Announcement: 'test'} as ClientLicense);
expect(models).toBeDefined();
expect(models.length).toBe(1); // config
});
});
describe('storeConfig', () => {
const zeroPersistenceConfig = {
MobileEphemeralModeEnabled: 'true',
MobileEphemeralModeAutoCacheCleanupDays: '0',
AsymmetricSigningPublicKey: 'key-1',
} as ClientConfig;
describe('storing the signing key', () => {
it('mirrors the signing key when the config is zero-persistence', async () => {
await storeConfig(serverUrl, zeroPersistenceConfig);
expect(storePushSigningKey).toHaveBeenCalledWith(serverUrl, 'key-1');
expect(removePushSigningKey).not.toHaveBeenCalled();
});
it('mirrors the signing key even when only preparing records', async () => {
await storeConfig(serverUrl, zeroPersistenceConfig, true);
expect(storePushSigningKey).toHaveBeenCalledWith(serverUrl, 'key-1');
});
it('does not store the signing key when the config is not zero-persistence', async () => {
await storeConfig(serverUrl, {AsymmetricSigningPublicKey: 'key-1'} as ClientConfig);
expect(storePushSigningKey).not.toHaveBeenCalled();
});
it('mirrors the signing key when entering zero-persistence even if the key is unchanged from the stored config', async () => {
await operator.handleConfigs({
configs: [
{id: 'MobileEphemeralModeEnabled', value: 'false'},
{id: 'AsymmetricSigningPublicKey', value: 'key-1'},
],
configsToDelete: [],
prepareRecordsOnly: false,
});
await storeConfig(serverUrl, zeroPersistenceConfig);
expect(storePushSigningKey).toHaveBeenCalledWith(serverUrl, 'key-1');
expect(removePushSigningKey).not.toHaveBeenCalled();
});
it('does not store the signing key when it is unchanged from the stored config', async () => {
await operator.handleConfigs({
configs: [
{id: 'MobileEphemeralModeEnabled', value: 'true'},
{id: 'MobileEphemeralModeAutoCacheCleanupDays', value: '0'},
{id: 'AsymmetricSigningPublicKey', value: 'key-1'},
{id: 'AboutLink', value: 'old-link'},
],
configsToDelete: [],
prepareRecordsOnly: false,
});
await storeConfig(serverUrl, {...zeroPersistenceConfig, AboutLink: 'new-link'} as ClientConfig);
expect(storePushSigningKey).not.toHaveBeenCalled();
});
});
describe('removing the signing key', () => {
it('removes the mirrored signing key when the config transitions away from zero-persistence', async () => {
await operator.handleConfigs({
configs: [
{id: 'MobileEphemeralModeEnabled', value: 'true'},
{id: 'MobileEphemeralModeAutoCacheCleanupDays', value: '0'},
{id: 'AsymmetricSigningPublicKey', value: 'key-1'},
],
configsToDelete: [],
prepareRecordsOnly: false,
});
await storeConfig(serverUrl, {
MobileEphemeralModeEnabled: 'false',
MobileEphemeralModeAutoCacheCleanupDays: '0',
AsymmetricSigningPublicKey: 'key-1',
} as ClientConfig);
expect(removePushSigningKey).toHaveBeenCalledWith(serverUrl);
expect(storePushSigningKey).not.toHaveBeenCalled();
});
it('removes the signing key defensively when prior config is empty and the config is not zero-persistence', async () => {
await storeConfig(serverUrl, {AsymmetricSigningPublicKey: 'key-1'} as ClientConfig);
expect(removePushSigningKey).toHaveBeenCalledWith(serverUrl);
expect(storePushSigningKey).not.toHaveBeenCalled();
});
it('does not remove the signing key when a prior config already confirms the server was not zero-persistence', async () => {
await operator.handleConfigs({
configs: [
{id: 'MobileEphemeralModeEnabled', value: 'false'},
{id: 'AboutLink', value: 'old-link'},
],
configsToDelete: [],
prepareRecordsOnly: false,
});
await storeConfig(serverUrl, {MobileEphemeralModeEnabled: 'false', AboutLink: 'new-link'} as ClientConfig);
expect(removePushSigningKey).not.toHaveBeenCalled();
});
it('removes the mirrored signing key when it is dropped from the config while the server remains zero-persistence', async () => {
await operator.handleConfigs({
configs: [
{id: 'MobileEphemeralModeEnabled', value: 'true'},
{id: 'MobileEphemeralModeAutoCacheCleanupDays', value: '0'},
{id: 'AsymmetricSigningPublicKey', value: 'key-1'},
],
configsToDelete: [],
prepareRecordsOnly: false,
});
await storeConfig(serverUrl, {
MobileEphemeralModeEnabled: 'true',
MobileEphemeralModeAutoCacheCleanupDays: '0',
} as ClientConfig);
expect(removePushSigningKey).toHaveBeenCalledWith(serverUrl);
expect(storePushSigningKey).not.toHaveBeenCalled();
});
});
});
describe('dataRetention', () => {
it('handle not found database - storeDataRetentionPolicies', async () => {
const models = await storeDataRetentionPolicies('foo', {} as DataRetentionPoliciesRequest);
expect(models).toBeDefined();
expect(models.length).toBe(0);
});
it('base case - storeDataRetentionPolicies', async () => {
const models = await storeDataRetentionPolicies(serverUrl, {globalPolicy: {} as GlobalDataRetentionPolicy, teamPolicies: [], channelPolicies: []} as DataRetentionPoliciesRequest);
expect(models).toBeDefined();
expect(models.length).toBe(2); // data retention and granular data retention policies
});
it('empty case - storeDataRetentionPolicies', async () => {
const models = await storeDataRetentionPolicies(serverUrl, {} as DataRetentionPoliciesRequest);
expect(models).toBeDefined();
expect(models.length).toBe(2); // data retention and granular data retention policies
});
it('handle not found database - updateLastDataRetentionRun', async () => {
const {error} = await updateLastDataRetentionRun('foo', 0) as {error: unknown};
expect(error).toBeDefined();
});
it('base case - updateLastDataRetentionRun', async () => {
const models = await updateLastDataRetentionRun(serverUrl, 10) as SystemModel[];
expect(models).toBeDefined();
expect(models.length).toBe(1); // data retention
});
it('no time provided - updateLastDataRetentionRun', async () => {
const models = await updateLastDataRetentionRun(serverUrl) as SystemModel[];
expect(models).toBeDefined();
expect(models.length).toBe(1); // data retention
});
it('handle not found database - dataRetentionCleanup', async () => {
const {error} = await dataRetentionCleanup('foo');
expect(error).toBeDefined();
});
it('rentention off - dataRetentionCleanup', async () => {
const post = TestHelper.fakePost({channel_id: 'channelid1', id: 'postid', create_at: 1});
await operator.handlePosts({
actionType: ActionType.POSTS.RECEIVED_IN_CHANNEL,
order: [post.id],
posts: [post],
prepareRecordsOnly: false,
});
const spy = jest.spyOn(Database.prototype, 'unsafeVacuum').mockImplementation(jest.fn());
const {error} = await dataRetentionCleanup(serverUrl);
expect(error).toBeDefined(); // unsafeExecute loki error
spy.mockRestore();
});
it('retention on - dataRetentionCleanup', async () => {
const channel: Channel = {
id: 'channelid1',
team_id: 'teamid1',
total_msg_count: 0,
} as Channel;
await operator.handleConfigs({
configs: [
{id: 'DataRetentionEnableMessageDeletion', value: 'true'},
],
configsToDelete: [],
prepareRecordsOnly: false,
});
await operator.handleSystem({systems:
[
{id: SYSTEM_IDENTIFIERS.LICENSE, value: {IsLicensed: 'true', DataRetention: 'true'}},
{id: SYSTEM_IDENTIFIERS.GRANULAR_DATA_RETENTION_POLICIES, value: {team: [{team_id: 'teamid1', post_duration: 100}], channel: [{channel_id: 'channelid1', post_duration: 100}]}},
],
prepareRecordsOnly: false});
await operator.handleChannel({channels: [channel], prepareRecordsOnly: false});
const spy = jest.spyOn(Database.prototype, 'unsafeVacuum').mockImplementation(jest.fn());
const {error} = await dataRetentionCleanup(serverUrl);
expect(error).toBeDefined(); // LokiJSAdapter doesn't support unsafeSqlQuery
spy.mockRestore();
});
it('already cleaned today - dataRetentionCleanup', async () => {
const channel: Channel = {
id: 'channelid1',
team_id: 'teamid1',
total_msg_count: 0,
} as Channel;
await operator.handleConfigs({
configs: [
{id: 'DataRetentionEnableMessageDeletion', value: 'true'},
],
configsToDelete: [],
prepareRecordsOnly: false,
});
await operator.handleSystem({systems:
[
{id: SYSTEM_IDENTIFIERS.LAST_DATA_RETENTION_RUN, value: Date.now()},
{id: SYSTEM_IDENTIFIERS.LICENSE, value: {IsLicensed: 'true', DataRetention: 'true'}},
{id: SYSTEM_IDENTIFIERS.GRANULAR_DATA_RETENTION_POLICIES, value: {team: [{team_id: 'teamid1', post_duration: 100}], channel: [{channel_id: 'channelid1', post_duration: 100}]}},
],
prepareRecordsOnly: false});
await operator.handleChannel({channels: [channel], prepareRecordsOnly: false});
const {error} = await dataRetentionCleanup(serverUrl);
expect(error).toBeUndefined();
});
});
describe('setLastServerVersionCheck', () => {
it('handle not found database', async () => {
const {error} = await setLastServerVersionCheck('foo');
expect(error).toBeDefined();
});
it('base case', async () => {
const {error} = await setLastServerVersionCheck(serverUrl);
expect(error).toBeUndefined();
});
it('base case - reset', async () => {
const {error} = await setLastServerVersionCheck(serverUrl, true);
expect(error).toBeUndefined();
});
});
describe('setGlobalThreadsTab', () => {
it('handle not found database', async () => {
const {error} = await setGlobalThreadsTab('foo', 'all');
expect(error).toBeDefined();
});
it('base case', async () => {
const {error} = await setGlobalThreadsTab(serverUrl, 'all');
expect(error).toBeUndefined();
});
});
describe('dismissAnnouncement', () => {
it('handle not found database', async () => {
const {error} = await dismissAnnouncement('foo', 'text');
expect(error).toBeDefined();
});
it('base case', async () => {
const {error} = await dismissAnnouncement(serverUrl, 'text');
expect(error).toBeUndefined();
});
});
describe('setDisconnectedSince', () => {
it('should swallow when server database is not found', async () => {
await expect(setDisconnectedSince('unknown-server', 1719400000000)).resolves.toBeUndefined();
});
it('should write a timestamp to the system table', async () => {
const timestamp = 1719400000000;
await setDisconnectedSince(serverUrl, timestamp);
const records = await operator.database.get<SystemModel>('System').
query().
fetch();
const row = records.find((r) => r.id === SYSTEM_IDENTIFIERS.DISCONNECTED_SINCE);
expect(row?.value).toBe(timestamp);
});
it('should delete the row when passed null', async () => {
await setDisconnectedSince(serverUrl, 1719400000000);
await setDisconnectedSince(serverUrl, null);
const records = await operator.database.get<SystemModel>('System').
query().
fetch();
expect(records.find((r) => r.id === SYSTEM_IDENTIFIERS.DISCONNECTED_SINCE)).toBeUndefined();
});
it('should be a no-op when deleting a non-existent row', async () => {
await expect(setDisconnectedSince(serverUrl, null)).resolves.toBeUndefined();
});
});
describe('expiredBoRPostCleanup', () => {
it('should delete expired BoR posts', async () => {
const database = operator.database;
jest.spyOn(database.adapter, 'unsafeExecute').mockImplementation(() => Promise.resolve());
const channel: Channel = TestHelper.fakeChannel({
id: 'channelid1',
team_id: 'teamid1',
});
await operator.handleChannel({channels: [channel], prepareRecordsOnly: false});
const now = Date.now();
const borPostExpiredForAll = TestHelper.fakePost({
id: 'postid1',
channel_id: channel.id,
type: PostTypes.BURN_ON_READ,
props: {expire_at: now - 10000},
});
const borPostExpiredForMe = TestHelper.fakePost({
id: 'postid2',
channel_id: channel.id,
type: PostTypes.BURN_ON_READ,
props: {expire_at: now + 100000},
metadata: {expire_at: now - 10000},
});
await operator.handlePosts({
actionType: ActionType.POSTS.RECEIVED_IN_CHANNEL,
order: [borPostExpiredForAll.id, borPostExpiredForMe.id],
posts: [borPostExpiredForAll, borPostExpiredForMe],
prepareRecordsOnly: false,
});
// verify channel posts
const fetchedPosts = await getPosts(serverUrl, [borPostExpiredForAll.id, borPostExpiredForMe.id]);
expect(fetchedPosts.length).toBe(2);
await expiredBoRPostCleanup(serverUrl);
expect(database.adapter.unsafeExecute).toHaveBeenCalledWith({
sqls: [
[`DELETE FROM Post where id IN ('${borPostExpiredForMe.id}','${borPostExpiredForAll.id}')`, []],
[`DELETE FROM Reaction where post_id IN ('${borPostExpiredForMe.id}','${borPostExpiredForAll.id}')`, []],
[`DELETE FROM File where post_id IN ('${borPostExpiredForMe.id}','${borPostExpiredForAll.id}')`, []],
[`DELETE FROM Draft where root_id IN ('${borPostExpiredForMe.id}','${borPostExpiredForAll.id}')`, []],
[`DELETE FROM PostsInThread where root_id IN ('${borPostExpiredForMe.id}','${borPostExpiredForAll.id}')`, []],
[`DELETE FROM Thread where id IN ('${borPostExpiredForMe.id}','${borPostExpiredForAll.id}')`, []],
[`DELETE FROM ThreadParticipant where thread_id IN ('${borPostExpiredForMe.id}','${borPostExpiredForAll.id}')`, []],
[`DELETE FROM ThreadsInTeam where thread_id IN ('${borPostExpiredForMe.id}','${borPostExpiredForAll.id}')`, []],
],
});
});
it('should not run cleanup when called again within 15 minutes', async () => {
const database = operator.database;
const unsafeExecuteSpy = jest.spyOn(database.adapter, 'unsafeExecute').mockImplementation(() => Promise.resolve());
// Set up a recent last run time (within 15 minutes)
const recentRunTime = Date.now() - (10 * 60 * 1000); // 10 minutes ago
await operator.handleSystem({
systems: [{
id: SYSTEM_IDENTIFIERS.LAST_BOR_POST_CLEANUP_RUN,
value: recentRunTime,
}],
prepareRecordsOnly: false,
});
const channel: Channel = TestHelper.fakeChannel({
id: 'channelid1',
team_id: 'teamid1',
});
await operator.handleChannel({channels: [channel], prepareRecordsOnly: false});
const now = Date.now();
const borPostExpired = TestHelper.fakePost({
id: 'postid1',
channel_id: channel.id,
type: PostTypes.BURN_ON_READ,
props: {expire_at: now - 10000},
});
await operator.handlePosts({
actionType: ActionType.POSTS.RECEIVED_IN_CHANNEL,
order: [borPostExpired.id],
posts: [borPostExpired],
prepareRecordsOnly: false,
});
// Call cleanup - should not run because last run was within 15 minutes
await expiredBoRPostCleanup(serverUrl);
// Verify that unsafeExecute was not called (no cleanup performed)
expect(unsafeExecuteSpy).not.toHaveBeenCalled();
});
it('should handle no server database gracefully', async () => {
// Try to run cleanup on a non-existent server
await expect(expiredBoRPostCleanup('nonexistent.server.com')).resolves.not.toThrow();
});
it('should handle no BoR posts gracefully', async () => {
const database = operator.database;
const unsafeExecuteSpy = jest.spyOn(database.adapter, 'unsafeExecute').mockImplementation(() => Promise.resolve());
const channel: Channel = TestHelper.fakeChannel({
id: 'channelid1',
team_id: 'teamid1',
});
await operator.handleChannel({channels: [channel], prepareRecordsOnly: false});
await expiredBoRPostCleanup(serverUrl);
// Verify that unsafeExecute was not called (no BoR posts to clean)
expect(unsafeExecuteSpy).not.toHaveBeenCalled();
});
it('should handle BoR posts that are not expired', async () => {
const database = operator.database;
const unsafeExecuteSpy = jest.spyOn(database.adapter, 'unsafeExecute').mockImplementation(() => Promise.resolve());
const channel: Channel = TestHelper.fakeChannel({
id: 'channelid1',
team_id: 'teamid1',
});
await operator.handleChannel({channels: [channel], prepareRecordsOnly: false});
const now = Date.now();
// Create BoR posts that are not expired
const borPostNotExpired = TestHelper.fakePost({
id: 'postid1',
channel_id: channel.id,
type: PostTypes.BURN_ON_READ,
props: {expire_at: now + 100000}, // Future expiry
});
const borPostNotExpiredForMe = TestHelper.fakePost({
id: 'postid2',
channel_id: channel.id,
type: PostTypes.BURN_ON_READ,
props: {expire_at: now + 10000},
metadata: {expire_at: now + 100000},
});
await operator.handlePosts({
actionType: ActionType.POSTS.RECEIVED_IN_CHANNEL,
order: [borPostNotExpired.id, borPostNotExpiredForMe.id],
posts: [borPostNotExpired, borPostNotExpiredForMe],
prepareRecordsOnly: false,
});
await expiredBoRPostCleanup(serverUrl);
// Verify that unsafeExecute was not called (no expired BoR posts)
expect(unsafeExecuteSpy).not.toHaveBeenCalled();
});
it('should handle mixed expired and non-expired BoR posts', async () => {
const database = operator.database;
jest.spyOn(database.adapter, 'unsafeExecute').mockImplementation(() => Promise.resolve());
const channel: Channel = TestHelper.fakeChannel({
id: 'channelid1',
team_id: 'teamid1',
});
await operator.handleChannel({channels: [channel], prepareRecordsOnly: false});
const now = Date.now();
const borPostExpired = TestHelper.fakePost({
id: 'postid1',
channel_id: channel.id,
type: PostTypes.BURN_ON_READ,
props: {expire_at: now - 10000}, // Expired
});
const borPostNotExpired = TestHelper.fakePost({
id: 'postid2',
channel_id: channel.id,
type: PostTypes.BURN_ON_READ,
props: {expire_at: now + 100000}, // Not expired
});
await operator.handlePosts({
actionType: ActionType.POSTS.RECEIVED_IN_CHANNEL,
order: [borPostExpired.id, borPostNotExpired.id],
posts: [borPostExpired, borPostNotExpired],
prepareRecordsOnly: false,
});
await expiredBoRPostCleanup(serverUrl);
// Should only delete the expired post
expect(database.adapter.unsafeExecute).toHaveBeenCalledWith({
sqls: [
[`DELETE FROM Post where id IN ('${borPostExpired.id}')`, []],
[`DELETE FROM Reaction where post_id IN ('${borPostExpired.id}')`, []],
[`DELETE FROM File where post_id IN ('${borPostExpired.id}')`, []],
[`DELETE FROM Draft where root_id IN ('${borPostExpired.id}')`, []],
[`DELETE FROM PostsInThread where root_id IN ('${borPostExpired.id}')`, []],
[`DELETE FROM Thread where id IN ('${borPostExpired.id}')`, []],
[`DELETE FROM ThreadParticipant where thread_id IN ('${borPostExpired.id}')`, []],
[`DELETE FROM ThreadsInTeam where thread_id IN ('${borPostExpired.id}')`, []],
],
});
});
it('should handle database errors gracefully', async () => {
const database = operator.database;
// Mock database query to throw an error
jest.spyOn(database, 'get').mockImplementation(() => {
throw new Error('Database error');
});
// Should not throw an error, just log it
await expect(expiredBoRPostCleanup(serverUrl)).resolves.not.toThrow();
expect(logError).toHaveBeenCalledWith('An error occurred while performing BoR post cleanup', expect.any(Error));
});
it('should handle updateLastBoRCleanupRun error gracefully', async () => {
const database = operator.database;
jest.spyOn(database.adapter, 'unsafeExecute').mockImplementation(() => Promise.resolve());
// Mock handleSystem to throw an error when updating last cleanup run
const handleSystemSpy = jest.spyOn(operator, 'handleSystem').mockImplementation(() => {
throw new Error('System update error');
});
const channel: Channel = TestHelper.fakeChannel({
id: 'channelid1',
team_id: 'teamid1',
});
await operator.handleChannel({channels: [channel], prepareRecordsOnly: false});
// Restore handleSystem for channel creation, then mock it again for the cleanup run update
handleSystemSpy.mockRestore();
jest.spyOn(operator, 'handleSystem').mockImplementation((args) => {
if (args.systems?.[0]?.id === SYSTEM_IDENTIFIERS.LAST_BOR_POST_CLEANUP_RUN) {
throw new Error('System update error');
}
return Promise.resolve([]);
});
const now = Date.now();
const borPostExpired = TestHelper.fakePost({
id: 'postid1',
channel_id: channel.id,
type: PostTypes.BURN_ON_READ,
props: {expire_at: now - 10000},
});
await operator.handlePosts({
actionType: ActionType.POSTS.RECEIVED_IN_CHANNEL,
order: [borPostExpired.id],
posts: [borPostExpired],
prepareRecordsOnly: false,
});
// Should not throw an error, just log it
await expect(expiredBoRPostCleanup(serverUrl)).resolves.not.toThrow();
expect(logError).toHaveBeenCalledWith('Failed updateLastBoRCleanupRun', expect.any(Error));
});
});