public function testInsertWithModelCallsInsertOnParentAndReturnsModel() { $data = array('product_id' => 1, 'uom_code' => 'EA', 'quantity' => 1); $productUom = new \SpeckCatalog\Model\ProductUom(); $productUom->setProductId(1)->setUomCode('EA')->setQuantity(1); $mockMapper = $this->getMock('\\SpeckCatalog\\Mapper\\ProductUom'); $mockMapper->expects($this->once())->method('insert')->with($productUom); $mockMapper->expects($this->once())->method('find')->with($data)->will($this->returnValue($productUom)); $service = $this->getService(); $service->setEntityMapper($mockMapper); $return = $service->insert($productUom); $this->assertInstanceOf('\\SpeckCatalog\\Model\\ProductUom', $return); }