/**
* Updates an object state
*
* @param mixed $stateId
* @param \eZ\Publish\SPI\Persistence\Content\ObjectState\InputStruct $input
*
* @return \eZ\Publish\SPI\Persistence\Content\ObjectState
*/
public function update($stateId, InputStruct $input)
{
$objectState = $this->objectStateMapper->createObjectStateFromInputStruct($input);
$objectState->id = (int) $stateId;
$this->objectStateGateway->updateObjectState($objectState);
return $this->load($objectState->id);
}