diff --git a/core/tests/Drupal/Tests/Core/Entity/ContentEntityDatabaseStorageTest.php b/core/tests/Drupal/Tests/Core/Entity/ContentEntityDatabaseStorageTest.php index e107453..a0a7671 100644 --- a/core/tests/Drupal/Tests/Core/Entity/ContentEntityDatabaseStorageTest.php +++ b/core/tests/Drupal/Tests/Core/Entity/ContentEntityDatabaseStorageTest.php @@ -1106,12 +1106,8 @@ public function testLoadMultiplePersistentCached() { $key = 'values:' . $this->entityTypeId . ':1'; $id = 1; - $class_name = '\Drupal\Tests\Core\Entity\ContentEntityDatabaseStorageTestEntityInterface'; - $methods = get_class_methods($class_name); - unset($methods[array_search('postLoad', $methods)]); - $entity = $this->getMockBuilder($class_name) - ->setMethods($methods) - ->getMock(); + $entity = $this->getMockBuilder('\Drupal\Tests\Core\Entity\ContentEntityDatabaseStorageTestEntityInterface') + ->getMockForAbstractClass(); $entity->expects($this->any()) ->method('id') ->will($this->returnValue($id)); @@ -1150,12 +1146,8 @@ public function testLoadMultipleNoPersistentCache() { $this->setUpModuleHandlerNoImplementations(); $id = 1; - $class_name = '\Drupal\Tests\Core\Entity\ContentEntityDatabaseStorageTestEntityInterface'; - $methods = get_class_methods($class_name); - unset($methods[array_search('postLoad', $methods)]); - $entity = $this->getMockBuilder($class_name) - ->setMethods($methods) - ->getMock(); + $entity = $this->getMockBuilder('\Drupal\Tests\Core\Entity\ContentEntityDatabaseStorageTestEntityInterface') + ->getMockForAbstractClass(); $entity->expects($this->any()) ->method('id') ->will($this->returnValue($id)); @@ -1201,12 +1193,8 @@ public function testLoadMultiplePersistentCacheMiss() { $this->setUpModuleHandlerNoImplementations(); $id = 1; - $class_name = '\Drupal\Tests\Core\Entity\ContentEntityDatabaseStorageTestEntityInterface'; - $methods = get_class_methods($class_name); - unset($methods[array_search('postLoad', $methods)]); - $entity = $this->getMockBuilder($class_name) - ->setMethods($methods) - ->getMock(); + $entity = $this->getMockBuilder('\Drupal\Tests\Core\Entity\ContentEntityDatabaseStorageTestEntityInterface') + ->getMockForAbstractClass(); $entity->expects($this->any()) ->method('id') ->will($this->returnValue($id));