class EntityContentBaseTest

class EntityContentBaseTest extends EntityTestBase {
    
    /**
     * Test basic entity save.
     *
     * @covers ::import
     */
    public function testImport() {
        $bundles = [];
        $destination = new EntityTestDestination([], '', [], $this->migration
            ->reveal(), $this->storage
            ->reveal(), $bundles, $this->entityFieldManager
            ->reveal(), $this->prophesize(FieldTypePluginManagerInterface::class)
            ->reveal());
        $entity = $this->prophesize(ContentEntityInterface::class);
        $entity->isValidationRequired()
            ->shouldBeCalledTimes(1);
        // Assert that save is called.
        $entity->save()
            ->shouldBeCalledTimes(1);
        // Set an id for the entity
        $entity->id()
            ->willReturn(5);
        $destination->setEntity($entity->reveal());
        // Ensure the id is saved entity id is returned from import.
        $this->assertEquals([
            5,
        ], $destination->import(new Row()));
        // Assert that import set the rollback action.
        $this->assertEquals(MigrateIdMapInterface::ROLLBACK_DELETE, $destination->rollbackAction());
    }
    
    /**
     * Test row skipping when we can't get an entity to save.
     *
     * @covers ::import
     */
    public function testImportEntityLoadFailure() {
        $bundles = [];
        $destination = new EntityTestDestination([], '', [], $this->migration
            ->reveal(), $this->storage
            ->reveal(), $bundles, $this->entityFieldManager
            ->reveal(), $this->prophesize(FieldTypePluginManagerInterface::class)
            ->reveal());
        $destination->setEntity(FALSE);
        $this->expectException(MigrateException::class);
        $this->expectExceptionMessage('Unable to get entity');
        $destination->import(new Row());
    }
    
    /**
     * Test that translation destination fails for untranslatable entities.
     */
    public function testUntranslatable() {
        // An entity type without a language.
        $this->entityType
            ->getKey('langcode')
            ->willReturn('');
        $this->entityType
            ->getKey('id')
            ->willReturn('id');
        $this->entityFieldManager
            ->getBaseFieldDefinitions('foo')
            ->willReturn([
            'id' => BaseFieldDefinitionTest::create('integer'),
        ]);
        $destination = new EntityTestDestination([
            'translations' => TRUE,
        ], '', [], $this->migration
            ->reveal(), $this->storage
            ->reveal(), [], $this->entityFieldManager
            ->reveal(), $this->prophesize(FieldTypePluginManagerInterface::class)
            ->reveal());
        $this->expectException(MigrateException::class);
        $this->expectExceptionMessage('The "foo" entity type does not support translations.');
        $destination->getIds();
    }

}