silverstripe-framework/tests/php/View/ViewableDataTest.php

<?php

namespace SilverStripe\View\Tests;

use ReflectionMethod;
use SilverStripe\ORM\FieldType\DBField;
use SilverStripe\Dev\SapphireTest;
use SilverStripe\ORM\ArrayList;
use SilverStripe\ORM\FieldType\DBText;
use SilverStripe\View\ArrayData;
use SilverStripe\View\SSViewer;
use SilverStripe\View\Tests\ViewableDataTest\ViewableDataTestExtension;
use SilverStripe\View\Tests\ViewableDataTest\ViewableDataTestObject;
use SilverStripe\View\ViewableData;

/**
 * See {@link SSViewerTest->testCastingHelpers()} for more tests related to casting and ViewableData behaviour,
 * from a template-parsing perspective.
 */
class ViewableDataTest extends SapphireTest
{
    protected static $required_extensions = [
        ViewableDataTestObject::class => [
            ViewableDataTestExtension::class,
        ],
    ];

    public function testCasting()
    {
        $htmlString = "&quot;";
        $textString = '"';

        $htmlField = DBField::create_field('HTMLFragment', $textString);

        $this->assertEquals($textString, $htmlField->forTemplate());
        $this->assertEquals($htmlString, $htmlField->obj('HTMLATT')->forTemplate());
        $this->assertEquals('%22', $htmlField->obj('URLATT')->forTemplate());
        $this->assertEquals('%22', $htmlField->obj('RAWURLATT')->forTemplate());
        $this->assertEquals($htmlString, $htmlField->obj('ATT')->forTemplate());
        $this->assertEquals($textString, $htmlField->obj('RAW')->forTemplate());
        $this->assertEquals('\"', $htmlField->obj('JS')->forTemplate());
        $this->assertEquals($htmlString, $htmlField->obj('HTML')->forTemplate());
        $this->assertEquals($htmlString, $htmlField->obj('XML')->forTemplate());

        $textField = DBField::create_field('Text', $textString);
        $this->assertEquals($htmlString, $textField->forTemplate());
        $this->assertEquals($htmlString, $textField->obj('HTMLATT')->forTemplate());
        $this->assertEquals('%22', $textField->obj('URLATT')->forTemplate());
        $this->assertEquals('%22', $textField->obj('RAWURLATT')->forTemplate());
        $this->assertEquals($htmlString, $textField->obj('ATT')->forTemplate());
        $this->assertEquals($textString, $textField->obj('RAW')->forTemplate());
        $this->assertEquals('\"', $textField->obj('JS')->forTemplate());
        $this->assertEquals($htmlString, $textField->obj('HTML')->forTemplate());
        $this->assertEquals($htmlString, $textField->obj('XML')->forTemplate());
    }

    public function testRequiresCasting()
    {
        $caster = new ViewableDataTest\Castable();

        $this->assertInstanceOf(ViewableDataTest\RequiresCasting::class, $caster->obj('alwaysCasted'));
        $this->assertInstanceOf(ViewableDataTest\Caster::class, $caster->obj('noCastingInformation'));

        $this->assertInstanceOf(DBText::class, $caster->obj('arrayOne'));
        $this->assertInstanceOf(ArrayList::class, $caster->obj('arrayTwo'));
    }

    public function testFailoverRequiresCasting()
    {
        $caster = new ViewableDataTest\Castable();
        $container = new ViewableDataTest\Container();
        $container->setFailover($caster);

        $this->assertInstanceOf(ViewableDataTest\RequiresCasting::class, $container->obj('alwaysCasted'));
        $this->assertInstanceOf(ViewableDataTest\RequiresCasting::class, $caster->obj('alwaysCasted'));

        $this->assertInstanceOf(ViewableDataTest\Caster::class, $container->obj('noCastingInformation'));
        $this->assertInstanceOf(ViewableDataTest\Caster::class, $caster->obj('noCastingInformation'));
    }

    public function testCastingXMLVal()
    {
        $caster = new ViewableDataTest\Castable();

        $this->assertEquals('casted', $caster->XML_val('alwaysCasted'));
        $this->assertEquals('casted', $caster->XML_val('noCastingInformation'));

        // Test automatic escaping is applied even to fields with no 'casting'
        $this->assertEquals('casted', $caster->XML_val('unsafeXML'));
        $this->assertEquals('&lt;foo&gt;', $caster->XML_val('castedUnsafeXML'));
    }

    public function testArrayCustomise()
    {
        $viewableData    = new ViewableDataTest\Castable();
        $newViewableData = $viewableData->customise(
            [
            'test'         => 'overwritten',
            'alwaysCasted' => 'overwritten'
             ]
        );

        $this->assertEquals('test', $viewableData->XML_val('test'));
        $this->assertEquals('casted', $viewableData->XML_val('alwaysCasted'));

        $this->assertEquals('overwritten', $newViewableData->XML_val('test'));
        $this->assertEquals('overwritten', $newViewableData->XML_val('alwaysCasted'));

        $this->assertEquals('castable', $viewableData->forTemplate());
        $this->assertEquals('castable', $newViewableData->forTemplate());
    }

    public function testObjectCustomise()
    {
        $viewableData    = new ViewableDataTest\Castable();
        $newViewableData = $viewableData->customise(new ViewableDataTest\RequiresCasting());

        $this->assertEquals('test', $viewableData->XML_val('test'));
        $this->assertEquals('casted', $viewableData->XML_val('alwaysCasted'));

        $this->assertEquals('overwritten', $newViewableData->XML_val('test'));
        $this->assertEquals('casted', $newViewableData->XML_val('alwaysCasted'));

        $this->assertEquals('castable', $viewableData->forTemplate());
        $this->assertEquals('casted', $newViewableData->forTemplate());
    }

    public function testDefaultValueWrapping()
    {
        $data = new ArrayData(['Title' => 'SomeTitleValue']);
        // this results in a cached raw string in ViewableData:
        $this->assertTrue($data->hasValue('Title'));
        $this->assertFalse($data->hasValue('SomethingElse'));
        // this should cast the raw string to a StringField since we are
        // passing true as the third argument:
        $obj = $data->obj('Title', null, true);
        $this->assertTrue(is_object($obj));
        // and the string field should have the value of the raw string:
        $this->assertEquals('SomeTitleValue', $obj->forTemplate());
    }

    public function testCastingClass()
    {
        $expected = [
            //'NonExistant'   => null,
            'Field'         => 'CastingType',
            'Argument'      => 'ArgumentType',
            'ArrayArgument' => 'ArrayArgumentType'
        ];
        $obj = new ViewableDataTest\CastingClass();

        foreach ($expected as $field => $class) {
            $this->assertEquals(
                $class,
                $obj->castingClass($field),
                "castingClass() returns correct results for ::\$$field"
            );
        }
    }

    public function testObjWithCachedStringValueReturnsValidObject()
    {
        $obj = new ViewableDataTest\NoCastingInformation();

        // Save a literal string into cache
        $cache = true;
        $uncastedData = $obj->obj('noCastingInformation', null, false, $cache);

        // Fetch the cached string as an object
        $forceReturnedObject = true;
        $castedData = $obj->obj('noCastingInformation', null, $forceReturnedObject);

        // Uncasted data should always be the nonempty string
        $this->assertNotEmpty($uncastedData, 'Uncasted data was empty.');
        //$this->assertTrue(is_string($uncastedData), 'Uncasted data should be a string.');

        // Casted data should be the string wrapped in a DBField-object.
        $this->assertNotEmpty($castedData, 'Casted data was empty.');
        $this->assertInstanceOf(DBField::class, $castedData, 'Casted data should be instance of DBField.');

        $this->assertEquals($uncastedData, $castedData->getValue(), 'Casted and uncasted strings are not equal.');
    }

    public function testCaching()
    {
        $objCached = new ViewableDataTest\Cached();
        $objNotCached = new ViewableDataTest\NotCached();

        $objCached->Test = 'AAA';
        $objNotCached->Test = 'AAA';

        $this->assertEquals('AAA', $objCached->obj('Test', null, true, true));
        $this->assertEquals('AAA', $objNotCached->obj('Test', null, true, true));

        $objCached->Test = 'BBB';
        $objNotCached->Test = 'BBB';

        // Cached data must be always the same
        $this->assertEquals('AAA', $objCached->obj('Test', null, true, true));
        $this->assertEquals('BBB', $objNotCached->obj('Test', null, true, true));
    }

    public function testSetFailover()
    {
        $failover = new ViewableData();
        $container = new ViewableDataTest\Container();
        $container->setFailover($failover);

        $this->assertSame($failover, $container->getFailover(), 'getFailover() returned a different object');
        $this->assertFalse($container->hasMethod('testMethod'), 'testMethod() is already defined when it shouldn’t be');

        // Ensure that defined methods detected from the failover aren't cached when setting a new failover
        $container->setFailover(new ViewableDataTest\Failover);
        $this->assertTrue($container->hasMethod('testMethod'));

        // Test the reverse - that defined methods previously detected in a failover are removed if they no longer exist
        $container->setFailover($failover);
        $this->assertSame($failover, $container->getFailover(), 'getFailover() returned a different object');
        $this->assertFalse($container->hasMethod('testMethod'), 'testMethod() incorrectly reported as existing');
    }

    public function testIsAccessibleMethod()
    {
        $reflectionMethod = new ReflectionMethod(ViewableData::class, 'isAccessibleMethod');
        $reflectionMethod->setAccessible(true);
        $object = new ViewableDataTestObject();
        $viewableData = new ViewableData();

        $output = $reflectionMethod->invokeArgs($object, ['privateMethod']);
        $this->assertFalse($output, 'Method should not be accessible');

        $output = $reflectionMethod->invokeArgs($object, ['protectedMethod']);
        $this->assertTrue($output, 'Method should be accessible');

        $output = $reflectionMethod->invokeArgs($object, ['publicMethod']);
        $this->assertTrue($output, 'Method should be accessible');

        $output = $reflectionMethod->invokeArgs($object, ['missingMethod']);
        $this->assertFalse($output, 'Method should not be accessible');

        $output = $reflectionMethod->invokeArgs($viewableData, ['isAccessibleProperty']);
        $this->assertTrue($output, 'Method should be accessible');

        $output = $reflectionMethod->invokeArgs($object, ['publicMethodFromExtension']);
        $this->assertTrue($output, 'Method should be accessible');

        $output = $reflectionMethod->invokeArgs($object, ['protectedMethodFromExtension']);
        $this->assertFalse($output, 'Method should not be accessible');

        $output = $reflectionMethod->invokeArgs($object, ['privateMethodFromExtension']);
        $this->assertFalse($output, 'Method should not be accessible');
    }

    public function testIsAccessibleProperty()
    {
        $reflectionMethod = new ReflectionMethod(ViewableData::class, 'isAccessibleProperty');
        $reflectionMethod->setAccessible(true);
        $object = new ViewableDataTestObject();

        $output = $reflectionMethod->invokeArgs($object, ['privateProperty']);
        $this->assertFalse($output, 'Property should not be accessible');

        $output = $reflectionMethod->invokeArgs($object, ['protectedProperty']);
        $this->assertTrue($output, 'Property should be accessible');

        $output = $reflectionMethod->invokeArgs($object, ['publicProperty']);
        $this->assertTrue($output, 'Property should be accessible');

        $output = $reflectionMethod->invokeArgs($object, ['missingProperty']);
        $this->assertFalse($output, 'Property should not be accessible');

        $output = $reflectionMethod->invokeArgs(new ViewableData(), ['objCache']);
        $this->assertTrue($output, 'Property should be accessible');
    }

    public function testDynamicData()
    {
        $obj = (object) ['SomeField' => [1, 2, 3]];
        $viewableData = new ViewableData();
        $this->assertFalse($viewableData->hasDynamicData('abc'));
        $viewableData->setDynamicData('abc', $obj);
        $this->assertTrue($viewableData->hasDynamicData('abc'));
        $this->assertSame($obj, $viewableData->getDynamicData('abc'));
        $this->assertSame($obj, $viewableData->abc);
    }

    public function provideWrapArrayInObj(): array
    {
        return [
            'empty array' => [
                'arr' => [],
                'expectedClass' => ArrayList::class,
            ],
            'fully indexed array' => [
                'arr' => [
                    'value1',
                    'value2',
                ],
                'expectedClass' => ArrayList::class,
            ],
        ];
    }

    /**
     * @dataProvider provideWrapArrayInObj
     */
    public function testWrapArrayInObj(array $arr, string $expectedClass): void
    {
        $viewableData = new ViewableData();
        $viewableData->arr = $arr;
        $this->assertInstanceOf($expectedClass, $viewableData->obj('arr'));
    }
}