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 """)