def test_get_entry_type(self):
from invenio.modules.records.models import Record
from invenio.modules.records.api import get_record
records = []
record_list = Record.query.all()
for rec in record_list:
rec = get_record(rec.id)
if rec:
records.append(rec)
for r in records:
b = Bibtex(r)
self.assertEqual(r['upload_type']['type'], b._get_entry_type())
self.assertEqual(test_record['upload_type']['type'],
self.record_good._get_entry_type())
self.assertEqual('default', self.record_bad._get_entry_type())
def bibtex_filter(record):
return Bibtex(record).format()
def setUp(self):
from invenio.modules.records.api import Record
self.record_good = Bibtex(Record.create(test_record, 'json'))
self.record_bad = Bibtex(Record.create(test_bad_record, 'json'))
self.record_empty = Bibtex({})
