152 lines
4.6 KiB
Python
152 lines
4.6 KiB
Python
"""Test link checker for RT links"""
|
|
# 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
|
|
|
|
import pytest
|
|
|
|
from . import testutil
|
|
|
|
from conservancy_beancount import errors as errormod
|
|
from conservancy_beancount.plugin import meta_rt_links
|
|
|
|
METADATA_KEYS = [
|
|
'approval',
|
|
'check',
|
|
'contract',
|
|
'invoice',
|
|
'purchase-order',
|
|
'receipt',
|
|
'rt-id',
|
|
'statement',
|
|
]
|
|
|
|
GOOD_LINKS = [
|
|
'rt:1',
|
|
'rt:1/5',
|
|
'rt://ticket/2',
|
|
'rt://ticket/3/attachments/15',
|
|
]
|
|
|
|
MALFORMED_LINKS = [
|
|
'rt:one',
|
|
'rt:two/three',
|
|
'rt://4',
|
|
'rt://ticket/5/attach/6',
|
|
]
|
|
|
|
NOT_FOUND_LINKS = [
|
|
'rt:1/10',
|
|
'rt:10',
|
|
'rt://ticket/9',
|
|
'rt://ticket/3/attachments/99',
|
|
]
|
|
|
|
MALFORMED_MSG = '{} link is malformed: {}'.format
|
|
NOT_FOUND_MSG = '{} not found in RT: {}'.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(rt_client=testutil.RTClient())
|
|
return meta_rt_links.MetaRTLinks(config)
|
|
|
|
def test_error_with_no_rt():
|
|
config = testutil.TestConfig()
|
|
with pytest.raises(errormod.ConfigurationError):
|
|
meta_rt_links.MetaRTLinks(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))
|
|
|
|
@pytest.mark.parametrize('link_source,format_error', [
|
|
(MALFORMED_LINKS, MALFORMED_MSG),
|
|
(NOT_FOUND_LINKS, NOT_FOUND_MSG),
|
|
])
|
|
def test_bad_txn_links(hook, link_source, format_error):
|
|
meta = build_meta(None, link_source)
|
|
txn = testutil.Transaction(**meta, postings=[
|
|
('Income:Donations', -5),
|
|
('Assets:Cash', 5),
|
|
])
|
|
expected = {format_error(key, value) for key, value in meta.items()}
|
|
actual = {error.message for error in hook.run(txn)}
|
|
assert expected == actual
|
|
|
|
@pytest.mark.parametrize('link_source,format_error', [
|
|
(MALFORMED_LINKS, MALFORMED_MSG),
|
|
(NOT_FOUND_LINKS, NOT_FOUND_MSG),
|
|
])
|
|
def test_bad_post_links(hook, link_source, format_error):
|
|
meta = build_meta(None, link_source)
|
|
txn = testutil.Transaction(postings=[
|
|
('Income:Donations', -5, meta.copy()),
|
|
('Assets:Cash', 5),
|
|
])
|
|
expected = {format_error(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', [
|
|
'statement.txt',
|
|
'https://example.org/',
|
|
])
|
|
def test_docs_outside_rt_not_checked(hook, ext_doc):
|
|
txn = testutil.Transaction(
|
|
receipt='{} {} {}'.format(GOOD_LINKS[0], ext_doc, MALFORMED_LINKS[1]),
|
|
postings=[
|
|
('Income:Donations', -5),
|
|
('Assets:Cash', 5),
|
|
])
|
|
expected = {MALFORMED_MSG('receipt', MALFORMED_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(MALFORMED_LINKS[0], GOOD_LINKS[1]),
|
|
postings=[
|
|
('Income:Donations', -5, {'invoice': '{} {}'.format(*NOT_FOUND_LINKS)}),
|
|
('Assets:Cash', 5, {'statement': '{} {}'.format(GOOD_LINKS[0], MALFORMED_LINKS[1])}),
|
|
])
|
|
expected = {
|
|
MALFORMED_MSG('contract', MALFORMED_LINKS[0]),
|
|
NOT_FOUND_MSG('invoice', NOT_FOUND_LINKS[0]),
|
|
NOT_FOUND_MSG('invoice', NOT_FOUND_LINKS[1]),
|
|
MALFORMED_MSG('statement', MALFORMED_LINKS[1]),
|
|
}
|
|
actual = {error.message for error in hook.run(txn)}
|
|
assert expected == actual
|