feat: Initial clean release of CHM

packages/app-mobile/components/screens/NoteLockMigrationScreen.test.tsx

@@ -0,0 +1,100 @@
+import * as React from 'react';
+import { Store } from 'redux';
+import Setting from '@joplin/lib/models/Setting';
+import { setupDatabaseAndSynchronizer, switchClient } from '@joplin/lib/testing/test-utils';
+import { confirmAndSkipNoteLockKeyMigration, startNoteLockKeyMigration } from '@joplin/lib/services/noteLock/NoteLockKeyMigration';
+import { NoteLockMigrationScreenComponent } from './NoteLockMigrationScreen';
+import TestProviderStack from '../testing/TestProviderStack';
+import createMockReduxStore from '../../utils/testing/createMockReduxStore';
+import setupGlobalStore from '../../utils/testing/setupGlobalStore';
+import { AppState } from '../../utils/types';
+import { fireEvent, render, screen, waitFor } from '../../utils/testing/testingLibrary';
+import BackButtonService from '../../services/BackButtonService';
+
+jest.mock('@joplin/lib/services/noteLock/NoteLockKeyMigration', () => ({
+	startNoteLockKeyMigration: jest.fn(),
+	confirmAndSkipNoteLockKeyMigration: jest.fn(),
+}));
+
+const startMock = startNoteLockKeyMigration as jest.Mock;
+const skipMock = confirmAndSkipNoteLockKeyMigration as jest.Mock;
+
+let store: Store<AppState>;
+const renderScreen = () => {
+	const dispatch = jest.fn();
+	render(
+		<TestProviderStack store={store}>
+			<NoteLockMigrationScreenComponent themeId={Setting.THEME_LIGHT} dispatch={dispatch}/>
+		</TestProviderStack>,
+	);
+	fireEvent.changeText(screen.getByLabelText('Note lock password on this device'), 'local');
+	fireEvent.changeText(screen.getByLabelText('Note lock password on the sync target'), 'target');
+	return dispatch;
+};
+
+describe('NoteLockMigrationScreen', () => {
+	beforeEach(async () => {
+		await setupDatabaseAndSynchronizer(0);
+		await switchClient(0);
+		store = createMockReduxStore();
+		setupGlobalStore(store);
+		jest.clearAllMocks();
+	});
+	afterEach(() => {
+		screen.unmount();
+	});
+
+	test('should go back once both passwords are accepted and leave the run to the banner', async () => {
+		startMock.mockImplementation(async (_local: string, _target: string, _dispatch: unknown, onStarted: ()=> void) => onStarted());
+		const dispatch = renderScreen();
+
+		fireEvent.press(screen.getByText('Migrate'));
+
+		await waitFor(() => expect(dispatch).toHaveBeenCalledWith({ type: 'NAV_BACK' }));
+		expect(startMock).toHaveBeenCalledWith('local', 'target', dispatch, expect.any(Function));
+		expect(skipMock).not.toHaveBeenCalled();
+	});
+
+	test('should block going back while the passwords are checked', async () => {
+		startMock.mockReturnValue(new Promise(() => {}));
+		renderScreen();
+		expect(screen.getByLabelText('Back')).toBeVisible();
+
+		fireEvent.press(screen.getByText('Migrate'));
+
+		await waitFor(() => expect(screen.queryByLabelText('Back')).toBeNull());
+		expect(await BackButtonService.back()).toBe(true);
+	});
+
+	test('should not show the migration banner on its own screen', async () => {
+		Setting.setValue('featureFlag.noteLock', true);
+		Setting.setValue('noteLock.conflictNoteLockKey', { noteLockKey: { id: 'target-key' }, syncMigrationId: 'lineage' });
+		store = createMockReduxStore();
+		setupGlobalStore(store);
+		renderScreen();
+
+		expect(screen.queryByText(/Press to migrate/)).toBeNull();
+	});
+
+	test('should report a wrong password and stay', async () => {
+		startMock.mockRejectedValue(Object.assign(new Error('bad'), { name: 'OperationError' }));
+		const dispatch = renderScreen();
+
+		fireEvent.press(screen.getByText('Migrate'));
+
+		expect(await screen.findByRole('alert')).toHaveTextContent('Invalid password');
+		expect(dispatch).not.toHaveBeenCalled();
+	});
+
+	test.each([true, false])('should leave the screen only once the skip is confirmed (%s)', async (confirmed) => {
+		skipMock.mockResolvedValue(confirmed);
+		const dispatch = renderScreen();
+
+		fireEvent.press(screen.getByText('Skip'));
+
+		await waitFor(() => expect(skipMock).toHaveBeenCalledWith(dispatch));
+		if (confirmed) await waitFor(() => expect(dispatch).toHaveBeenCalledWith({ type: 'NAV_BACK' }));
+		else expect(dispatch).not.toHaveBeenCalled();
+		expect(startMock).not.toHaveBeenCalled();
+	});
+});