mynode/packages/app-desktop/services/autoUpdater/AutoUpdaterService.test.ts

import { LoggerWrapper } from '@joplin/utils/Logger';
import { releases3 } from '../../utils/checkForUpdatesUtilsTestData';
import AutoUpdaterService from './AutoUpdaterService';
import { BrowserWindow } from 'electron';
import { autoUpdater } from 'electron-updater';

jest.mock('electron', () => ({
	BrowserWindow: jest.fn(),
}));

jest.mock('electron-updater', () => {
	const mockAutoUpdater = {
		setFeedURL: jest.fn(),
		checkForUpdates: jest.fn(),
		on: jest.fn(),
		quitAndInstall: jest.fn(),
	};
	return { autoUpdater: mockAutoUpdater };
});


describe('AutoUpdaterService', () => {
	let service: AutoUpdaterService;
	let mockLogger: LoggerWrapper;

	beforeEach(() => {
		const mockWindow = new BrowserWindow();
		mockLogger = { info: jest.fn(), error: jest.fn(), debug: jest.fn(), warn: jest.fn() };
		service = new AutoUpdaterService(mockWindow, mockLogger, false, true);
		global.fetch = jest.fn(() =>
			Promise.resolve({
				ok: true,
				json: () => Promise.resolve(releases3),
			}),
		) as jest.Mock;
	});

	// electron-updater rebuilds the file name from the channel, so a wrong
	// channel would silently fetch the x64 metadata on an arm64 machine.
	test.each([
		['win32', 'arm64', 'latest-win-arm64'],
		['win32', 'x64', 'latest'],
		['darwin', 'arm64', 'latest'],
	])('should set the update channel for %s %s', async (platform, arch, expectedChannel) => {
		// releases3 predates the Windows arm64 build, so add its asset.
		const releases = JSON.parse(JSON.stringify(releases3));
		releases[0].assets.push({
			name: 'latest-win-arm64.yml',
			browser_download_url: 'https://github.com/laurent22/joplin/releases/download/v3.1.3/latest-win-arm64.yml',
		});
		global.fetch = jest.fn(() => Promise.resolve({
			ok: true,
			json: () => Promise.resolve(releases),
		})) as jest.Mock;

		const originalPlatform = process.platform;
		const originalArch = process.arch;
		Object.defineProperty(process, 'platform', { value: platform, configurable: true });
		Object.defineProperty(process, 'arch', { value: arch, configurable: true });

		try {
			await service.checkForUpdates(true);
		} finally {
			Object.defineProperty(process, 'platform', { value: originalPlatform, configurable: true });
			Object.defineProperty(process, 'arch', { value: originalArch, configurable: true });
		}

		expect(autoUpdater.setFeedURL).toHaveBeenCalledWith(
			expect.objectContaining({ channel: expectedChannel }),
		);
	});

	it('should correctly fetch and process the latest prerelease', async () => {
		const release = await service.fetchLatestRelease(true);
		expect(release).toBeDefined();
		expect(release.tag_name).toBe('v3.1.3');
	});

	it('should correctly fetch and process the latest release', async () => {
		const release = await service.fetchLatestRelease(false);
		expect(release).toBeDefined();
		expect(release.tag_name).toBe('v3.1.2');
	});

	it('should return the correct download URL for Windows x32', async () => {
		const release = await service.fetchLatestRelease(true);
		expect(release).toBeDefined();
		const url = service.getDownloadUrlForPlatform(release, 'win32', 'ia32');
		expect(url).toBe('https://github.com/laurent22/joplin/releases/download/v3.1.3/latest.yml');
	});

	it('should return the correct download URL for Windows x64', async () => {
		const release = await service.fetchLatestRelease(true);
		expect(release).toBeDefined();
		const url = service.getDownloadUrlForPlatform(release, 'win32', 'x64');
		expect(url).toBe('https://github.com/laurent22/joplin/releases/download/v3.1.3/latest.yml');
	});

	it('should return the correct download URL for Windows arm64', async () => {
		const release = await service.fetchLatestRelease(true);
		expect(release).toBeDefined();
		const url = 'https://github.com/laurent22/joplin/releases/download/v3.1.3/latest-win-arm64.yml';
		release.assets.push({ name: 'latest-win-arm64.yml', browser_download_url: url });
		expect(service.getDownloadUrlForPlatform(release, 'win32', 'arm64')).toBe(url);
	});

	it('should return the correct download URL for Mac x64', async () => {
		const release = await service.fetchLatestRelease(true);
		expect(release).toBeDefined();
		const url = service.getDownloadUrlForPlatform(release, 'darwin', 'x64');
		expect(url).toBe('https://github.com/laurent22/joplin/releases/download/v3.1.3/latest-mac.yml');
	});

	it('should return the correct download URL for Mac arm64', async () => {
		const release = await service.fetchLatestRelease(true);
		expect(release).toBeDefined();
		const url = service.getDownloadUrlForPlatform(release, 'darwin', 'arm64');
		expect(url).toBe('https://github.com/laurent22/joplin/releases/download/v3.1.3/latest-mac-arm64.yml');
	});

	it('should throw an error for Linux', async () => {
		const release = await service.fetchLatestRelease(true);
		expect(release).toBeDefined();
		expect(() => service.getDownloadUrlForPlatform(release, 'linux', 'amd64')).toThrow('The AutoUpdaterService does not support the following platform: linux');
	});
});