def test_token_list_params(self):
"""
Test kuebadm.token_list with parameters
"""
output = "H1 H2 H31 H32 H4\n1 2 3.1 3.2 4"
result = {"retcode": 0, "stdout": output}
salt_mock = {
"cmd.run_all": MagicMock(return_value=result),
}
with patch.dict(kubeadm.__salt__, salt_mock):
result = kubeadm.token_list(kubeconfig="/kube.cfg", rootfs="/mnt")
assert result == [{
"h1": "1",
"h2": "2",
"h31 h32": "3.1 3.2",
"h4": "4"
}]
salt_mock["cmd.run_all"].assert_called_with([
"kubeadm",
"token",
"list",
"--kubeconfig",
"/kube.cfg",
"--rootfs",
"/mnt",
])
def test_token_list_error(self):
'''
Test kuebadm.token_generate error
'''
result = {'retcode': 1, 'stderr': 'error'}
salt_mock = {
'cmd.run_all': MagicMock(return_value=result),
}
with patch.dict(kubeadm.__salt__, salt_mock):
with pytest.raises(CommandExecutionError):
assert kubeadm.token_list()
def test_token_list_error(self):
"""
Test kuebadm.token_generate error
"""
result = {"retcode": 1, "stderr": "error"}
salt_mock = {
"cmd.run_all": MagicMock(return_value=result),
}
with patch.dict(kubeadm.__salt__, salt_mock):
with pytest.raises(CommandExecutionError):
assert kubeadm.token_list()
def test_token_empty(self):
"""
Test kuebadm.token_list when no outout
"""
result = {"retcode": 0, "stdout": ""}
salt_mock = {
"cmd.run_all": MagicMock(return_value=result),
}
with patch.dict(kubeadm.__salt__, salt_mock):
assert kubeadm.token_list() == []
salt_mock["cmd.run_all"].assert_called_with(
["kubeadm", "token", "list"])
def test_token_list_broken_lines(self):
'''
Test kuebadm.token_list with multiple tokens, one broken
'''
output = 'H1 H2 H31 H32 H4\n1 2 3.1 3.2 4\na b c d e'
result = {'retcode': 0, 'stdout': output}
salt_mock = {
'cmd.run_all': MagicMock(return_value=result),
}
with patch.dict(kubeadm.__salt__, salt_mock):
assert kubeadm.token_list() == [{
'h1': '1', 'h2': '2', 'h31 h32': '3.1 3.2', 'h4': '4'
}]
def test_token_list_broken_lines(self):
"""
Test kuebadm.token_list with multiple tokens, one broken
"""
output = "H1 H2 H31 H32 H4\n1 2 3.1 3.2 4\na b c d e"
result = {"retcode": 0, "stdout": output}
salt_mock = {
"cmd.run_all": MagicMock(return_value=result),
}
with patch.dict(kubeadm.__salt__, salt_mock):
assert kubeadm.token_list() == [{
"h1": "1",
"h2": "2",
"h31 h32": "3.1 3.2",
"h4": "4"
}]
def test_token_list(self):
'''
Test kuebadm.token_list without parameters
'''
output = 'H1 H2 H31 H32 H4\n1 2 3.1 3.2 4'
result = {'retcode': 0, 'stdout': output}
salt_mock = {
'cmd.run_all': MagicMock(return_value=result),
}
with patch.dict(kubeadm.__salt__, salt_mock):
assert kubeadm.token_list() == [{
'h1': '1', 'h2': '2', 'h31 h32': '3.1 3.2', 'h4': '4'
}]
salt_mock['cmd.run_all'].assert_called_with(
['kubeadm', 'token', 'list']
)
def test_token_list(self):
"""
Test kuebadm.token_list without parameters
"""
output = "H1 H2 H31 H32 H4\n1 2 3.1 3.2 4"
result = {"retcode": 0, "stdout": output}
salt_mock = {
"cmd.run_all": MagicMock(return_value=result),
}
with patch.dict(kubeadm.__salt__, salt_mock):
assert kubeadm.token_list() == [{
"h1": "1",
"h2": "2",
"h31 h32": "3.1 3.2",
"h4": "4"
}]
salt_mock["cmd.run_all"].assert_called_with(
["kubeadm", "token", "list"])
