protected function AcquiadamMockedMediaEntityTrait::getMockedMediaEntity in Media: Acquia DAM 8

Mocks a Media entity to add shared functionality for tests.

Parameters

int $assetId: The asset ID to assign to the entity.

string $sourceField: The source field name to use. Allows overriding the success path.

int $mediaEntityId: The ID to assign to the media entity.

Return value

\Drupal\media\MediaInterface|\PHPUnit\Framework\MockObject\MockObject The mocked MediaInterface entity.

File

tests/src/Traits/AcquiadamMockedMediaEntityTrait.php, line 29

Class

AcquiadamMockedMediaEntityTrait

Shared complex mocked media and related entities.

Namespace

Drupal\Tests\media_acquiadam\Traits

Code

protected function getMockedMediaEntity($assetId, $sourceField = NULL, $mediaEntityId = 47247625) {
  $sourceField = $sourceField ?? 'phpunit_asset_id_field';
  $source_field_definition = $this
    ->getMockBuilder(FieldDefinitionInterface::class)
    ->disableOriginalConstructor()
    ->getMock();
  $source_field_definition
    ->method('getName')
    ->willReturn($sourceField);
  $media_source = $this
    ->getMockBuilder(MediaSourceInterface::class)
    ->disableOriginalConstructor()
    ->getMock();
  $media_source
    ->method('getSourceFieldDefinition')
    ->willReturn($source_field_definition);
  $media_bundle = $this
    ->getMockBuilder(MediaTypeInterface::class)
    ->disableOriginalConstructor()
    ->getMock();
  $bundle_definition = $this
    ->getMockBuilder(EntityReferenceFieldItemListInterface::class)
    ->disableOriginalConstructor()
    ->getMock();
  $bundle_definition->entity = $media_bundle;
  $media = $this
    ->getMockBuilder(MediaInterface::class)
    ->disableOriginalConstructor()
    ->getMock();
  $media
    ->method('getSource')
    ->willReturn($media_source);
  $media
    ->method('uuid')
    ->willReturn('e63ce44d-4cfe-44d4-af7d-0692821d52cc');
  $media
    ->method('hasField')
    ->willReturnMap([
    [
      'phpunit_asset_id_field',
      TRUE,
    ],
    [
      'phpunit_file_field',
      TRUE,
    ],
    [
      'phpunit_test_fail',
      FALSE,
    ],
  ]);
  $media
    ->method('getEntityTypeId')
    ->willReturn('media');
  $media
    ->method('bundle')
    ->willReturn('acquiadam');
  $media
    ->method('id')
    ->willReturn($mediaEntityId);
  $media
    ->method('get')
    ->with('bundle')
    ->willReturn($bundle_definition);
  $file_field = $this
    ->getMockBuilder(\stdClass::class)
    ->disableOriginalConstructor()
    ->setMethods([
    'first',
    'mainPropertyName',
  ])
    ->getMock();
  $file_field
    ->method('first')
    ->willReturnSelf();
  $file_field
    ->method('mainPropertyName')
    ->willReturn('target_id');
  $media->phpunit_file_field = $file_field;
  $media->phpunit_file_field->target_id = $this
    ->getMockedFileEntity()
    ->id();
  $asset_id_field = $this
    ->getMockBuilder(\stdClass::class)
    ->disableOriginalConstructor()
    ->setMethods([
    'first',
    'mainPropertyName',
  ])
    ->getMock();
  $asset_id_field
    ->method('first')
    ->willReturnSelf();
  $asset_id_field
    ->method('mainPropertyName')
    ->willReturn('value');
  $media->phpunit_asset_id_field = $asset_id_field;
  $media->phpunit_asset_id_field->value = $assetId;
  return $media;
}