diff --git a/core/tests/Drupal/Tests/Core/Entity/EntityTypeTest.php b/core/tests/Drupal/Tests/Core/Entity/EntityTypeTest.php index c350b48..738c233 100644 --- a/core/tests/Drupal/Tests/Core/Entity/EntityTypeTest.php +++ b/core/tests/Drupal/Tests/Core/Entity/EntityTypeTest.php @@ -32,6 +32,28 @@ protected function setUpEntityType($definition) { } /** + * @covers ::get + * + * @dataProvider providerTestGet + */ + public function testGet(array $defintion, $key, $expected) { + $entity_type = $this->setUpEntityType($defintion); + $this->assertSame($expected, $entity_type->get($key)); + } + + /** + * @covers ::set + * @covers ::get + * + * @dataProvider providerTestSet + */ + public function testSet($key, $value) { + $entity_type = $this->setUpEntityType([]); + $this->assertInstanceOf('Drupal\Core\Entity\EntityTypeInterface', $entity_type->set($key, $value)); + $this->assertSame($value, $entity_type->get($key)); + } + + /** * Tests the getKeys() method. * * @dataProvider providerTestGetKeys @@ -65,6 +87,34 @@ public function testHasKey($entity_keys, $expected) { } /** + * Provides test data for testGet. + */ + public function providerTestGet() { + return [ + [[], 'provider', NULL], + [['provider' => ''], 'provider', ''], + [['provider' => 'test'], 'provider', 'test'], + [[], 'something_additional', NULL], + [['something_additional' => ''], 'something_additional', ''], + [['something_additional' => 'additional'], 'something_additional', 'additional'], + ]; + } + + /** + * Provides test data for testSet. + */ + public function providerTestSet() { + return [ + ['provider', NULL], + ['provider', ''], + ['provider', 'test'], + ['something_additional', NULL], + ['something_additional', ''], + ['something_additional', 'additional'], + ]; + } + + /** * Provides test data. */ public function providerTestGetKeys() {