function MediaRevisionTest::assertRevisionCount

Asserts that an entity has a certain number of revisions.

Parameters

\Drupal\Core\Entity\EntityInterface $entity: The entity in question.

int $expected_revisions: The expected number of revisions.

File

core/modules/media/tests/src/Functional/MediaRevisionTest.php, line 194

Class

MediaRevisionTest

Tests the revisionability of media entities.

Namespace

Code

protected function assertRevisionCount(EntityInterface $entity, $expected_revisions) {
    $entity_type = $entity->getEntityType();
    $count = $this->container
        ->get('entity_type.manager')
        ->getStorage($entity_type->id())
        ->getQuery()
        ->count()
        ->allRevisions()
        ->condition($entity_type->getKey('id'), $entity->id())
        ->execute();
    $this->assertSame($expected_revisions, (int) $count);
}