class EntityRepositoryTest

class EntityRepositoryTest extends UnitTestCase {
  
  /**
   * The entity type manager.
   *
   * @var \Drupal\Core\Entity\EntityTypeManagerInterface|\Prophecy\Prophecy\ProphecyInterface
   */
  protected $entityTypeManager;
  
  /**
   * The language manager.
   *
   * @var \Drupal\Core\Language\LanguageManagerInterface|\Prophecy\Prophecy\ProphecyInterface
   */
  protected $languageManager;
  
  /**
   * The context repository.
   *
   * @var \Drupal\Core\Plugin\Context\ContextRepositoryInterface|\Prophecy\Prophecy\ProphecyInterface
   */
  protected $contextRepository;
  
  /**
   * The entity repository under test.
   *
   * @var \Drupal\Core\Entity\EntityRepository
   */
  protected $entityRepository;
  
  /**
   * {@inheritdoc}
   */
  protected function setUp() : void {
    parent::setUp();
    $this->entityTypeManager = $this->prophesize(EntityTypeManagerInterface::class);
    $this->languageManager = $this->prophesize(LanguageManagerInterface::class);
    $this->contextRepository = $this->prophesize(ContextRepositoryInterface::class);
    $this->entityRepository = new EntityRepository($this->entityTypeManager
      ->reveal(), $this->languageManager
      ->reveal(), $this->contextRepository
      ->reveal());
  }
  
  /**
   * Tests the getTranslationFromContext() method.
   *
   * @covers ::getTranslationFromContext
   */
  public function testGetTranslationFromContext() : void {
    $language = new Language([
      'id' => 'en',
    ]);
    $this->languageManager
      ->getCurrentLanguage(LanguageInterface::TYPE_CONTENT)
      ->willReturn($language)
      ->shouldBeCalledTimes(1);
    $this->languageManager
      ->getFallbackCandidates(Argument::type('array'))
      ->will(function ($args) {
      $context = $args[0];
      $candidates = [];
      if (!empty($context['langcode'])) {
        $candidates[$context['langcode']] = $context['langcode'];
      }
      return $candidates;
    })
      ->shouldBeCalledTimes(1);
    $translated_entity = $this->prophesize(ContentEntityInterface::class);
    $entity = $this->prophesize(ContentEntityInterface::class);
    $entity->getUntranslated()
      ->willReturn($entity);
    $entity->language()
      ->willReturn($language);
    $entity->hasTranslation(LanguageInterface::LANGCODE_DEFAULT)
      ->willReturn(FALSE);
    $entity->hasTranslation('custom_langcode')
      ->willReturn(TRUE);
    $entity->getTranslation('custom_langcode')
      ->willReturn($translated_entity->reveal());
    $entity->getTranslationLanguages()
      ->willReturn([
      new Language([
        'id' => 'en',
      ]),
      new Language([
        'id' => 'custom_langcode',
      ]),
    ]);
    $entity->addCacheContexts([
      'languages:language_content',
    ])
      ->shouldBeCalled();
    $this->assertSame($entity->reveal(), $this->entityRepository
      ->getTranslationFromContext($entity->reveal()));
    $this->assertSame($translated_entity->reveal(), $this->entityRepository
      ->getTranslationFromContext($entity->reveal(), 'custom_langcode'));
  }

}