def test_persist_tree_after_tasks_inserted(self, mock_get_current_context): """ Make sure a marker tree isn't(Raises exception) persisted after the current context's tasks have been inserted """ from furious.context.context import Context from furious.marker_tree.identity_utils import leaf_persistence_id_from_group_id from furious.marker_tree.marker import Marker from furious.marker_tree.exceptions import NotSafeToSave a_context = Context(id="zebra") a_context._tasks_inserted = True mock_get_current_context.return_value = a_context root_marker = Marker(id="zebra") for x in xrange(3): root_marker.children.append(Marker(id=leaf_persistence_id_from_group_id(root_marker.id, x))) self.assertRaises(NotSafeToSave, root_marker.persist)