perf(worker): skip the premium sweep on self-hosted instances (#2195)

fluxer_api/src/api/worker/tests/ProcessExpiredPremiumSweep.test.ts

@@ -0,0 +1,69 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+
+import type {WorkerTaskHelpers} from '@pkgs/worker/src/contracts/WorkerTask';
+import {afterEach, describe, expect, test} from 'vitest';
+import {getConfig} from '../../Config';
+import {NoopLogger} from '../../test/mocks/NoopLogger';
+import type {UserRepository} from '../../user/repositories/UserRepository';
+import processExpiredPremiumSweep from '../tasks/ProcessExpiredPremiumSweep';
+import {clearWorkerDependencies, setWorkerDependenciesForTest} from '../WorkerContext';
+
+function createHarness() {
+	const scanLimits: Array<number> = [];
+	const userRepository = {
+		async scanAllUsersPage(limit: number): Promise<{users: []; pageState: null}> {
+			scanLimits.push(limit);
+			return {users: [], pageState: null};
+		},
+	} as unknown as UserRepository;
+	setWorkerDependenciesForTest({userRepository});
+	return {scanLimits};
+}
+
+function createHelpers(): WorkerTaskHelpers {
+	return {
+		logger: new NoopLogger(),
+		jobId: 1n,
+		addJob: async () => 0n,
+		reportProgress: async () => {},
+		shouldCancel: async () => false,
+		setContextLink: async () => {},
+	};
+}
+
+async function withSelfHosted(selfHosted: boolean, callback: () => Promise<void>): Promise<void> {
+	const config = getConfig();
+	const originalSelfHosted = config.instance.selfHosted;
+	try {
+		config.instance.selfHosted = selfHosted;
+		await callback();
+	} finally {
+		config.instance.selfHosted = originalSelfHosted;
+	}
+}
+
+describe('processExpiredPremiumSweep', () => {
+	afterEach(() => {
+		clearWorkerDependencies();
+	});
+
+	test('scans no users on a self-hosted instance', async () => {
+		const harness = createHarness();
+
+		await withSelfHosted(true, async () => {
+			await processExpiredPremiumSweep({}, createHelpers());
+		});
+
+		expect(harness.scanLimits).toEqual([]);
+	});
+
+	test('scans users on a hosted instance', async () => {
+		const harness = createHarness();
+
+		await withSelfHosted(false, async () => {
+			await processExpiredPremiumSweep({}, createHelpers());
+		});
+
+		expect(harness.scanLimits).toEqual([100]);
+	});
+});