def save(self, doc):
        if doc.id:
            return self.update(doc)
        else:
            return self.insert(doc)
def wrapper%(signature)s:
    with ldap3mock:
        return func%(funcargs)s