linode-cli/tests/unit/test_completion.py at 07adbe288ff2eb68753da270d962f31c1acf3ba4 · linode/linode-cli · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
#!/usr/local/bin/python3
"""
Unit tests for linodecli.completion
"""

from linodecli import completion


class TestCompletion:
    """
    Unit tests for linodecli.completion
    """

    ops = {"temp_key": {"temp_action": "description"}}
    fish_expected = """# This is a generated file by Linode-CLI! Do not modify!
complete -c linode-cli -n "not __fish_seen_subcommand_from temp_key" -x -a 'temp_key --help'
complete -c linode -n "not __fish_seen_subcommand_from temp_key" -x -a 'temp_key --help'
complete -c lin -n "not __fish_seen_subcommand_from temp_key" -x -a 'temp_key --help'
complete -c linode-cli -n "__fish_seen_subcommand_from temp_key" -x -a 'temp_action --help'
complete -c linode -n "__fish_seen_subcommand_from temp_key" -x -a 'temp_action --help'
complete -c lin -n "__fish_seen_subcommand_from temp_key" -x -a 'temp_action --help'"""
    bash_expected = """# This is a generated file by Linode-CLI! Do not modify!
_linode_cli()
{
local cur prev opts
COMPREPLY=()
cur="${COMP_WORDS[COMP_CWORD]}"
prev="${COMP_WORDS[COMP_CWORD-1]}"

case "${prev}" in
    linode-cli | linode | lin)
        COMPREPLY=( $(compgen -W "temp_key --help" -- ${cur}) )
        return 0
        ;;
    temp_key)
        COMPREPLY=( $(compgen -W "temp_action --help" -- ${cur}) )
        return 0
        ;;
    *)
        ;;
esac
}

complete -F _linode_cli linode-cli
complete -F _linode_cli linode
complete -F _linode_cli lin"""
    zsh_expected = """#compdef linode-cli linode lin

# This is a generated file by Linode-CLI! Do not modify!
local -a subcommands
subcommands=(
'temp_key --help'
)

local -a command
local -a opts

_arguments -C \\
  "1: :(temp_key)" \\
  '*:: :->subcmds' && return 0

if (( CURRENT == 2 )); then
  case $words[1] in
    temp_key)
      command=(
          'temp_action --help'
      )
      _describe -t commands "temp_key command" command
      ;;
  esac
fi
"""

    def test_fish_completion(self, mocker):
        """
        Test if the fish completion renders correctly
        """
        actual = completion.get_fish_completions(self.ops)
        assert actual == self.fish_expected

    def test_bash_completion(self, mocker):
        """
        Test if the bash completion renders correctly
        """
        # mocker = mocker.patch('linodecli-completion.get_bash_completions', return_value=self.bash_expected)
        actual = completion.get_bash_completions(self.ops)
        assert actual == self.bash_expected

    def test_zsh_completion(self):
        """
        Test if the Zsh completion renders correctly
        """
        actual = completion.get_zsh_completions(self.ops)
        assert actual == self.zsh_expected

    def test_get_completions(self):
        """
        Test get_completions for arg parse
        """
        actual = completion.get_completions(self.ops, False, "bash")
        assert actual == self.bash_expected

        actual = completion.get_completions(self.ops, False, "fish")
        assert actual == self.fish_expected

        actual = completion.get_completions(self.ops, False, "zsh")
        assert actual == self.zsh_expected

        actual = completion.get_completions(self.ops, False, "notrealshell")
        assert "invoke" in actual

        actual = completion.get_completions(self.ops, True, "")
        assert "[SHELL]" in actual