Make WordPress Core

Ticket #39776: 39776.5.diff

File 39776.5.diff, 17.1 KB (added by welcher, 5 years ago)

Adds unit tests and better logic around negative and large numbers

  • src/wp-admin/includes/plugin.php

    diff --git src/wp-admin/includes/plugin.php src/wp-admin/includes/plugin.php
    index 683cb7c215..2ad6beda1c 100644
    function add_menu_page( $page_title, $menu_title, $capability, $menu_slug, $func 
    13381338 *                              and only include lowercase alphanumeric, dashes, and underscores characters
    13391339 *                              to be compatible with sanitize_key().
    13401340 * @param callable $function    The function to be called to output the content for this page.
     1341 * @param int      $position    The position order in the menu where this item should appear.
     1342 *
    13411343 * @return false|string The resulting page's hook_suffix, or false if the user does not have the capability required.
    13421344 */
    1343 function add_submenu_page( $parent_slug, $page_title, $menu_title, $capability, $menu_slug, $function = '' ) {
     1345function add_submenu_page( $parent_slug, $page_title, $menu_title, $capability, $menu_slug, $function = '', $position = null ) {
    13441346        global $submenu, $menu, $_wp_real_parent_file, $_wp_submenu_nopriv,
    13451347                $_registered_pages, $_parent_pages;
    13461348
    function add_submenu_page( $parent_slug, $page_title, $menu_title, $capability, 
    13701372                }
    13711373        }
    13721374
    1373         $submenu[ $parent_slug ][] = array( $menu_title, $capability, $menu_slug, $page_title );
     1375        $new_sub_menu = array( $menu_title, $capability, $menu_slug, $page_title );
     1376        if ( null === $position ) {
     1377                $submenu[ $parent_slug ][] = $new_sub_menu;
     1378        } else {
     1379                // If position is equal or higher than the number of items in the array, just append it.
     1380                if ( $position >= count( $submenu[ $parent_slug ] ) ) {
     1381                        $submenu[ $parent_slug ][] = $new_sub_menu;
     1382                } else {
     1383                        // Ensure we don't have a negative position.
     1384                        $position = max( $position, 0 );
     1385                        // Grab all of the items before the insertion point.
     1386                        $before_items = array_slice( $submenu[ $parent_slug ], 0, $position, true );
     1387                        // Grab all of the items after the insertion point
     1388                        $after_items = array_slice( $submenu[ $parent_slug ], $position, null, true );
     1389                        // Add the new item
     1390                        $before_items[] = $new_sub_menu;
     1391                        // Merge the items.
     1392                        $submenu[ $parent_slug ] = array_merge( $before_items, $after_items );
     1393                }
     1394        }
     1395        // Sort the parent array
     1396        ksort( $submenu[ $parent_slug ] );
    13741397
    13751398        $hookname = get_plugin_page_hookname( $menu_slug, $parent_slug );
    13761399        if ( ! empty( $function ) && ! empty( $hookname ) ) {
    function add_submenu_page( $parent_slug, $page_title, $menu_title, $capability, 
    14091432 * @param string   $capability The capability required for this menu to be displayed to the user.
    14101433 * @param string   $menu_slug  The slug name to refer to this menu by (should be unique for this menu).
    14111434 * @param callable $function   The function to be called to output the content for this page.
     1435 * @param int      $position   The position in the menu order this one should appear.
    14121436 * @return false|string The resulting page's hook_suffix, or false if the user does not have the capability required.
    14131437 */
    1414 function add_management_page( $page_title, $menu_title, $capability, $menu_slug, $function = '' ) {
    1415         return add_submenu_page( 'tools.php', $page_title, $menu_title, $capability, $menu_slug, $function );
     1438function add_management_page( $page_title, $menu_title, $capability, $menu_slug, $function = '', $position = null ) {
     1439        return add_submenu_page( 'tools.php', $page_title, $menu_title, $capability, $menu_slug, $function, $position );
    14161440}
    14171441
    14181442/**
    function add_management_page( $page_title, $menu_title, $capability, $menu_slug, 
    14311455 * @param string   $capability The capability required for this menu to be displayed to the user.
    14321456 * @param string   $menu_slug  The slug name to refer to this menu by (should be unique for this menu).
    14331457 * @param callable $function   The function to be called to output the content for this page.
     1458 * @param int      $position   The position in the menu order this one should appear.
    14341459 * @return false|string The resulting page's hook_suffix, or false if the user does not have the capability required.
    14351460 */
    1436 function add_options_page( $page_title, $menu_title, $capability, $menu_slug, $function = '' ) {
    1437         return add_submenu_page( 'options-general.php', $page_title, $menu_title, $capability, $menu_slug, $function );
     1461function add_options_page( $page_title, $menu_title, $capability, $menu_slug, $function = '', $position = null ) {
     1462        return add_submenu_page( 'options-general.php', $page_title, $menu_title, $capability, $menu_slug, $function, $position );
    14381463}
    14391464
    14401465/**
    function add_options_page( $page_title, $menu_title, $capability, $menu_slug, $f 
    14531478 * @param string   $capability The capability required for this menu to be displayed to the user.
    14541479 * @param string   $menu_slug  The slug name to refer to this menu by (should be unique for this menu).
    14551480 * @param callable $function   The function to be called to output the content for this page.
     1481 * @param int      $position   The position in the menu order this one should appear.
    14561482 * @return false|string The resulting page's hook_suffix, or false if the user does not have the capability required.
    14571483 */
    1458 function add_theme_page( $page_title, $menu_title, $capability, $menu_slug, $function = '' ) {
    1459         return add_submenu_page( 'themes.php', $page_title, $menu_title, $capability, $menu_slug, $function );
     1484function add_theme_page( $page_title, $menu_title, $capability, $menu_slug, $function = '', $position = null ) {
     1485        return add_submenu_page( 'themes.php', $page_title, $menu_title, $capability, $menu_slug, $function, $position );
    14601486}
    14611487
    14621488/**
    function add_theme_page( $page_title, $menu_title, $capability, $menu_slug, $fun 
    14751501 * @param string   $capability The capability required for this menu to be displayed to the user.
    14761502 * @param string   $menu_slug  The slug name to refer to this menu by (should be unique for this menu).
    14771503 * @param callable $function   The function to be called to output the content for this page.
     1504 * @param int      $position   The position in the menu order this one should appear.
    14781505 * @return false|string The resulting page's hook_suffix, or false if the user does not have the capability required.
    14791506 */
    1480 function add_plugins_page( $page_title, $menu_title, $capability, $menu_slug, $function = '' ) {
    1481         return add_submenu_page( 'plugins.php', $page_title, $menu_title, $capability, $menu_slug, $function );
     1507function add_plugins_page( $page_title, $menu_title, $capability, $menu_slug, $function = '', $position = null ) {
     1508        return add_submenu_page( 'plugins.php', $page_title, $menu_title, $capability, $menu_slug, $function, $position );
    14821509}
    14831510
    14841511/**
    function add_plugins_page( $page_title, $menu_title, $capability, $menu_slug, $f 
    14971524 * @param string   $capability The capability required for this menu to be displayed to the user.
    14981525 * @param string   $menu_slug  The slug name to refer to this menu by (should be unique for this menu).
    14991526 * @param callable $function   The function to be called to output the content for this page.
     1527 * @param int      $position   The position in the menu order this one should appear.
    15001528 * @return false|string The resulting page's hook_suffix, or false if the user does not have the capability required.
    15011529 */
    1502 function add_users_page( $page_title, $menu_title, $capability, $menu_slug, $function = '' ) {
    1503         if ( current_user_can( 'edit_users' ) ) {
     1530
     1531function add_users_page( $page_title, $menu_title, $capability, $menu_slug, $function = '', $position = null ) {
     1532        if ( current_user_can('edit_users') ) {
    15041533                $parent = 'users.php';
    15051534        } else {
    15061535                $parent = 'profile.php';
    15071536        }
    1508         return add_submenu_page( $parent, $page_title, $menu_title, $capability, $menu_slug, $function );
     1537        return add_submenu_page( $parent, $page_title, $menu_title, $capability, $menu_slug, $function, $position );
    15091538}
    15101539/**
    15111540 * Add submenu page to the Dashboard main menu.
    function add_users_page( $page_title, $menu_title, $capability, $menu_slug, $fun 
    15231552 * @param string   $capability The capability required for this menu to be displayed to the user.
    15241553 * @param string   $menu_slug  The slug name to refer to this menu by (should be unique for this menu).
    15251554 * @param callable $function   The function to be called to output the content for this page.
     1555 * @param int      $position   The position in the menu order this one should appear.
    15261556 * @return false|string The resulting page's hook_suffix, or false if the user does not have the capability required.
    15271557 */
    1528 function add_dashboard_page( $page_title, $menu_title, $capability, $menu_slug, $function = '' ) {
    1529         return add_submenu_page( 'index.php', $page_title, $menu_title, $capability, $menu_slug, $function );
     1558function add_dashboard_page( $page_title, $menu_title, $capability, $menu_slug, $function = '', $position = null ) {
     1559        return add_submenu_page( 'index.php', $page_title, $menu_title, $capability, $menu_slug, $function, $position );
    15301560}
    15311561
    15321562/**
    function add_dashboard_page( $page_title, $menu_title, $capability, $menu_slug, 
    15451575 * @param string   $capability The capability required for this menu to be displayed to the user.
    15461576 * @param string   $menu_slug  The slug name to refer to this menu by (should be unique for this menu).
    15471577 * @param callable $function   The function to be called to output the content for this page.
     1578 * @param int      $position   The position in the menu order this one should appear.
    15481579 * @return false|string The resulting page's hook_suffix, or false if the user does not have the capability required.
    15491580 */
    1550 function add_posts_page( $page_title, $menu_title, $capability, $menu_slug, $function = '' ) {
    1551         return add_submenu_page( 'edit.php', $page_title, $menu_title, $capability, $menu_slug, $function );
     1581function add_posts_page( $page_title, $menu_title, $capability, $menu_slug, $function = '', $position = null ) {
     1582        return add_submenu_page( 'edit.php', $page_title, $menu_title, $capability, $menu_slug, $function, $position );
    15521583}
    15531584
    15541585/**
    function add_posts_page( $page_title, $menu_title, $capability, $menu_slug, $fun 
    15671598 * @param string   $capability The capability required for this menu to be displayed to the user.
    15681599 * @param string   $menu_slug  The slug name to refer to this menu by (should be unique for this menu).
    15691600 * @param callable $function   The function to be called to output the content for this page.
     1601 * @param int      $position   The position in the menu order this one should appear.
    15701602 * @return false|string The resulting page's hook_suffix, or false if the user does not have the capability required.
    15711603 */
    1572 function add_media_page( $page_title, $menu_title, $capability, $menu_slug, $function = '' ) {
    1573         return add_submenu_page( 'upload.php', $page_title, $menu_title, $capability, $menu_slug, $function );
     1604function add_media_page( $page_title, $menu_title, $capability, $menu_slug, $function = '', $position = null ) {
     1605        return add_submenu_page( 'upload.php', $page_title, $menu_title, $capability, $menu_slug, $function, $position );
    15741606}
    15751607
    15761608/**
    function add_media_page( $page_title, $menu_title, $capability, $menu_slug, $fun 
    15891621 * @param string   $capability The capability required for this menu to be displayed to the user.
    15901622 * @param string   $menu_slug  The slug name to refer to this menu by (should be unique for this menu).
    15911623 * @param callable $function   The function to be called to output the content for this page.
     1624 * @param int      $position   The position in the menu order this one should appear.
    15921625 * @return false|string The resulting page's hook_suffix, or false if the user does not have the capability required.
    15931626 */
    1594 function add_links_page( $page_title, $menu_title, $capability, $menu_slug, $function = '' ) {
    1595         return add_submenu_page( 'link-manager.php', $page_title, $menu_title, $capability, $menu_slug, $function );
     1627function add_links_page( $page_title, $menu_title, $capability, $menu_slug, $function = '', $position = null ) {
     1628        return add_submenu_page( 'link-manager.php', $page_title, $menu_title, $capability, $menu_slug, $function, $position );
    15961629}
    15971630
    15981631/**
    function add_links_page( $page_title, $menu_title, $capability, $menu_slug, $fun 
    16111644 * @param string   $capability The capability required for this menu to be displayed to the user.
    16121645 * @param string   $menu_slug  The slug name to refer to this menu by (should be unique for this menu).
    16131646 * @param callable $function   The function to be called to output the content for this page.
     1647 * @param int      $position   The position in the menu order this one should appear.
    16141648 * @return false|string The resulting page's hook_suffix, or false if the user does not have the capability required.
    16151649 */
    1616 function add_pages_page( $page_title, $menu_title, $capability, $menu_slug, $function = '' ) {
    1617         return add_submenu_page( 'edit.php?post_type=page', $page_title, $menu_title, $capability, $menu_slug, $function );
     1650function add_pages_page( $page_title, $menu_title, $capability, $menu_slug, $function = '', $position = null ) {
     1651        return add_submenu_page( 'edit.php?post_type=page', $page_title, $menu_title, $capability, $menu_slug, $function, $position);
    16181652}
    16191653
    16201654/**
    function add_pages_page( $page_title, $menu_title, $capability, $menu_slug, $fun 
    16331667 * @param string   $capability The capability required for this menu to be displayed to the user.
    16341668 * @param string   $menu_slug  The slug name to refer to this menu by (should be unique for this menu).
    16351669 * @param callable $function   The function to be called to output the content for this page.
     1670 * @param int      $position   The position in the menu order this one should appear.
    16361671 * @return false|string The resulting page's hook_suffix, or false if the user does not have the capability required.
    16371672 */
    1638 function add_comments_page( $page_title, $menu_title, $capability, $menu_slug, $function = '' ) {
    1639         return add_submenu_page( 'edit-comments.php', $page_title, $menu_title, $capability, $menu_slug, $function );
     1673function add_comments_page( $page_title, $menu_title, $capability, $menu_slug, $function = '', $position = null ) {
     1674        return add_submenu_page( 'edit-comments.php', $page_title, $menu_title, $capability, $menu_slug, $function, $position );
    16401675}
    16411676
    16421677/**
  • tests/phpunit/tests/admin/includesPlugin.php

    diff --git tests/phpunit/tests/admin/includesPlugin.php tests/phpunit/tests/admin/includesPlugin.php
    index 916ca755b7..ce090d1643 100644
    class Tests_Admin_includesPlugin extends WP_UnitTestCase { 
    5757                wp_set_current_user( $current_user );
    5858        }
    5959
     60        /**
     61         * Tests the priority parameter.
     62         *
     63         * @param int $priority          The position of the new item.
     64         * @param int $expected_position Where the new item is expected to appear.
     65         *
     66         * @dataProvider data_submenu_priority
     67         */
     68        function test_submenu_priority( $priority, $expected_position  ) {
     69                global $submenu;
     70                $current_user = get_current_user_id();
     71                wp_set_current_user( self::factory()->user->create( array( 'role' => 'administrator' ) ) );
     72                set_current_screen( 'dashboard' );
     73
     74                // Setup a menu with some items.
     75                $parent = add_menu_page( 'Test Toplevel', 'Test Toplevel', 'manage_options', 'mt-top-level-handle', 'mt_toplevel_page' );
     76                foreach ( $this->submenus_to_add() as $menu ) {
     77                        add_submenu_page( $parent, $menu[0], $menu[1], $menu[2], $menu[3], $menu[4] );
     78                }
     79
     80                // Insert the new page.
     81                add_submenu_page( $parent, 'New Page', 'New Page', 'manage_options', 'custom-position', 'custom_pos' , $priority );
     82
     83                $this->assertSame( 'custom-position', $submenu[ $parent ][ $expected_position ][2] );
     84
     85                wp_set_current_user( $current_user );
     86        }
     87
     88        /**
     89         * Helper to store the menus to add so getting the length is programtically done.
     90         *
     91         * @return array
     92         */
     93        function submenus_to_add() {
     94                return array(
     95                        array( 'Submenu Priority', 'Submenu Priority', 'manage_options', 'sub-page', '' ),
     96                        array( 'Submenu Priority 2', 'Submenu Priority 2', 'manage_options', 'sub-page2', '' ),
     97                        array( 'Submenu Priority 3', 'Submenu Priority 3', 'manage_options', 'sub-page3', '' ),
     98                        array( 'Submenu Priority 4', 'Submenu Priority 4', 'manage_options', 'sub-page4', '' ),
     99                        array( 'Submenu Priority 5', 'Submenu Priority 5', 'manage_options', 'sub-page5', '' ),
     100                );
     101        }
     102        /**
     103         * Data provider for the above tests
     104         * @return array
     105         */
     106        function data_submenu_priority() {
     107                $menu_count = count( $this->submenus_to_add() );
     108                return array(
     109                        array( null, $menu_count ),        // Insert at the end of the menu is null is passed. Default behaviour.
     110                        array( 0, 0 ),                     // Insert at the beginning of the menu if 0 is passed.
     111                        array( -1, 0 ),                    // Negative numbers are treated the same as passing 0.
     112                        array( -7, 0 ),                    // Negative numbers are treated the same as passing 0.
     113                        array( 1, 1 ),                     // Insert as the second item.
     114                        array( 3, 3 ),                     // Insert as the 4th item.
     115                        array( $menu_count, $menu_count ), // Numbers equal to the number of items are added at the end.
     116                        array( 123456, $menu_count )       // Numbers higher than the number of items are added at the end.
     117                );
     118        }
     119
    60120        function test_is_plugin_active_true() {
    61121                activate_plugin( 'hello.php' );
    62122                $test = is_plugin_active( 'hello.php' );