Add unit tests

packages/edit-site/src/store/test/actions.js

@@ -13,7 +13,7 @@ import { store as preferencesStore } from '@wordpress/preferences';
  * Internal dependencies
  */
 import { store as editSiteStore } from '..';
-import { setHasPageContentFocus } from '../actions';
+import { setHasPageContentFocus, setPageContentFocusType } from '../actions';
 
 const ENTITY_TYPES = {
 	wp_template: {
@@ -246,4 +246,13 @@ describe( 'actions', () => {
 			} );
 		} );
 	} );
+
+	describe( 'setPageContentFocusType', () => {
+		it( 'sets the page content focus type', () => {
+			expect( setPageContentFocusType( 'disableTemplate' ) ).toEqual( {
+				type: 'SET_PAGE_CONTENT_FOCUS_TYPE',
+				pageContentFocusType: 'disableTemplate',
+			} );
+		} );
+	} );
 } );

packages/edit-site/src/store/test/reducer.js

@@ -12,6 +12,7 @@ import {
 	blockInserterPanel,
 	listViewPanel,
 	hasPageContentFocus,
+	pageContentFocusType,
 } from '../reducer';
 
 import { setIsInserterOpened, setIsListViewOpened } from '../actions';
@@ -179,4 +180,21 @@ describe( 'state', () => {
 			).toBe( false );
 		} );
 	} );
+
+	describe( 'pageContentFocusType', () => {
+		it( 'defaults to disableTemplate', () => {
+			expect( pageContentFocusType( undefined, {} ) ).toBe(
+				'disableTemplate'
+			);
+		} );
+
+		it( 'can be set', () => {
+			expect(
+				pageContentFocusType( 'disableTemplate', {
+					type: 'SET_PAGE_CONTENT_FOCUS_TYPE',
+					pageContentFocusType: 'enableTemplate',
+				} )
+			).toBe( 'enableTemplate' );
+		} );
+	} );
 } );

packages/edit-site/src/store/test/selectors.js

@@ -17,6 +17,7 @@ import {
 	__unstableGetPreference,
 	isPage,
 	hasPageContentFocus,
+	getPageContentFocusType,
 } from '../selectors';
 
 describe( 'selectors', () => {
@@ -204,4 +205,41 @@ describe( 'selectors', () => {
 			expect( hasPageContentFocus( state ) ).toBe( false );
 		} );
 	} );
+
+	describe( 'getPageContentFocusType', () => {
+		it( 'returns the current content focus type', () => {
+			const state = {
+				editedPost: {
+					postType: 'wp_template',
+					context: { postType: 'page', postId: 123 },
+				},
+				hasPageContentFocus: true,
+				pageContentFocusType: 'disableTemplate',
+			};
+			expect( getPageContentFocusType( state ) ).toBe(
+				'disableTemplate'
+			);
+		} );
+
+		it( 'returns null if the page does not have content focus', () => {
+			const state = {
+				editedPost: {
+					postType: 'wp_template',
+					context: { postType: 'page', postId: 123 },
+				},
+				hasPageContentFocus: false,
+			};
+			expect( getPageContentFocusType( state ) ).toBe( null );
+		} );
+
+		it( 'returns null if the edited post type is a template', () => {
+			const state = {
+				editedPost: {
+					postType: 'wp_template',
+				},
+				hasPageContentFocus: true,
+			};
+			expect( getPageContentFocusType( state ) ).toBe( null );
+		} );
+	} );
 } );