Exemple #1
0
def test_transformer_when_block(patch, tree):
    """
    Ensures when_block nodes are transformed correctly
    """
    patch.object(Transformer, 'implicit_output')
    result = Transformer.when_block([tree])
    Transformer.implicit_output.assert_called_with(tree)
    assert result == Tree('when_block', [tree])
Exemple #2
0
def test_transformer_when_block(patch, tree):
    """
    Ensures when_block nodes are transformed correctly
    """
    tree.data = 'when_service'
    tree.output = None
    tree.children = [Token('NAME', '.name.')]
    tree.child_token.return_value = Token('NAME', '.child.', line=42)
    tree.path.child_token.return_value = Token('NAME', '.path.')
    result = Transformer.when_block([tree, 'block'])
    assert result == Tree('concise_when_block', [
        Token('NAME', '.child.'),
        Token('NAME', '.path.'),
        Tree('when_block', [tree, 'block'])
    ])
Exemple #3
0
def test_transformer_when_block_no_command(patch, tree):
    """
    Ensures when_block nodes without command are transformed correctly
    """
    patch.object(Transformer, 'argument_shorthand')
    tree.data = 'when_service'
    tree.output = None
    tree.children = [Token('NAME', '.name.')]
    tree.service_fragment.command = None
    result = Transformer.when_block([tree, 'block'])
    assert result == Tree('when_block', [
        Tree('service', [
            Tree('path', [tree.child_token()]),
            tree.service_fragment,
        ]),
        'block',
    ])
Exemple #4
0
def test_transformer_when_block(patch, tree):
    """
    Ensures when_block nodes are transformed correctly
    """
    tree.data = "when_service"
    tree.output = None
    tree.children = [Token("NAME", ".name.")]
    tree.child_token.return_value = Token("NAME", ".child.", line=42)
    tree.path.child_token.return_value = Token("NAME", ".path.")
    result = Transformer.when_block([tree, "block"])
    assert result == Tree(
        "concise_when_block",
        [
            Token("NAME", ".child."),
            Token("NAME", ".path."),
            Tree("when_block", [tree, "block"]),
        ],
    )
Exemple #5
0
def test_transformer_when_block_no_command(patch, tree):
    """
    Ensures when_block nodes without command are transformed correctly
    """
    patch.object(Transformer, "argument_shorthand")
    tree.data = "when_service"
    tree.output = None
    tree.children = [Token("NAME", ".name.")]
    tree.service_fragment.command = None
    result = Transformer.when_block([tree, "block"])
    assert result == Tree(
        "when_block",
        [
            Tree(
                "service",
                [
                    Tree("path", [tree.child_token()]),
                    tree.service_fragment,
                ],
            ),
            "block",
        ],
    )