Ticket #29839: 20141116_05-getTerms.diff

tests/phpunit/tests/term/getTerms.php

@@ -424 +424 @@
                 $this->assertEquals( array( $t1, $t3 ), $found );
         }
 
+        /**
+         * @ticket 29839
+         */
+        function test_get_terms_childless_flat() {
+                // If run on a flat hierarchy it should return everything.
+                $flat_tax = 'countries';
+                register_taxonomy( $flat_tax, 'post', array( 'hierarchical' => false ) );
+                $australia = $this->factory->term->create( array( 'name' => 'Australia', 'taxonomy' => $flat_tax ) );
+                $china  =    $this->factory->term->create( array( 'name' => 'China',     'taxonomy' => $flat_tax ) );
+                $tanzania =  $this->factory->term->create( array( 'name' => 'Tanzania',  'taxonomy' => $flat_tax ) );
+
+                // Note that wp_set_post_terms() expects parameter 2 to be a string when the taxonomy is flat
+                //  so lets redefine the term IDs created above to be the term objects
+                $australia = get_term( $australia, $flat_tax );
+                $china  =    get_term( $china, $flat_tax );
+                $tanzania =  get_term( $tanzania, $flat_tax );
+
+                $post4_id = $this->factory->post->create();
+                $post5_id = $this->factory->post->create();
+
+                wp_set_post_terms( $post4_id, $australia->name, $flat_tax );
+                wp_set_post_terms( $post5_id, $tanzania->name,  $flat_tax );
+
+
+                // Test 1: hide_empty = false
+                $terms = get_terms( $flat_tax, array( 'childless' => true, 'hide_empty' => false) );
+
+                // Expected: Australia, China, Tanzania
+                $this->assertEquals( 3, count($terms) );
+                $this->assertEqualSets( array($australia->term_id, $china->term_id, $tanzania->term_id), wp_list_pluck( $terms, 'term_id' ) );
+
+
+                // Test 2: hide_empty = true
+                $terms = get_terms( $flat_tax, array( 'childless' => true, 'hide_empty' => true) );
+
+                // Expected: Australia, Tanzania
+                $this->assertEquals( 2, count($terms) );
+                $this->assertEqualSets( array($australia->term_id, $tanzania->term_id), wp_list_pluck( $terms, 'term_id' ) );
+
+
+                // Test 3: these two should be equal
+                $terms_1 = get_terms( $flat_tax, array( 'childless' => true, 'hide_empty' => false) );
+                $terms_2 = get_terms( $flat_tax, array( 'get' => 'all',      'hide_empty' => false ) );
+
+                $this->assertEqualSets(wp_list_pluck( $terms_1, 'term_id' ), wp_list_pluck( $terms_2, 'term_id' ));
+
+                _unregister_taxonomy( $flat_tax );
+        }
+
+
+        /**
+         * @ticket 29839
+         */
+        function test_get_terms_childless_hierarchical() {
+                $tax = 'location';
+                register_taxonomy( $tax, 'post', array( 'hierarchical' => true ) );
+                /*
+                Canada
+                        Ontario
+                                Ottawa
+                                        Nepean
+                                Toronto
+                        Quebec
+                                Montreal
+                        PEI
+                */
+                // Level 1
+                $canada = $this->factory->term->create( array( 'name' => 'Canada', 'taxonomy' => $tax ) );
+
+                // Level 2
+                $ontario = $this->factory->term->create( array( 'name' => 'Ontario', 'parent' => $canada, 'taxonomy' => $tax ) );
+                $quebec  = $this->factory->term->create( array( 'name' => 'Quebec',  'parent' => $canada, 'taxonomy' => $tax ) );
+                $pei     = $this->factory->term->create( array( 'name' => 'PEI', '    parent' => $canada, 'taxonomy' => $tax ) );
+
+                // Level 3
+                $toronto=  $this->factory->term->create( array( 'name' => 'Toronto',  'parent' => $ontario, 'taxonomy' => $tax ) );
+                $ottawa=   $this->factory->term->create( array( 'name' => 'Ottawa',   'parent' => $ontario, 'taxonomy' => $tax ) );
+                $montreal= $this->factory->term->create( array( 'name' => 'Montreal', 'parent' => $quebec,  'taxonomy' => $tax ) );
+
+                // Level 4
+                $nepean = $this->factory->term->create( array( 'name' => 'Nepean', 'parent' => $ottawa, 'taxonomy' => $tax ) );
+
+                $post1_id = $this->factory->post->create();
+                $post2_id = $this->factory->post->create();
+                $post3_id = $this->factory->post->create();
+
+                wp_set_post_terms( $post1_id, $nepean, $tax );
+                wp_set_post_terms( $post2_id, $montreal, $tax );
+                wp_set_post_terms( $post3_id, $ottawa, $tax );
+
+
+                // Test 1: hide_empty = false
+                $terms = get_terms( $tax, array( 'childless' => true, 'hide_empty' => false) );
+
+                // Expected: Montreal, Nepean, PEI, Toronto
+                $this->assertEquals( 4, count($terms) );
+                $this->assertEqualSets( array($montreal, $nepean, $pei, $toronto), wp_list_pluck( $terms, 'term_id' ) );
+
+
+                // Test 2: hide_empty = true
+                $terms = get_terms( $tax, array( 'childless' => true, 'hide_empty' => true) );
+
+                // Expected: Montreal, Nepean
+                $this->assertEquals( 2, count($terms) );
+                $this->assertEqualSets( array($montreal, $nepean), wp_list_pluck( $terms, 'term_id' ) );
+
+
+                // Test 3: child_of = Quebec
+                $terms = get_terms( $tax, array( 'childless' => true, 'hide_empty' => false, 'child_of' => $quebec ) );
+
+                // Expected: Montreal
+                $this->assertEquals( 1, count($terms) );
+                $this->assertEqualSets( array( $montreal ), wp_list_pluck( $terms, 'term_id' ) );
+
+
+                // Test 4: all
+                $terms = get_terms( $tax, array( 'childless' => true, 'hide_empty' => false, 'get' => 'all' ) );
+                $terms_2 = get_terms( $tax, array( 'hide_empty' => false, 'get' => 'all' ) );
+
+                // Expected: both match
+                $this->assertEqualSets( wp_list_pluck( $terms, 'term_id' ), wp_list_pluck( $terms_2, 'term_id' ) );
+
+                _unregister_taxonomy( $tax );
+        }
+
         public function test_get_terms_hierarchical_tax_hide_empty_false_fields_ids() {
                 // Set up a clean taxonomy.
                 $tax = 'hierarchical_fields';