def setUp(self):
"""Test."""
from invenio_deposit.models import DepositionType
from invenio_deposit.registry import deposit_types, \
deposit_default_type
# Unregister any default types
try:
deposit_default_type.unregister()
except RegistryError:
pass
# Create some test types.
class DefaultType(DepositionType):
pass
class AnotherType(DepositionType):
pass
# Register types
self.DefaultType = DefaultType
self.AnotherType = AnotherType
deposit_types.register(DefaultType)
deposit_types.register(AnotherType)
deposit_default_type.register(DefaultType)
def setUp(self):
"""Setup tests."""
from invenio_deposit.models import Deposition, DepositionType
from invenio_deposit.registry import deposit_types, deposit_default_type
from invenio_deposit.form import WebDepositForm
from invenio_deposit.tasks import prefill_draft, prepare_sip
celery.conf["CELERY_ALWAYS_EAGER"] = True
def agnostic_task(obj, eng):
data_model = eng.workflow_definition.model(obj)
sip = data_model.get_latest_sip()
print(sip.metadata)
class DefaultType(DepositionType):
pass
class SimpleRecordTestForm(WebDepositForm):
pass
class DepositModelTest(DepositionType):
"""A test workflow for the model."""
model = Deposition
draft_definitions = {"default": SimpleRecordTestForm}
workflow = [prefill_draft(draft_id="default"), prepare_sip(), agnostic_task]
deposit_types.register(DefaultType)
deposit_types.register(DepositModelTest)
deposit_default_type.register(DefaultType)
def setUp(self):
"""Test."""
from invenio_deposit.models import DepositionType
from invenio_deposit.registry import deposit_types, \
deposit_default_type
# Unregister any default types
try:
deposit_default_type.unregister()
except RegistryError:
pass
# Create some test types.
class DefaultType(DepositionType):
pass
class AnotherType(DepositionType):
pass
# Register types
self.DefaultType = DefaultType
self.AnotherType = AnotherType
deposit_types.register(DefaultType)
deposit_types.register(AnotherType)
deposit_default_type.register(DefaultType)
def register(self, deposition_type):
""" Register the deposition type """
from invenio_deposit.registry import deposit_types
from invenio_deposit.url_converters import refresh_url_map
deposit_types.register(deposition_type)
assert deposition_type in deposit_types
self.deposition_type = deposition_type
refresh_url_map(self.app)
def register(self, deposition_type):
""" Register the deposition type """
from invenio_deposit.registry import deposit_types
from invenio_deposit.url_converters import refresh_url_map
deposit_types.register(deposition_type)
assert deposition_type in deposit_types
self.deposition_type = deposition_type
refresh_url_map(self.app)