@wordpress/edit-post/src/components/browser-url/test/index.jsdom.test.jsx

Edit Post module for WordPress.

import { render, act } from '@testing-library/react';
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
import { useSelect } from '@wordpress/data';
import { default as BrowserURL, getPostEditURL } from '../';

vi.hoisted( () => globalThis.wpVitest.mockMatchMedia() );

vi.mock( '@wordpress/data/src/components/use-select', () => ( {
	default: vi.fn(),
} ) );

function setupUseSelectMock( { postId, postStatus, currentRevisionId } ) {
	useSelect.mockImplementation( () => {
		return {
			postId,
			postStatus,
			currentRevisionId,
		};
	} );
}

function flushURLWrites() {
	act( () => vi.runAllTimers() );
}

describe( 'getPostEditURL', () => {
	it( 'should generate relative path with post and action arguments', () => {
		const url = getPostEditURL( 1 );

		expect( url ).toBe( 'post.php?post=1&action=edit' );
	} );
} );

describe( 'BrowserURL', () => {
	let replaceStateSpy;

	beforeEach( () => {
		vi.useFakeTimers();
		replaceStateSpy = vi.spyOn( window.history, 'replaceState' );
		window.history.replaceState( {}, '', '/' );
		replaceStateSpy.mockReset();
	} );

	afterEach( () => {
		act( () => vi.runOnlyPendingTimers() );
		vi.useRealTimers();
	} );

	it( 'not update URL if post is auto-draft', () => {
		setupUseSelectMock( {
			postId: 1,
			postStatus: 'auto-draft',
		} );

		render( <BrowserURL /> );
		flushURLWrites();
		expect( replaceStateSpy ).not.toHaveBeenCalled();
	} );

	it( 'updates URL immediately if post is no longer auto-draft', () => {
		setupUseSelectMock( {
			postId: 1,
			postStatus: 'auto-draft',
		} );
		const { rerender } = render( <BrowserURL /> );
		flushURLWrites();

		setupUseSelectMock( {
			postId: 1,
			postStatus: 'draft',
		} );

		rerender( <BrowserURL /> );
		expect( replaceStateSpy ).toHaveBeenCalledWith(
			{ id: 1 },
			'Post 1',
			'post.php?post=1&action=edit'
		);
	} );

	it( 'updates URL immediately if post ID changes', () => {
		setupUseSelectMock( {
			postId: 1,
			postStatus: 'draft',
		} );
		const { rerender } = render( <BrowserURL /> );
		flushURLWrites();

		setupUseSelectMock( {
			postId: 2,
			postStatus: 'draft',
		} );
		replaceStateSpy.mockReset();

		rerender( <BrowserURL /> );
		expect( replaceStateSpy ).toHaveBeenCalledWith(
			{ id: 2 },
			'Post 2',
			'post.php?post=2&action=edit'
		);
	} );

	it( 'appends the revision arg while previewing a revision', () => {
		setupUseSelectMock( {
			postId: 1,
			postStatus: 'draft',
			currentRevisionId: 5,
		} );

		render( <BrowserURL /> );
		flushURLWrites();
		expect( replaceStateSpy ).toHaveBeenCalledWith(
			{ id: 1 },
			'Post 1',
			'post.php?post=1&action=edit&revision=5'
		);
	} );

	it( 'writes immediately after a quiet period', () => {
		setupUseSelectMock( {
			postId: 1,
			postStatus: 'draft',
			currentRevisionId: 5,
		} );
		const { rerender } = render( <BrowserURL /> );
		flushURLWrites();

		act( () => vi.advanceTimersByTime( 301 ) );
		setupUseSelectMock( {
			postId: 1,
			postStatus: 'draft',
			currentRevisionId: 6,
		} );
		replaceStateSpy.mockReset();

		rerender( <BrowserURL /> );
		expect( replaceStateSpy ).toHaveBeenCalledWith(
			{ id: 1 },
			'Post 1',
			'post.php?post=1&action=edit&revision=6'
		);
	} );

	it( 'removes the revision arg after exiting revisions mode', () => {
		setupUseSelectMock( {
			postId: 1,
			postStatus: 'draft',
			currentRevisionId: 5,
		} );
		const { rerender } = render( <BrowserURL /> );

		setupUseSelectMock( {
			postId: 1,
			postStatus: 'draft',
			currentRevisionId: null,
		} );
		replaceStateSpy.mockReset();

		rerender( <BrowserURL /> );
		expect( replaceStateSpy ).not.toHaveBeenCalled();
		flushURLWrites();
		expect( replaceStateSpy ).toHaveBeenCalledWith(
			{ id: 1 },
			'Post 1',
			'post.php?post=1&action=edit'
		);
	} );
} );