@@ -28,24 +28,37 @@ def iter_entries(in_file):
else:
lines.append(line)
if lines:
yield ''.join(lines)
def entries2set(in_file):
return set(normalize_whitespace(e) for e in iter_entries(in_file))
def expected_entries(path):
path = pathlib.Path(path)
if not path.is_absolute():
path = DATA_DIR / path
with path.open() as in_file:
return entries2set(in_file)
def test_fees_import():
arglist = ARGLIST + [
'-c', 'One',
pathlib.Path(DATA_DIR, 'PatreonEarnings.csv').as_posix(),
]
exitcode, stdout, _ = run_main(arglist)
assert exitcode == 0
actual = entries2set(stdout)
assert actual == expected_entries('test_main_fees_import.ledger')
def test_date_range_import():
'--date-range', '2017/10/01-',
expected = {entry for entry in expected_entries('test_main_fees_import.ledger')
if entry.startswith('2017/10/')}
assert actual == expected