patilswapnilv / shortcodely

        $this->assertEquals(2, $found);

    }

    public function test_post_type_single_value() 

    {

        register_post_type('post-type-1');

        register_post_type('post-type-2');

        $p1 = self::factory()->post->create(array( 'post_type' => 'post-type-1' ));

        $p2 = self::factory()->post->create(array( 'post_type' => 'post-type-2' ));

        $c1 = self::factory()->comment->create_post_comments($p1, 1);

        $c2 = self::factory()->comment->create_post_comments($p2, 1);

        $q = new WP_Comment_Query();

        $found = $q->query(

            array(

            'fields' => 'ids',

            'post_type' => 'post-type-2',

            ) 

        );

        $this->assertEqualSets($c2, $found);

        _unregister_post_type('post-type-1');

        _unregister_post_type('post-type-2');

    }

    /**

     * @ticket 20006

     */

    public function test_post_type_singleton_array() 

    {

        register_post_type('post-type-1');

        register_post_type('post-type-2');

        $p1 = self::factory()->post->create(array( 'post_type' => 'post-type-1' ));

        $p2 = self::factory()->post->create(array( 'post_type' => 'post-type-2' ));

        $c1 = self::factory()->comment->create_post_comments($p1, 1);

        $c2 = self::factory()->comment->create_post_comments($p2, 1);

        $q = new WP_Comment_Query();

        $found = $q->query(

            array(

            'fields' => 'ids',

            'post_type' => array( 'post-type-2' ),

            ) 

        );

        $this->assertEqualSets($c2, $found);

        _unregister_post_type('post-type-1');

        _unregister_post_type('post-type-2');

    }