135 lines
4.2 KiB
Python
135 lines
4.2 KiB
Python
"""Test link checker for repository documents"""
|
|
# Copyright © 2020 Brett Smith
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
import itertools
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from . import testutil
|
|
|
|
from conservancy_beancount import errors as errormod
|
|
from conservancy_beancount.plugin import meta_repo_links
|
|
|
|
METADATA_KEYS = [
|
|
'approval',
|
|
'check',
|
|
'contract',
|
|
'invoice',
|
|
'purchase-order',
|
|
'receipt',
|
|
'statement',
|
|
]
|
|
|
|
GOOD_LINKS = [Path(s) for s in [
|
|
'Projects/project-data.yml',
|
|
'Projects/project-list.yml',
|
|
]]
|
|
|
|
BAD_LINKS = [Path(s) for s in [
|
|
'NonexistentDirectory/NonexistentFile1.txt',
|
|
'NonexistentDirectory/NonexistentFile2.txt',
|
|
]]
|
|
|
|
NOT_FOUND_MSG = '{} not found in repository: {}'.format
|
|
|
|
def build_meta(keys=None, *sources):
|
|
if keys is None:
|
|
keys = iter(METADATA_KEYS)
|
|
sources = (itertools.cycle(src) for src in sources)
|
|
return {key: ' '.join(str(x) for x in rest)
|
|
for key, *rest in zip(keys, *sources)}
|
|
|
|
@pytest.fixture(scope='module')
|
|
def hook():
|
|
config = testutil.TestConfig(repo_path='repository')
|
|
return meta_repo_links.MetaRepoLinks(config)
|
|
|
|
def test_error_with_no_repository():
|
|
config = testutil.TestConfig(repo_path=None)
|
|
with pytest.raises(errormod.ConfigurationError):
|
|
meta_repo_links.MetaRepoLinks(config)
|
|
|
|
def test_good_txn_links(hook):
|
|
meta = build_meta(None, GOOD_LINKS)
|
|
txn = testutil.Transaction(**meta, postings=[
|
|
('Income:Donations', -5),
|
|
('Assets:Cash', 5),
|
|
])
|
|
assert not list(hook.run(txn))
|
|
|
|
def test_good_post_links(hook):
|
|
meta = build_meta(None, GOOD_LINKS)
|
|
txn = testutil.Transaction(postings=[
|
|
('Income:Donations', -5, meta),
|
|
('Assets:Cash', 5),
|
|
])
|
|
assert not list(hook.run(txn))
|
|
|
|
def test_bad_txn_links(hook):
|
|
meta = build_meta(None, BAD_LINKS)
|
|
txn = testutil.Transaction(**meta, postings=[
|
|
('Income:Donations', -5),
|
|
('Assets:Cash', 5),
|
|
])
|
|
expected = {NOT_FOUND_MSG(key, value) for key, value in meta.items()}
|
|
actual = {error.message for error in hook.run(txn)}
|
|
assert expected == actual
|
|
|
|
def test_bad_post_links(hook):
|
|
meta = build_meta(None, BAD_LINKS)
|
|
txn = testutil.Transaction(postings=[
|
|
('Income:Donations', -5, meta.copy()),
|
|
('Assets:Cash', 5),
|
|
])
|
|
expected = {NOT_FOUND_MSG(key, value) for key, value in meta.items()}
|
|
actual = {error.message for error in hook.run(txn)}
|
|
assert expected == actual
|
|
|
|
@pytest.mark.parametrize('ext_doc', [
|
|
'rt:123',
|
|
'rt:456/789',
|
|
'rt://ticket/23',
|
|
'rt://ticket/34/attachments/567890',
|
|
])
|
|
def test_docs_outside_repository_not_checked(hook, ext_doc):
|
|
txn = testutil.Transaction(
|
|
receipt='{} {} {}'.format(GOOD_LINKS[0], ext_doc, BAD_LINKS[1]),
|
|
postings=[
|
|
('Income:Donations', -5),
|
|
('Assets:Cash', 5),
|
|
])
|
|
expected = {NOT_FOUND_MSG('receipt', BAD_LINKS[1])}
|
|
actual = {error.message for error in hook.run(txn)}
|
|
assert expected == actual
|
|
|
|
def test_mixed_results(hook):
|
|
txn = testutil.Transaction(
|
|
approval='{} {}'.format(*GOOD_LINKS),
|
|
contract='{} {}'.format(BAD_LINKS[0], GOOD_LINKS[1]),
|
|
postings=[
|
|
('Income:Donations', -5, {'invoice': '{} {}'.format(*BAD_LINKS)}),
|
|
('Assets:Cash', 5, {'statement': '{} {}'.format(GOOD_LINKS[0], BAD_LINKS[1])}),
|
|
])
|
|
expected = {
|
|
NOT_FOUND_MSG('contract', BAD_LINKS[0]),
|
|
NOT_FOUND_MSG('invoice', BAD_LINKS[0]),
|
|
NOT_FOUND_MSG('invoice', BAD_LINKS[1]),
|
|
NOT_FOUND_MSG('statement', BAD_LINKS[1]),
|
|
}
|
|
actual = {error.message for error in hook.run(txn)}
|
|
assert expected == actual
|