def test_insert_metadata_in_file(tmpdir):
file_content = dedent(
"""
2016-02-26 * "Uncle Boons" "Eating out alone"
Liabilities:US:Chase:Slate -24.84 USD
Expenses:Food:Restaurant 24.84 USD
"""
)
samplefile = tmpdir.mkdir("fava_util_file").join("example.beancount")
samplefile.write(file_content)
assert samplefile.read() == dedent(file_content)
assert len(tmpdir.listdir()) == 1
insert_metadata_in_file(str(samplefile), 1, "metadata", "test1")
assert samplefile.read() == dedent(
"""
2016-02-26 * "Uncle Boons" "Eating out alone"
metadata: "test1"
Liabilities:US:Chase:Slate -24.84 USD
Expenses:Food:Restaurant 24.84 USD
"""
)
insert_metadata_in_file(str(samplefile), 1, "metadata", "test2")
assert samplefile.read() == dedent(
"""
2016-02-26 * "Uncle Boons" "Eating out alone"
metadata: "test2"
metadata: "test1"
Liabilities:US:Chase:Slate -24.84 USD
Expenses:Food:Restaurant 24.84 USD
"""
)
def test_insert_metadata_in_file(tmpdir):
file_content = dedent("""
2016-02-26 * "Uncle Boons" "Eating out alone"
Liabilities:US:Chase:Slate -24.84 USD
Expenses:Food:Restaurant 24.84 USD
""")
samplefile = tmpdir.mkdir('fava_util_file').join('example.beancount')
samplefile.write(file_content)
assert samplefile.read() == dedent(file_content)
assert len(tmpdir.listdir()) == 1
insert_metadata_in_file(str(samplefile), 1, 'metadata', 'test1')
assert samplefile.read() == dedent("""
2016-02-26 * "Uncle Boons" "Eating out alone"
metadata: "test1"
Liabilities:US:Chase:Slate -24.84 USD
Expenses:Food:Restaurant 24.84 USD
""")
insert_metadata_in_file(str(samplefile), 1, 'metadata', 'test2')
assert samplefile.read() == dedent("""
2016-02-26 * "Uncle Boons" "Eating out alone"
metadata: "test2"
metadata: "test1"
Liabilities:US:Chase:Slate -24.84 USD
Expenses:Food:Restaurant 24.84 USD
""")
def test_insert_metadata_in_file(tmp_path) -> None:
file_content = dedent("""\
2016-02-26 * "Uncle Boons" "Eating out alone"
Liabilities:US:Chase:Slate -24.84 USD
Expenses:Food:Restaurant 24.84 USD
""")
samplefile = tmp_path / "example.beancount"
samplefile.write_text(file_content)
# Insert some metadata lines.
insert_metadata_in_file(str(samplefile), 1, "metadata", "test1")
insert_metadata_in_file(str(samplefile), 1, "metadata", "test2")
assert samplefile.read_text("utf-8") == dedent("""\
2016-02-26 * "Uncle Boons" "Eating out alone"
metadata: "test2"
metadata: "test1"
Liabilities:US:Chase:Slate -24.84 USD
Expenses:Food:Restaurant 24.84 USD
""")
# Check that inserting also works if the next line is empty.
insert_metadata_in_file(str(samplefile), 5, "metadata", "test1")
assert samplefile.read_text("utf-8") == dedent("""\
2016-02-26 * "Uncle Boons" "Eating out alone"
metadata: "test2"
metadata: "test1"
Liabilities:US:Chase:Slate -24.84 USD
Expenses:Food:Restaurant 24.84 USD
metadata: "test1"
""")
