public function testBeforeSave() { $object = $this->getMockBuilder('Magento\\Framework\\Object')->disableOriginalConstructor()->setMethods(array('getDefaultShipping', 'unsetDefaultShipping'))->getMock(); $object->expects($this->once())->method('getDefaultShipping')->will($this->returnValue(null)); $object->expects($this->once())->method('unsetDefaultShipping')->will($this->returnSelf()); /** @var \Magento\Framework\Object $object */ $this->testable->beforeSave($object); }