function ModerationInformationTest::testGetWorkflowForEntity

@dataProvider providerWorkflow @covers ::getWorkflowForEntity

File

core/modules/content_moderation/tests/src/Unit/ModerationInformationTest.php, line 109

Class

ModerationInformationTest

@coversDefaultClass \Drupal\content_moderation\ModerationInformation @group content_moderation

Namespace

Code

public function testGetWorkflowForEntity($workflow) {
    $entity_type_manager = $this->prophesize(EntityTypeManagerInterface::class);
    if ($workflow) {
        $workflow_entity = $this->prophesize(WorkflowInterface::class)
            ->reveal();
        $workflow_storage = $this->prophesize(EntityStorageInterface::class);
        $workflow_storage->load('workflow')
            ->willReturn($workflow_entity)
            ->shouldBeCalled();
        $entity_type_manager->getStorage('workflow')
            ->willReturn($workflow_storage->reveal());
    }
    else {
        $workflow_entity = NULL;
    }
    $moderation_information = new ModerationInformation($entity_type_manager->reveal(), $this->setupModerationBundleInfo('test_bundle', $workflow));
    $entity = $this->prophesize(ContentEntityInterface::class);
    $entity->getEntityTypeId()
        ->willReturn('test_entity_type');
    $entity->bundle()
        ->willReturn('test_bundle');
    $this->assertEquals($workflow_entity, $moderation_information->getWorkflowForEntity($entity->reveal()));
}