def test_add_yaml_node_at_end_when_before_not_found():
source_lines = textwrap.dedent("""\
language: python
python:
- 3.6
script: pytest tests
""").splitlines(True)
result = add_yaml_node(source_lines, 'dist', 'xenial', before='sudo')
assert "".join(result) == textwrap.dedent("""\
language: python
python:
- 3.6
script: pytest tests
dist: xenial
""")
def test_add_yaml_node_before():
source_lines = textwrap.dedent("""\
language: python
python:
- 3.6
script: pytest tests
""").splitlines(True)
result = add_yaml_node(source_lines, 'dist', 'xenial', before='python')
assert "".join(result) == textwrap.dedent("""\
language: python
dist: xenial
python:
- 3.6
script: pytest tests
""")