def setUp(self): """Test.""" from invenio.modules.deposit.models import DepositionType from invenio.modules.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.modules.deposit.registry import deposit_types from invenio.modules.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)