InternalTypedDataTestTrait.php

<?php

declare (strict_types=1);
namespace Drupal\Tests\serialization\Unit\Normalizer;

use Drupal\Core\TypedData\DataDefinitionInterface;
use Drupal\Core\TypedData\TypedDataInterface;

/**
 * Trait that provides mocked typed data objects.
 */
trait InternalTypedDataTestTrait {
    
    /**
     * Gets a typed data property.
     *
     * @param bool $internal
     *   Whether the typed data property is internal.
     *
     * @return \Drupal\Core\TypedData\TypedDataInterface
     *   The typed data property.
     */
    protected function getTypedDataProperty($internal = TRUE) {
        $definition = $this->prophesize(DataDefinitionInterface::class);
        $definition->isInternal()
            ->willReturn($internal)
            ->shouldBeCalled();
        $definition = $definition->reveal();
        $property = $this->prophesize(TypedDataInterface::class);
        $property->getDataDefinition()
            ->willReturn($definition)
            ->shouldBeCalled();
        return $property->reveal();
    }

}