diff --git a/tests/testutil.py b/tests/testutil.py index 9534bff4382ead304b93a66256879a0ad1905e69..74ae8f24cfae926af23fe02f6de24dfdd83ffc89 100644 --- a/tests/testutil.py +++ b/tests/testutil.py @@ -17,12 +17,14 @@ import datetime import itertools import re +import unittest.mock import beancount.core.amount as bc_amount import beancount.core.data as bc_data import beancount.loader as bc_loader import beancount.parser.options as bc_options +import git import odf.element import odf.opendocument import odf.table @@ -288,6 +290,13 @@ class TestConfig: return self._rt_wrapper +def TestRepo(head_hexsha='abcd1234', dirty=False): + retval = unittest.mock.Mock(spec=git.Repo) + retval.is_dirty.return_value = dirty + retval.head.commit.hexsha = head_hexsha + return retval + + class _TicketBuilder: MESSAGE_ATTACHMENTS = [ ('(Unnamed)', 'multipart/alternative', '0b'),