def test_merge_error(capsys, mock_init):
# Simulate user input:
sys.argv = f"bibm merge fake_file.bib".split()
cli.main()
captured = capsys.readouterr()
assert captured.out \
== "\nError: Input BibTeX file 'fake_file.bib' does not exist.\n"
def test_cli_fetch_invalid_name(capsys, mock_init_sample, reqs):
sys.argv = "bibm fetch 1957RvMP...29..547B pdf_file".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == (
"Fetching PDF file from Journal website:\n"
"\nError: Invalid filename, must have a .pdf extension\n")
def test_cli_open_keycode_invalid(capsys, mock_init_sample, mock_call):
pathlib.Path(f"{u.BM_PDF()}Slipher1913.pdf").touch()
sys.argv = f"bibm open bad_keycode".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == ('\nError: Input is no key, bibcode, or PDF '
'of any entry in Bibmanager database\n')
def test_cli_search_year_invalid(capsys, mock_init_sample,
mock_prompt_session):
sys.argv = "bibm search".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == ("(Press 'tab' for autocomplete)\n\n"
"\nInvalid format for input year: 1984a\n")
def test_cli_pdf_set_pathed_filename(capsys, mock_init_sample):
sys.argv = "bibm pdf 1957RvMP...29..547B file.pdf ./new.pdf".split()
pathlib.Path("file.pdf").touch()
cli.main()
captured = capsys.readouterr()
assert captured.out == "\nError: filename must not have a path\n"
bib = bm.find(bibcode='1957RvMP...29..547B')
assert bib.pdf is None
def test_version(capsys):
sys.argv = "bibm --version".split()
try:
cli.main()
except SystemExit:
pass
captured = capsys.readouterr()
assert captured.out == f"bibmanager version {bibmanager.__version__}\n"
def test_help(capsys):
sys.argv = "bibm -h".split()
try:
cli.main()
except:
pass
captured = capsys.readouterr()
assert captured.out == main_description
def test_ads_search(capsys, reqs, mock_prompt_session):
cm.set('ads_display', '2')
am.search.__defaults__ = 0, 2, 'pubdate+desc'
sys.argv = "bibm ads-search".split()
captured = capsys.readouterr()
cli.main()
captured = capsys.readouterr()
assert captured.out == f"""(Press 'tab' for autocomplete)\n
def test_cli_pdf_set_missing_pdf(capsys, mock_init_sample):
sys.argv = f"bibm pdf 1957RvMP...29..547B".split()
pathlib.Path(f"file.pdf").touch()
cli.main()
captured = capsys.readouterr()
assert captured.out == "\nError: Path to PDF file is missing.\n"
bib = bm.find(bibcode='1957RvMP...29..547B')
assert bib.pdf is None
def test_merge_default(capsys, mock_init):
bibfile = u.HOME + "examples/sample.bib"
# Simulate user input:
sys.argv = f"bibm merge {bibfile}".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == "\nMerged 17 new entries.\n\n" \
f"Merged BibTeX file '{bibfile}' into bibmanager database.\n"
def test_cli_pdf_set_renamed(capsys, mock_init_sample, mock_call):
sys.argv = f"bibm pdf 1957RvMP...29..547B file.pdf new.pdf".split()
pathlib.Path(f"file.pdf").touch()
cli.main()
captured = capsys.readouterr()
assert captured.out == f"Saved PDF to: '{u.BM_PDF()}new.pdf'.\n"
bib = bm.find(bibcode='1957RvMP...29..547B')
assert bib.pdf == 'new.pdf'
assert 'new.pdf' in os.listdir(u.BM_PDF())
def test_cli_pdf_set_invalid_name(capsys, mock_init_sample):
sys.argv = "bibm pdf 1957RvMP...29..547B file.pdf pdf_file".split()
pathlib.Path("file.pdf").touch()
cli.main()
captured = capsys.readouterr()
assert captured.out == \
"\nError: Invalid filename, must have a .pdf extension\n"
bib = bm.find(bibcode='1957RvMP...29..547B')
assert bib.pdf is None
def test_cli_pdf_set_missing_pdf_file(capsys, mock_init_sample):
sys.argv = "bibm pdf 1957RvMP...29..547B file.pdf".split()
with u.ignored(OSError):
os.remove('file.pdf')
cli.main()
captured = capsys.readouterr()
assert captured.out == "\nError: input PDF file does not exist.\n"
bib = bm.find(bibcode='1957RvMP...29..547B')
assert bib.pdf is None
def test_cli_pdf_set_missing_bib(capsys, mock_init_sample):
sys.argv = f"bibm pdf 1957RvMP...00..000X file.pdf".split()
pathlib.Path("file.pdf").touch()
cli.main()
captured = capsys.readouterr()
assert captured.out == \
"\nError: BibTex entry is not in Bibmanager database.\n"
bib = bm.find(bibcode='1957RvMP...29..547B')
assert bib.pdf is None
def test_reset_config(capsys, mock_init_sample):
pathlib.Path(u.BM_BIBFILE).touch()
# Simulate user input:
sys.argv = "bibm reset -c".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == "Resetting config parameters.\n"
assert filecmp.cmp(u.HOME + "config", u.ROOT + "config")
assert set(os.listdir(u.HOME)) == \
set(["bm_database.pickle", "bm_bibliography.bib", "config", "examples"])
def test_config_help(capsys, mock_init_sample):
sys.argv = "bibm config paper".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == (
"\nThe 'paper' parameter sets the default paper "
"format for latex compilation outputs\n"
"(not for pdflatex, which is automatic). Typical options are 'letter'\n"
"(e.g., for ApJ articles) or 'A4' (e.g., for A&A).\n\n"
"The current paper format is: 'letter'.\n")
def test_cli_ads_search_next(capsys, reqs, mock_prompt_session, mock_init):
cm.set('ads_display', '2')
am.search.__defaults__ = 0, 2, 'pubdate+desc'
sys.argv = "bibm ads-search".split()
cli.main()
captured = capsys.readouterr()
sys.argv = "bibm ads-search -n".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == f"""
def test_cli_open_prompt(capsys, mock_init_sample, mock_call,
mock_prompt_session):
pathlib.Path(f"{u.BM_PDF()}Slipher1913.pdf").touch()
sys.argv = f"bibm open".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == ("Syntax is: key: KEY_VALUE\n"
" or: bibcode: BIBCODE_VALUE\n"
" or: pdf: PDF_VALUE\n"
"(Press 'tab' for autocomplete)\n\n")
def test_cli_open_fetch(capsys, mock_init_sample, mock_call, reqs, mock_input):
sys.argv = f"bibm open 1957RvMP...29..547B".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == (
"\nError: Entry does not have a PDF in the database\n"
"Fetch from ADS?\n"
"[]yes [n]o\n\n"
"Fetching PDF file from Journal website:\n"
f"Saved PDF to: '{u.BM_PDF()}Burbidge1957_RvMP_29_547.pdf'.\n")
def test_older_pickle(capsys, mock_init_sample, mock_prompt_session):
# Mock pickle DB file with older version than bibmanager:
with open(u.BM_DATABASE, 'wb') as handle:
pickle.dump([], handle, protocol=pickle.HIGHEST_PROTOCOL)
# Simulate user input:
sys.argv = "bibm search".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == f"""Updating database file from version 0.0.0 to version {bibmanager.__version__}.
def test_cli_reset_all(capsys, mock_init_sample):
pathlib.Path(u.BM_BIBFILE()).touch()
cm.set("ads_display", "10")
captured = capsys.readouterr()
# Simulate user input:
sys.argv = "bibm reset".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == "Initializing new bibmanager database.\n" \
"Resetting config parameters.\n"
assert set(os.listdir(u.HOME)) == set(["config", "examples", "pdf"])
def test_config_display(capsys, mock_init_sample):
sys.argv = "bibm config".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == ("\nbibmanager configuration file:\n"
"PARAMETER VALUE\n"
"----------- -----\n"
"style autumn\n"
"text_editor default\n"
"paper letter\n"
"ads_token None\n"
"ads_display 20\n")
def test_reset_keep_database(capsys, mock_init_sample):
bibfile = u.HOME + "examples/sample.bib"
# Simulate user input:
sys.argv = f"bibm reset {bibfile}".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == f"Initializing new bibmanager database with BibTeX file: '{bibfile}'.\n" \
"Resetting config parameters.\n"
assert set(os.listdir(u.HOME)) == set(
["bm_database.pickle", "bm_bibliography.bib", "config", "examples"])
bibs = bm.loadfile(bibfile)
assert len(bibs) == 17
def test_cli_pdf_set_prompt_missing_pdf(capsys, mock_init_sample,
mock_prompt_session):
sys.argv = f"bibm pdf".split()
pathlib.Path(f"file.pdf").touch()
cli.main()
captured = capsys.readouterr()
assert captured.out == (
"Syntax is: key: KEY_VALUE PDF_FILE FILENAME\n"
" or: bibcode: BIBCODE_VALUE PDF_FILE FILENAME\n"
"(output FILENAME is optional, set it to guess for automated naming)\n"
"\n\nError: Path to PDF file is missing.\n")
bib = bm.find(bibcode='1957RvMP...29..547B')
assert bib.pdf is None
def test_cli_pdf_set_prompt_guess(capsys, mock_init_sample, mock_call,
mock_prompt_session):
sys.argv = f"bibm pdf".split()
pathlib.Path(f"file.pdf").touch()
cli.main()
captured = capsys.readouterr()
assert captured.out == (
"Syntax is: key: KEY_VALUE PDF_FILE FILENAME\n"
" or: bibcode: BIBCODE_VALUE PDF_FILE FILENAME\n"
"(output FILENAME is optional, set it to guess for automated naming)\n"
f"\nSaved PDF to: '{u.BM_PDF()}Burbidge1957_RvMP_29_547.pdf'.\n")
bib = bm.find(bibcode='1957RvMP...29..547B')
assert bib.pdf == 'Burbidge1957_RvMP_29_547.pdf'
assert 'Burbidge1957_RvMP_29_547.pdf' in os.listdir(u.BM_PDF())
def test_future_pickle(capsys, mock_init_sample):
# Mock pickle DB file with later version than bibmanager:
future_version = '2.0.0'
with open(u.BM_DATABASE, 'wb') as handle:
pickle.dump([], handle, protocol=pickle.HIGHEST_PROTOCOL)
pickle.dump(future_version, handle, protocol=pickle.HIGHEST_PROTOCOL)
# Simulate user input:
sys.argv = "bibm reset".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == \
(f"Bibmanager version ({bibmanager.__version__}) is older "
f"than saved database. Please update to version {future_version}.\n")
def test_reset_database(capsys, mock_init_sample):
pathlib.Path(u.BM_BIBFILE).touch()
cm.set("ads_display", "10")
captured = capsys.readouterr()
# Simulate user input:
sys.argv = "bibm reset -d".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == "Initializing new bibmanager database.\n"
# filecmp does not seem to work fine, will do manually:
# assert not filecmp.cmp(u.HOME+"config", u.ROOT+"config")
with open(u.ROOT + "config", "r") as f:
rconfig = f.read()
with open(u.HOME + "config", "r") as f:
hconfig = f.read()
assert rconfig != hconfig
assert set(os.listdir(u.HOME)) == set(["config", "examples"])
def test_config_invalid_value(capsys, mock_init_sample):
sys.argv = "bibm config ads_display A224".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == \
"\nError: The ads_display value must be a positive integer.\n"
def test_ads_search_next_empty(capsys, reqs, mock_prompt_session, mock_init):
sys.argv = "bibm ads-search -n".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == f"""There are no more entries for this querry.\n"""
def test_ads_search_empty(capsys, reqs, mock_prompt_session, mock_init):
sys.argv = "bibm ads-search".split()
cli.main()
captured = capsys.readouterr()
assert captured.out == f"""(Press 'tab' for autocomplete)\n\n"""
