Function makeMockAlarmScheduler

apps/dotcom/sync-worker/src/AlarmScheduler.test.ts:7–56

(alarms: {
	[K in Key]: ReturnType<typeof vi.fn<(...args: any[]) => any>>
})

vi.useFakeTimers()

function makeMockAlarmScheduler<Key extends string>(alarms: {
	[K in Key]: ReturnType<typeof vi.fn<(...args: any[]) => any>>
}) {
	const data = new Map<string, number>()
	let scheduledAlarm: number | null = null

	const storage = {
		getAlarm: async () => scheduledAlarm,
		setAlarm: vi.fn((time: number | Date) => {
			scheduledAlarm = typeof time === 'number' ? time : time.getTime()
		}),
		get: async (key: string) => data.get(key),
		list: async () => new Map(data),
		delete: async (keys: string[]) => {
			let count = 0
			for (const key of keys) {
				if (data.delete(key)) count++
			}
			return count
		},
		put: async (entries: Record<string, number>) => {
			for (const [key, value] of Object.entries(entries)) {
				data.set(key, value)
			}
		},
		asObject: () => Object.fromEntries(data),
	}

	const scheduler = new AlarmScheduler({
		alarms,
		storage: () => storage,
	})

	const advanceTime = async (time: number) => {
		vi.advanceTimersByTime(time)
		if (scheduledAlarm !== null && scheduledAlarm <= Date.now()) {
			scheduledAlarm = null
			await scheduler.onAlarm()
			// process the alarms that were scheduled during the onAlarm call:
			if (scheduledAlarm) await advanceTime(0)
		}
	}

	return {
		scheduler,
		storage,
		alarms,
		advanceTime,
	}
}

describe('AlarmScheduler', () => {
	beforeEach(() => {