cubicweb/server/test/data/entities.py
author Sylvain Thénault <sylvain.thenault@logilab.fr>
Fri, 10 Mar 2017 16:55:57 +0100
changeset 12034 e7e469d0539b
parent 11355 47b0b08fbb4b
permissions -rw-r--r--
[cleanup] Drop references to the old internal_session method dropped a while ago, no need to talk about it in a docstring nor to fake it.

from cubicweb.server.sources import datafeed


class SourceParserSuccess(datafeed.DataFeedParser):
    __regid__ = 'test_source_parser_success'

    def process(self, url, raise_on_error=False):
        entity = self._cw.create_entity('Card', title=u'success')
        self.notify_updated(entity)


class SourceParserFail(SourceParserSuccess):
    __regid__ = 'test_source_parser_fail'

    def process(self, url, raise_on_error=False):
        entity = self._cw.create_entity('Card', title=u'fail')
        self.notify_updated(entity)
        raise RuntimeError("fail")