def test_Archive(db):
"""Test the Archive model class."""
assert Archive.query.count() == 0
# we create an SIP, it will automatically create an Archive via signals
user = create_test_user('*****@*****.**')
sip = SIP.create(True, user_id=user.id, agent={'test': 'test'})
db.session.commit()
assert Archive.query.count() == 1
ark = Archive.get_from_sip(sip.id)
assert ark.sip.user.id == sip.user.id
assert ark.status == ArchiveStatus.NEW
assert ark.accession_id is None
assert ark.archivematica_id is None
# let's change the object
ark.status = ArchiveStatus.REGISTERED
ark.accession_id = '08'
ark.archivematica_id = sip.id
db.session.commit()
ark = Archive.get_from_accession_id('08')
assert Archive.query.count() == 1
assert ark.status == ArchiveStatus.REGISTERED
assert ark.archivematica_id == sip.id
# we try to get a non existing record
assert Archive.get_from_sip(uuid.uuid4()) is None
def test(accession_id):
"""Show a test page."""
ark = Archive.get_from_accession_id(accession_id)
return """<DOCTYPE html><html><head></head><body>
<h1>{}</h1>
</body></html>""".format(create_accession_id(ark))
def decorate(*args, **kwargs):
accession_id = kwargs.pop('accession_id')
archive = Archive_.get_from_accession_id(accession_id)
if not archive:
abort(404, 'Accession_id {} not found.'.format(accession_id))
return f(archive=archive, *args, **kwargs)