Ticket #37008: 37008.diff

tests/phpunit/tests/functions.php

@@ -887 +887 @@
 
                 $this->assertNull( wp_ext2type( 'unknown_format' ) );
         }
+
+        /**
+         * @ticket 37008
+         * @dataProvider uri_list_provider
+         */
+        public function test_remove_query_arg( $actual, $expected, $keys_to_remove ) {
+                $old_request_uri = $_SERVER['REQUEST_URI'];
+                $_SERVER['REQUEST_URI'] = $actual;
+
+                $uri_result = remove_query_arg( $keys_to_remove );
+
+                $this->assertNotEmpty( $uri_result, 'The URI cannot to be empty' );
+                $this->assertEquals(
+                        $expected,
+                        $uri_result,
+                        "The final URI after removing the keys isn't the expected one"
+                );
+
+                $_SERVER['REQUEST_URI'] = $old_request_uri;
+        }
+
+        function uri_list_provider() {
+                return array(
+                        array( 'edit.php?foo=test1&baz=test1', 'edit.php?baz=test1' , 'foo' ),
+                        array( 'edit.php?foo=test2&baz=test2', 'edit.php?baz=test2' , array( 'foo' ) ),
+                        array( 'edit.php?foo=test3&baz=test3', 'edit.php' , array( 'foo', 'baz' ) ),
+                        array( 'edit.php?foo=test4&baz=test4', 'edit.php?foo=test4&baz=test4' , array( 'fakefoo', 'fakebaz' ) ),
+                        array( 'edit.php?foo=test4&baz=test4', 'edit.php?foo=test4' , array( 'fakefoo', 'baz' ) ),
+                );
+        }
 }