Ticket #41111: 41111.diff

src/wp-includes/js/wp-api.js

@@ -1402 +1402 @@
         // The wp.api.init function returns a promise that will resolve with the endpoint once it is ready.
         wp.api.loadPromise = wp.api.init();
 
+        /**
+         * Determine model based on API route.
+         *
+         * @param {string} route    The API route.
+         *
+         * @return {Backbone Model} The model found at given route.
+         */
+        wp.api.getModelByRoute = function( route ) {
+                return _.first( _.filter( wp.api.models, function( model ) {
+                        return model.prototype.route && route === model.prototype.route.index;
+                } ) );
+        };
+
+        /**
+         * Determine collection based on API route.
+         *
+         * @param {string} route    The API route.
+         *
+         * @return {Backbone Model} The collection found at given route.
+         */
+        wp.api.getCollectionByRoute = function( route ) {
+                return _.first( _.filter( wp.api.collections, function( collection ) {
+                        return collection.prototype.route && route === collection.prototype.route.index;
+                } ) );
+        };
+
 } )();

tests/qunit/wp-includes/js/wp-api.js

@@ -194 +194 @@
                 } );
         } );
 
+        // Find models by route.
+        var modelsToFetchByRoute = [
+                'Category',
+                'Comment',
+                'Media',
+                'Page',
+                'PageRevision',
+                'Post',
+                'PostRevision',
+                'Status',
+                'Tag',
+                'Taxonomy',
+                'Type',
+                'User',
+        ];
+
+        _.each( modelsToFetchByRoute, function( model ) {
+                QUnit.test( 'Test fetching ' + model + ' by route.', function( assert ) {
+
+                        var done = assert.async();
+
+                        assert.expect( 1 );
+
+                        wp.api.loadPromise.done( function() {
+
+                                var theModel = wp.api.models[ model ];
+                                var route = theModel.prototype.route.index;
+
+                                assert.equal(
+                                        wp.api.getModelByRoute( route ),
+                                        theModel,
+                                        'wp.api.models.' + model + ' found at route ' + route
+                                );
+
+                                // Trigger Qunit async completion.
+                                done();
+                        } );
+                } );
+        } );
+
+        // Find collections by route.
+        var collectionsToFetchByRoute = [
+                'Categories',
+                'Comments',
+                'Media',
+                'PageRevisions',
+                'Pages',
+                'PostRevisions',
+                'Posts',
+                'Statuses',
+                'Tags',
+                'Taxonomies',
+                'Types',
+                'Users',
+        ];
+
+        _.each( collectionsToFetchByRoute, function( collection ) {
+                QUnit.test( 'Test fetching ' + collection + ' by route.', function( assert ) {
+
+                        var done = assert.async();
+
+                        assert.expect( 1 );
+
+                        wp.api.loadPromise.done( function() {
+
+                                var theCollection = wp.api.collections[ collection ];
+                                var route = theCollection.prototype.route.index;
+
+                                assert.equal(
+                                        wp.api.getCollectionByRoute( route ),
+                                        theCollection,
+                                        'wp.api.collections.' + collection + ' found at ' + route
+                                );
+
+                                // Trigger Qunit async completion.
+                                done();
+                        } );
+                } );
+        } );
+
         // Test the jswidget custom namespace and endpoints.
         wp.api.init( {
                 'versionString': 'js-widgets/v1/'