de2a50eeb2
First this brings a change to the paper API: the terminology of bibdata
and bibentry is clarified: now bibentry is a dictionary of the form
{citekey: bibdata} where bibdata corresponds to the actual dictionary of
bibliographic fields and values {author: ..., year: ...}.
Now bibentry is an attribute of the paper object that is generated from
citekey and bibdata.
This commit fixes all tests in particular an issue with citekey in
bibentry not updated.
Also removes prints in tests and deprecated assertEquals.
Usecase tests now fail if the command ask for unexpected inputs.
Removes queries for user input in attach and add commands (for deletion
of a copied document file). The input was not coherent with tests and is
annoying.
78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
import unittest
|
|
import os
|
|
|
|
import dotdot
|
|
import fake_env
|
|
|
|
from pubs import content, filebroker, databroker, datacache, configs
|
|
|
|
import str_fixtures
|
|
from pubs import endecoder
|
|
|
|
|
|
class TestDataBroker(unittest.TestCase):
|
|
|
|
def test_databroker(self):
|
|
|
|
ende = endecoder.EnDecoder()
|
|
page99_metadata = ende.decode_metadata(str_fixtures.metadata_raw0)
|
|
page99_bibentry = ende.decode_bibdata(str_fixtures.bibtex_raw0)
|
|
|
|
for db_class in [databroker.DataBroker, datacache.DataCache]:
|
|
self.fs = fake_env.create_fake_fs([content, filebroker, configs])
|
|
|
|
db = db_class('tmp', create=True)
|
|
|
|
db.push_metadata('citekey1', page99_metadata)
|
|
self.assertFalse(db.exists('citekey1', meta_check=True))
|
|
self.assertFalse(db.exists('citekey1', meta_check=False))
|
|
|
|
db.push_bibentry('citekey1', page99_bibentry)
|
|
self.assertTrue(db.exists('citekey1', meta_check=False))
|
|
self.assertTrue(db.exists('citekey1', meta_check=True))
|
|
|
|
self.assertEqual(db.pull_metadata('citekey1'), page99_metadata)
|
|
pulled = db.pull_bibentry('citekey1')['Page99']
|
|
for key, value in pulled.items():
|
|
self.assertEqual(pulled[key], page99_bibentry['Page99'][key])
|
|
self.assertEqual(db.pull_bibentry('citekey1'), page99_bibentry)
|
|
|
|
fake_env.unset_fake_fs([content, filebroker])
|
|
|
|
def test_existing_data(self):
|
|
|
|
ende = endecoder.EnDecoder()
|
|
page99_bibentry = ende.decode_bibdata(str_fixtures.bibtex_raw0)
|
|
|
|
for db_class in [databroker.DataBroker, datacache.DataCache]:
|
|
self.fs = fake_env.create_fake_fs([content, filebroker])
|
|
fake_env.copy_dir(self.fs, os.path.join(os.path.dirname(__file__), 'testrepo'), 'repo')
|
|
|
|
db = db_class('repo', create=False)
|
|
|
|
self.assertEqual(db.pull_bibentry('Page99'), page99_bibentry)
|
|
|
|
for citekey in ['10.1371_journal.pone.0038236',
|
|
'10.1371journal.pone.0063400',
|
|
'journal0063400']:
|
|
db.pull_bibentry(citekey)
|
|
db.pull_metadata(citekey)
|
|
|
|
with self.assertRaises(IOError):
|
|
db.pull_bibentry('citekey')
|
|
with self.assertRaises(IOError):
|
|
db.pull_metadata('citekey')
|
|
|
|
db.add_doc('Larry99', 'docsdir://Page99.pdf')
|
|
self.assertTrue(content.check_file('repo/doc/Page99.pdf', fail=False))
|
|
self.assertTrue(content.check_file('repo/doc/Larry99.pdf', fail=False))
|
|
|
|
db.remove_doc('docsdir://Page99.pdf')
|
|
|
|
fake_env.unset_fake_fs([content, filebroker])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|