protected function _mockFlash() { $this->Trait->Flash = $this->getMockBuilder('Cake\\Controller\\Component\\FlashComponent')->setMethods(['error', 'success'])->disableOriginalConstructor()->getMock(); }