test-timber-wp-functions.php

<?php

use Timber\FunctionWrapper;
use Timber\Helper;

class TestTimberWPFunctions extends Timber_UnitTestCase
{
    public function testFunctionFire()
    {
        $str = '{{function("my_test_function")}}';
        $output = Timber::compile_string($str);
        $this->assertEquals('jared sez hi', $output);
    }

    /**
     * @expectedDeprecated the_block_template_skip_link
     */
    public function testFooterOnFooterFW()
    {
        global $wp_scripts;
        $wp_scripts = null;
        wp_enqueue_script('jquery', false, [], false, true);
        $fw1 = new FunctionWrapper('wp_footer', [], true);
        $fw2 = new FunctionWrapper('wp_footer', [], true);
        $this->assertGreaterThan(50, strlen($fw1->call()));
        //this is bunk because footer scripts will only print once
        $this->assertSame(0, strlen($fw2->call()));
        wp_dequeue_script('jquery');
        $wp_footer_output1 = new FunctionWrapper('wp_footer', [], true);
        $this->assertSame(0, strlen($wp_footer_output1));
    }

    /**
     * @expectedDeprecated the_block_template_skip_link
     */
    public function testFooterAlone()
    {
        global $wp_scripts;
        $wp_scripts = null;
        wp_enqueue_script('jquery', false, [], false, true);
        $fw1 = new FunctionWrapper('wp_footer', [], true);
        $this->assertGreaterThan(50, strlen($fw1->call()));
    }

    public function testDoubleAction()
    {
        add_action('jared_action', function () {
            echo 'bar';
        });
        $fw1 = new FunctionWrapper('do_jared_action', [], true);
        $fw2 = new FunctionWrapper('do_jared_action', [], true);
        $this->assertEquals($fw1->call(), $fw2->call());
        $this->assertEquals('bar', $fw1->call());
    }

    /**
     * @expectedDeprecated the_block_template_skip_link
     */
    public function testDoubleActionWPFooter()
    {
        global $wp_scripts;
        $wp_scripts = null;
        add_action('wp_footer', 'echo_junk');
        $fw1 = new FunctionWrapper('wp_footer', [], true);
        $fw2 = new FunctionWrapper('wp_footer', [], true);
        $this->assertEquals($fw1->call(), $fw2->call());
        $pos = strpos($fw2->call(), 'foo');
        $this->assertGreaterThan(-1, $pos);
        remove_action('wp_footer', 'echo_junk');
    }

    /**
     * @expectedDeprecated the_block_template_skip_link
     */
    public function testInTwig()
    {
        global $wp_scripts;
        $wp_scripts = null;
        wp_enqueue_script('jquery', false, [], false, true);
        $str = Timber::compile('assets/wp-footer.twig', []);
        $pos = strpos($str, 'wp-includes/js/jquery/jquery');
        $this->assertGreaterThan(-1, $pos);
    }

    /**
     * @expectedDeprecated the_block_template_skip_link
     */
    public function testInTwigString()
    {
        global $wp_scripts;
        $wp_scripts = null;
        wp_enqueue_script('jquery', false, [], false, true);
        $str = Timber::compile_string('{{function("wp_footer")}}', []);
        $pos = strpos($str, 'wp-includes/js/jquery/jquery');
        $this->assertGreaterThan(-1, $pos);
    }

    /**
     * @expectedDeprecated the_block_template_skip_link
     */
    public function testAgainstFooterFunctionOutput()
    {
        global $wp_scripts;
        $wp_scripts = null;
        wp_enqueue_script('colorpicker', false, [], false, true);
        wp_enqueue_script('fake-js', 'http://example.org/fake-js.js', [], false, true);
        $wp_footer = Helper::ob_function('wp_footer');
        global $wp_scripts;
        $wp_scripts = null;
        wp_enqueue_script('colorpicker', false, [], false, true);
        wp_enqueue_script('fake-js', 'http://example.org/fake-js.js', [], false, true);
        $str = Timber::compile_string('{{function("wp_footer")}}');
        $this->assertEquals($wp_footer, $str);
        $this->assertGreaterThan(50, strlen($str));
    }

    public function testInTwigStringHeadAndFooter()
    {
        global $wp_scripts;
        $wp_scripts = null;
        //send colorpicker to the header
        wp_enqueue_script('colorpicker', false, [], false, false);
        //send fake-js to the footer
        wp_enqueue_script('fake-js', 'http://example.org/fake-js.js', [], false, true);
        $str = Timber::compile_string('<head>{{function("wp_head")}}</head><footer>{{function("wp_footer")}}</footer>');
        $footer_tag = strpos($str, '<footer>');
        $colorpicker = strpos($str, 'colorpicker');
        $this->assertGreaterThan(1, $colorpicker);
        //make sure that footer appears after colorpicker
        $this->assertGreaterThan($colorpicker, $footer_tag);
    }
}

function do_jared_action()
{
    do_action('jared_action');
}

function echo_junk()
{
    echo 'foo';
}

function my_test_function()
{
    return 'jared sez hi';
}