91 lines
3.1 KiB
Python
91 lines
3.1 KiB
Python
"""Test Metadata class"""
|
|
# 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 pytest
|
|
|
|
from . import testutil
|
|
|
|
from conservancy_beancount import data
|
|
|
|
@pytest.fixture
|
|
def simple_txn(index=None, key=None):
|
|
return testutil.Transaction(note='txn note', postings=[
|
|
('Assets:Cash', 5),
|
|
('Income:Donations', -5, {'note': 'donation love', 'extra': 'Extra'}),
|
|
])
|
|
SIMPLE_TXN_METAKEYS = frozenset(['filename', 'lineno', 'note'])
|
|
|
|
def test_metadata_transforms_source():
|
|
source = {'1': 'one'}
|
|
meta = data.Metadata(source)
|
|
meta['2'] = 'two'
|
|
assert source['2'] == 'two'
|
|
del meta['1']
|
|
assert set(source) == {'2'}
|
|
|
|
@pytest.mark.parametrize('value', [
|
|
'',
|
|
'link',
|
|
' link',
|
|
'link ',
|
|
'link1 link2',
|
|
' link1 link2 link3 ',
|
|
])
|
|
def test_get_links(value):
|
|
meta = data.Metadata({'key': value})
|
|
assert list(meta.get_links('key')) == value.split()
|
|
|
|
def test_get_links_missing():
|
|
meta = data.Metadata({})
|
|
assert not meta.get_links('key')
|
|
|
|
@pytest.mark.parametrize('value', testutil.NON_STRING_METADATA_VALUES)
|
|
def test_get_links_bad_type(value):
|
|
meta = data.Metadata({'key': value})
|
|
with pytest.raises(TypeError):
|
|
meta.get_links('key')
|
|
|
|
def test_first_link_from_txn(simple_txn):
|
|
meta = data.PostingMeta(simple_txn, 0)
|
|
assert meta.first_link('note') == 'txn'
|
|
|
|
def test_first_link_from_post_override(simple_txn):
|
|
meta = data.PostingMeta(simple_txn, 1)
|
|
assert meta.first_link('note') == 'donation'
|
|
|
|
def test_first_link_is_only_link(simple_txn):
|
|
meta = data.PostingMeta(simple_txn, 1)
|
|
assert meta.first_link('extra') == 'Extra'
|
|
|
|
def test_first_link_nonexistent_metadata(simple_txn):
|
|
meta = data.PostingMeta(simple_txn, 1)
|
|
assert meta.first_link('Nonexistent') is None
|
|
|
|
def test_first_link_nonexistent_default(simple_txn):
|
|
meta = data.PostingMeta(simple_txn, 1)
|
|
assert meta.first_link('Nonexistent', 'missing') == 'missing'
|
|
|
|
@pytest.mark.parametrize('meta_value', testutil.NON_STRING_METADATA_VALUES)
|
|
def test_first_link_bad_type_metadata(simple_txn, meta_value):
|
|
simple_txn.meta['badmeta'] = meta_value
|
|
meta = data.PostingMeta(simple_txn, 1)
|
|
assert meta.first_link('badmeta') is None
|
|
|
|
@pytest.mark.parametrize('meta_value', testutil.NON_STRING_METADATA_VALUES)
|
|
def test_first_link_bad_type_default(simple_txn, meta_value):
|
|
simple_txn.meta['badmeta'] = meta_value
|
|
meta = data.PostingMeta(simple_txn, 1)
|
|
assert meta.first_link('badmeta', '_') == '_'
|