Index: tests/phpunit/tests/customize/nav-menus-ajax.php
===================================================================
--- tests/phpunit/tests/customize/nav-menus-ajax.php	(revision 0)
+++ tests/phpunit/tests/customize/nav-menus-ajax.php	(working copy)
@@ -0,0 +1,577 @@
+<?php
+
+/**
+ * Testing the Ajax method functionality
+ *
+ * @package    WordPress
+ * @subpackage UnitTests
+ * @since      4.3.0
+ * @group     ajax
+ * @runTestsInSeparateProcesses
+ */
+class Test_WP_Customize_Nav_Menus_Ajax extends WP_Ajax_UnitTestCase {
+
+	/**
+	 * Instance of WP_Customize_Manager which is reset for each test.
+	 *
+	 * @var WP_Customize_Manager
+	 */
+	public $wp_customize;
+
+
+	/**
+	 * Set up the test fixture.
+	 */
+	public function setUp() {
+		parent::setUp();
+		require_once ABSPATH . WPINC . '/class-wp-customize-manager.php';
+		wp_set_current_user( $this->factory->user->create( array( 'role' => 'administrator' ) ) );
+		global $wp_customize;
+		$this->wp_customize = new WP_Customize_Manager();
+		$wp_customize = $this->wp_customize;
+	}
+
+	/**
+	 * Tear down the test fixture.
+	 */
+	public function tearDown() {
+		parent::tearDown();
+	}
+
+
+	/**
+	 * Helper to keep it DRY
+	 * @param $action
+	 */
+	protected function make_ajax_call( $action ) {
+		// Make the request
+		try {
+			$this->_handleAjax( $action );
+		} catch ( WPAjaxDieContinueException $e ) {
+			unset( $e );
+		}
+	}
+
+
+	/**
+	 * Testing capabilities check for ajax_load_available_items method
+	 *
+	 * @dataProvider data_ajax_load_available_items_cap_check
+	 *
+	 * @param string $role The role we're checking caps against
+	 * @param array $expected_results
+	 */
+	function test_ajax_load_available_items_cap_check( $role, $expected_results ) {
+
+		if ( 'administrator' != $role ) {
+			//if we're not an admin, we should get a wp_die(-1)
+			$this->setExpectedException( 'WPAjaxDieStopException' );
+		}
+
+		wp_set_current_user( $this->factory->user->create( array( 'role' => $role ) ) );
+
+		$_POST = array(
+			'action'                => 'load-available-menu-items-customizer',
+			'customize-menus-nonce' => wp_create_nonce( 'customize-menus' ),
+		);
+
+		$this->make_ajax_call( 'load-available-menu-items-customizer' );
+
+		//if we are an admin, we should get a proper response
+		if ( 'administrator' === $role ) {
+			//get the results
+			$response = json_decode( $this->_last_response, true );
+
+			$this->assertSame( $expected_results, $response );
+		}
+
+	}
+
+	/**
+	 *
+	 * Data provider.
+	 *
+	 * Provides various post_args to induce error messages in the that can be
+	 * compared to the expected_results.
+	 *
+	 * @since 4.3.0
+	 *
+	 * @return array {
+	 *     @type array {
+	 *         @string string $role             The role that will test caps for.
+	 *         @array  array  $expected_results The expected results from the ajax call.
+	 *     }
+	 * }
+	 *
+	 */
+	function data_ajax_load_available_items_cap_check() {
+		return array(
+			array(
+				'subscriber',
+				array(),
+			),
+			array(
+				'contributor',
+				array(),
+			),
+			array(
+				'author',
+				array(),
+			),
+			array(
+				'editor',
+				array(),
+			),
+			array(
+				'administrator',
+				array(
+					'success' => false,
+					'data' => 'nav_menus_missing_obj_type_or_type_parameter',
+				),
+			),
+		);
+	}
+
+
+	/**
+	 * Testing the error messaging for ajax_load_available_items
+	 *
+	 * @dataProvider data_ajax_load_available_items_error_messages
+	 *
+	 */
+	function test_ajax_load_available_items_error_messages( $post_args, $expected_results ) {
+
+		$_POST = array_merge( array(
+			'action'                => 'load-available-menu-items-customizer',
+			'customize-menus-nonce' => wp_create_nonce( 'customize-menus' ),
+		), $post_args );
+
+
+		// Make the request
+		$this->make_ajax_call( 'load-available-menu-items-customizer' );
+
+		//get the results
+		$response = json_decode( $this->_last_response, true );
+
+		$this->assertSame( $expected_results, $response );
+	}
+
+
+	/**
+	 *
+	 * Data provider.
+	 *
+	 * Provides various post_args to induce error messages in the that can be
+	 * compared to the expected_results.
+	 *
+	 * @since 4.3.0
+	 *
+	 * @return array {
+	 *     @type array {
+	 *         @array array $post_args        The arguments that will merged with the $_POST array.
+	 *         @array array $expected_results The expected results from the ajax call.
+	 *     }
+	 * }
+	 *
+	 */
+	function data_ajax_load_available_items_error_messages() {
+		return array(
+			/**
+			 * Testing empty obj_type and type
+			 */
+			array(
+				array(
+					'obj_type'  => '',
+					'type'      => '',
+				),
+
+				array(
+					'success' => false,
+					'data' => 'nav_menus_missing_obj_type_or_type_parameter',
+				),
+			),
+			/**
+			 * Testing empty obj_type
+			 */
+			array(
+				array(
+					'obj_type'  => '',
+					'type'      => 'post',
+				),
+
+				array(
+					'success' => false,
+					'data' => 'nav_menus_missing_obj_type_or_type_parameter',
+				),
+			),
+			/**
+			 * Testing empty type
+			 */
+			array(
+
+				array(
+					'obj_type'  => '',
+					'type'      => 'post',
+				),
+				array(
+					'success' => false,
+					'data' => 'nav_menus_missing_obj_type_or_type_parameter',
+				),
+			),
+			/**
+			 * Testing incorrect obj_type option
+			 */
+			array(
+				array(
+					'obj_type'  => 'invalid',
+					'type'      => 'post',
+				),
+
+				array(
+					'success' => false,
+					'data' => 'nav_menus_invalid_obj_type',
+				),
+			),
+
+			/**
+			 * Testing incorrect type option
+			 */
+			array(
+				array(
+					'obj_type'  => 'post_type',
+					'type'      => 'invalid',
+				),
+
+				array(
+					'success' => false,
+					'data' => 'nav_menus_invalid_post_type',
+				),
+			),
+		);
+	}
+
+
+	/**
+	 * Testing the success status
+	 *
+	 * @dataProvider data_ajax_load_available_items_sucess_status
+	 */
+	function test_ajax_load_available_items_sucess_status( $post_args, $success_status ) {
+
+		$_POST = array_merge( array(
+			'action'                => 'load-available-menu-items-customizer',
+			'customize-menus-nonce' => wp_create_nonce( 'customize-menus' ),
+		), $post_args );
+
+		// Make the request
+		$this->make_ajax_call( 'load-available-menu-items-customizer' );
+
+		//get the results
+		$response = json_decode( $this->_last_response, true );
+		$this->assertSame( $success_status, $response['success'] );
+
+	}
+
+	/**
+	 *
+	 * Data provider.
+	 *
+	 * Provides various post_args to retrieve results and compare against
+	 * the success status.
+	 *
+	 * @since 4.3.0
+	 *
+	 * @return array {
+	 *     @type array {
+	 *         @type array $post_args      The arguments that will merged with the $_POST array.
+	 *         @type bool  $success_status The expected success status.
+	 *     }
+	 * }
+	 *
+	 */
+	function data_ajax_load_available_items_sucess_status() {
+		return array(
+			array(
+				array(
+					'obj_type'  => 'post_type',
+					'type'      => 'post',
+				),
+				true,
+			),
+			array(
+				array(
+					'obj_type'  => 'post_type',
+					'type'      => 'page',
+				),
+				true,
+			),
+			array(
+				array(
+					'obj_type'  => 'post_type',
+					'type'      => 'custom',
+				),
+				false,
+			),
+			array(
+				array(
+					'obj_type'  => 'taxonomy',
+					'type'      => 'post_tag',
+				),
+				true,
+			),
+		);
+	}
+
+	/**
+	 * Testing the array structure for a single item
+	 *
+	 * @dataProvider data_ajax_load_available_items_structure
+	 *
+	 */
+	function test2_ajax_load_available_items_structure( $post_args ) {
+
+		$expected_keys = array(
+			'id',
+			'title',
+			'type',
+			'type_label',
+			'object',
+			'object_id',
+			'url',
+		);
+
+		//create some terms and pages
+		$this->factory->term->create_many( 5 );
+		$this->factory->post->create_many( 5, array( 'post_type' => 'page' ) );
+
+
+		$_POST = array_merge( array(
+			'action'                => 'load-available-menu-items-customizer',
+			'customize-menus-nonce' => wp_create_nonce( 'customize-menus' ),
+		), $post_args );
+
+		// Make the request
+		$this->make_ajax_call( 'load-available-menu-items-customizer' );
+
+		//get the results
+		$response = json_decode( $this->_last_response, true );
+
+		$this->assertNotEmpty( $response['data']['items'] );
+
+		//get the second index to avoid the home page edge case
+		$test_item = $response['data']['items'][1];
+
+		foreach ( $expected_keys as $key ) {
+			$this->assertArrayHasKey( $key, $test_item );
+			$this->assertNotEmpty( $test_item[ $key ] );
+		}
+
+		//special test for the home page
+		if ( 'page' === $test_item['object'] ) {
+			$home = $response['data']['items'][0];
+			foreach ( $expected_keys as $key ) {
+				if ( 'object_id' !== $key ){
+					$this->assertArrayHasKey( $key, $home );
+					if ( 'object' !== $key ) {
+						$this->assertNotEmpty( $home[ $key ] );
+					}
+				}
+			}
+		}
+	}
+
+
+	/**
+	 *
+	 * Data provider.
+	 *
+	 * Provides various post_args to return a list of items to test the array structure of.
+	 *
+	 * @since 4.3.0
+	 *
+	 * @return array {
+	 *     @type array {
+	 *         @type array $post_args The arguments that will merged with the $_POST array.
+	 *     }
+	 * }
+	 *
+	 */
+	function data_ajax_load_available_items_structure() {
+		return array(
+			array(
+				array(
+					'obj_type'  => 'post_type',
+					'type'      => 'post',
+				),
+			),
+			array(
+				array(
+					'obj_type'  => 'post_type',
+					'type'      => 'page',
+				),
+			),
+			array(
+				array(
+					'obj_type'  => 'taxonomy',
+					'type'      => 'post_tag',
+				),
+			),
+		);
+	}
+
+
+	/**
+	 * Testing the error messages for ajax_search_available_items
+	 *
+	 * @dataProvider data_ajax_search_available_items_caps_check
+	 */
+	function test_ajax_search_available_items_caps_check( $role, $expected_results ) {
+
+
+		if ( 'administrator' != $role ) {
+			//if we're not an admin, we should get a wp_die(-1)
+			$this->setExpectedException( 'WPAjaxDieStopException' );
+		}
+
+		wp_set_current_user( $this->factory->user->create( array( 'role' => $role ) ) );
+
+		$_POST = array(
+			'action'                => 'search-available-menu-items-customizer',
+			'customize-menus-nonce' => wp_create_nonce( 'customize-menus' ),
+		);
+
+		$this->make_ajax_call( 'search-available-menu-items-customizer' );
+
+		//if we are an admin, we should get a proper response
+		if ( 'administrator' === $role ) {
+			//get the results
+			$response = json_decode( $this->_last_response, true );
+
+			$this->assertSame( $expected_results, $response );
+		}
+	}
+
+	/**
+	 *
+	 * Data provider.
+	 *
+	 * Provides various post_args to induce error messages in the that can be
+	 * compared to the expected_results.
+	 *
+	 * @since 4.3.0
+	 *
+	 * @todo Make this more DRY
+	 *
+	 * @return array {
+	 *     @type array {
+	 *         @string string $role             The role that will test caps for.
+	 *         @array  array  $expected_results The expected results from the ajax call.
+	 *     }
+	 * }
+	 *
+	 */
+	function data_ajax_search_available_items_caps_check() {
+		return array(
+			array(
+				'subscriber',
+				array(),
+			),
+			array(
+				'contributor',
+				array(),
+			),
+			array(
+				'author',
+				array(),
+			),
+			array(
+				'editor',
+				array(),
+			),
+			array(
+				'administrator',
+				array(
+					'success' => false,
+					'data' => 'nav_menus_missing_search_parameter',
+				),
+			),
+		);
+	}
+
+
+	/**
+	 * Testing the results of various searches
+	 *
+	 * @dataProvider data_ajax_search_available_items_results
+	 */
+	function test_ajax_search_available_items_results( $post_args, $expected_results ) {
+
+		$this->factory->post->create_many( 5, array( 'post_title' => 'Test Post' ) );
+
+
+		$_POST = array_merge( array(
+			'action'                => 'search-available-menu-items-customizer',
+			'customize-menus-nonce' => wp_create_nonce( 'customize-menus' ),
+		), $post_args );
+
+		$this->make_ajax_call( 'search-available-menu-items-customizer' );
+
+		$response = json_decode( $this->_last_response, true );
+
+		if ( isset( $post_args['search'] ) && 'test' === $post_args['search'] ) {
+			$this->assertsame( true, $response['success'] );
+			$this->assertSame( 5, count( $response['data']['items'] ) );
+		} else {
+			$this->assertSame( $expected_results, $response );
+		}
+
+	}
+
+	/**
+	 *
+	 * Data provider.
+	 *
+	 * Provides various post_args to test the results.
+	 *
+	 * @since 4.3.0
+	 *
+	 * @return array {
+	 *     @type array {
+	 *         @string string $post_args        The args that will be passed to ajax.
+	 *         @array  array  $expected_results The expected results from the ajax call.
+	 *     }
+	 * }
+	 *
+	 */
+	function data_ajax_search_available_items_results() {
+		return array(
+			array(
+				array(),
+				array(
+					'success'   => false,
+					'data'      => 'nav_menus_missing_search_parameter',
+				),
+			),
+			array(
+				array(
+					'search' => 'all_the_things',
+				),
+				array(
+					'success' => false,
+					'data' => array(
+						'message' => 'No results found.',
+					),
+				),
+			),
+			array(
+				array(
+					'search' => 'test',
+				),
+				array(
+					'success' => true,
+					array(),
+				),
+			),
+		);
+	}
+}
Index: tests/phpunit/tests/customize/nav-menus.php
===================================================================
--- tests/phpunit/tests/customize/nav-menus.php	(revision 33208)
+++ tests/phpunit/tests/customize/nav-menus.php	(working copy)
@@ -48,29 +48,8 @@
 		$this->assertInstanceOf( 'WP_Customize_Manager', $menus->manager );
 	}
 
-	/**
-	 * Test the test_load_available_items_ajax method.
-	 *
-	 * @see WP_Customize_Nav_Menus::load_available_items_ajax()
-	 */
-	function test_load_available_items_ajax() {
 
-		$this->markTestIncomplete( 'This test has not been implemented.' );
-
-	}
-
 	/**
-	 * Test the search_available_items_ajax method.
-	 *
-	 * @see WP_Customize_Nav_Menus::search_available_items_ajax()
-	 */
-	function test_search_available_items_ajax() {
-
-		$this->markTestIncomplete( 'This test has not been implemented.' );
-
-	}
-
-	/**
 	 * Test the search_available_items_query method.
 	 *
 	 * @see WP_Customize_Nav_Menus::search_available_items_query()