/** * @param object $entity * @param string $scope */ public function showInfoMessage($entity, $scope) { $message = $this->messageManager->getInfoMessage($entity, $scope); if ($message) { $this->messageManager->addFlashMessage('info', $message); } }
/** * @param string|null $expected * @param string $scope * @param bool $shown * @param int|null $viewCount * @param int|null $editCount * @dataProvider getInfoMessageDataProvider */ public function testGetInfoMessage($expected, $scope, $shown, $viewCount = null, $editCount = null) { $entity = new \stdClass(); $entityName = get_class($entity); $this->storage->expects($this->once())->method('isInfoMessageShown')->with($entityName, $scope)->will($this->returnValue($shown)); $this->navigation->expects($this->any())->method('getTotalCount')->with($entity, $this->isType('string'))->will($this->returnValueMap([[$entity, EntityPaginationManager::VIEW_SCOPE, $viewCount], [$entity, EntityPaginationManager::EDIT_SCOPE, $editCount]])); if ($expected) { $this->storage->expects($this->once())->method('setInfoMessageShown')->with($entityName, $scope); } else { $this->storage->expects($this->never())->method('setInfoMessageShown'); } $this->assertSame($expected, $this->manager->getInfoMessage($entity, $scope)); }