def test_empty_func(self):
def noop(doc):
pass
handler = bahf.FunctionHandler(noop)
doc = Document()
handler.modify_document(doc)
if handler.failed:
raise RuntimeError(handler.error)
assert not doc.roots
def test_safe_to_fork(self):
def noop(doc):
pass
handler = bahf.FunctionHandler(noop)
doc = Document()
assert handler.safe_to_fork
handler.modify_document(doc)
if handler.failed:
raise RuntimeError(handler.error)
assert not handler.safe_to_fork
def test_func_adds_roots(self):
def add_roots(doc):
doc.add_root(AnotherModelInTestFunction())
doc.add_root(SomeModelInTestFunction())
handler = bahf.FunctionHandler(add_roots)
doc = Document()
handler.modify_document(doc)
if handler.failed:
raise RuntimeError(handler.error)
assert len(doc.roots) == 2
