diff --git a/tests/test_hooks.py b/tests/test_hooks.py index d5d61d7187f2dc1a968df48cc671af5193de58d9..95c2e60388d4089135a04442386d7900102d6e99 100644 --- a/tests/test_hooks.py +++ b/tests/test_hooks.py @@ -5,13 +5,19 @@ import itertools import pytest from import2ledger import hooks -from import2ledger.hooks import add_entity, default_date, filter_by_date +from import2ledger.hooks import add_entity, default_date, filter_by_date, ledger_entry def test_load_all(): all_hooks = list(hooks.load_all()) positions = {hook: index for index, hook in enumerate(all_hooks)} - assert positions[default_date.DefaultDateHook] < positions[add_entity.AddEntityHook] - assert positions[add_entity.AddEntityHook] < positions[filter_by_date.FilterByDateHook] + expected_order = [ + default_date.DefaultDateHook, + add_entity.AddEntityHook, + filter_by_date.FilterByDateHook, + ledger_entry.LedgerEntryHook, + ] + actual_order = list(sorted(expected_order, key=positions.__getitem__)) + assert actual_order == expected_order @pytest.mark.parametrize('in_key,payee,out_key,expected', [ ('payee', 'Alex Smith', 'entity', 'Smith-Alex'),