/**
* @expectedException \Magento\Framework\Model\Exception
* @expectedExceptionMessage Please define flag code.
*/
public function testBeforeSaveException()
{
$this->createInstance();
$this->flag->setData('block', 'blockNmae');
$this->flag->beforeSave();
}