/** * @param EntityPackage $entity_package * @return PackageContent */ public function getPackageContent(EntityPackage $entity_package) { if ($this->type == PackageContent::ENTITY) { return $entity_package->getEntityContent(); } else { return $entity_package->getRepositoryContent(); } }
public function testGetEntityContent() { $entity_content = self::createMock('Hostnet\\Component\\EntityPlugin\\PackageContentInterface'); $entity_package = new EntityPackage(new Package('hostnet/foo', 1.0, 1.0), $entity_content, self::createMock('Hostnet\\Component\\EntityPlugin\\PackageContentInterface')); $this->assertSame($entity_content, $entity_package->getEntityContent()); }