public function testAddSuccess()
{
$messageText = 'Some message success text';
$message = $this->getMock('Magento\\Framework\\Message\\MessageInterface');
$this->messageFactory->expects($this->once())->method('create')->with(MessageInterface::TYPE_SUCCESS, $messageText)->will($this->returnValue($message));
$collection = $this->initMessageCollection();
$collection->expects($this->once())->method('addMessage')->with($message);
$this->assertSame($this->messages, $this->messages->addSuccess($messageText));
}
/**
* @param string $type
* @param string $methodName
* @dataProvider addMessageDataProvider
*/
public function testAddMessage($type, $methodName)
{
$this->assertFalse($this->model->hasMessages());
$message = 'Message';
$messageCollection = $this->getMock('Magento\\Framework\\Message\\Collection', ['addMessage'], [], '', false);
$this->session->expects($this->any())->method('getData')->will($this->returnValue($messageCollection));
$this->eventManager->expects($this->once())->method('dispatch')->with('session_abstract_add_message');
$this->messageFactory->expects($this->once())->method('create')->with($type, $message)->will($this->returnValue($this->messageMock));
$this->model->{$methodName}($message, 'group');
$this->assertTrue($this->model->hasMessages());
}