aselimov/pubs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixes argument and adds tests for overwrite on import.

Browse Source
This commit is contained in:
Olivier Mangin 2017-12-11 15:49:06 -05:00
parent 1cd95bdeec
commit 3e31c4455e
2 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pubs/commands/import_cmd.py
View File

@ -20,7 +20,7 @@ def parser(subparsers, conf):
help="don't copy document files, just create a link.")
parser.add_argument('keys', nargs='*',
help="one or several keys to import from the file")
parser.add_argument('-O', '--overwrite', default=False,
parser.add_argument('-O', '--overwrite', action='store_true', default=False,
help="Overwrite keys already in the database")
return parser

20
tests/test_usecase.py
View File

@ -687,6 +687,26 @@ class TestUsecase(DataCommandTestCase):
outs = self.execute_cmds(cmds)
self.assertEqual(1 + 1, len(outs[-1].split('\n')))
def test_import_does_not_overwrite(self):
cmds = ['pubs init',
'pubs import data/ Page99',
'pubs import data/',
'pubs list'
]
with self.assertRaises(FakeSystemExit):
self.execute_cmds(cmds)
def test_import_overwrites(self):
cmds = ['pubs init',
'pubs import data/ Page99',
'pubs import --overwrite data/ Page99',
'pubs list'
]
outs = self.execute_cmds(cmds)
self.assertEqual(1 + 1, len(outs[-1].split('\n')))
def test_update(self):
cmds = ['pubs init',
'pubs add data/pagerank.bib',