def test_token_generate_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_generate()
def test_token_generate_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_generate()
def test_token_generate(self):
'''
Test kuebadm.token_generate without parameters
'''
result = {'retcode': 0, 'stdout': 'token'}
salt_mock = {
'cmd.run_all': MagicMock(return_value=result),
}
with patch.dict(kubeadm.__salt__, salt_mock):
assert kubeadm.token_generate() == 'token'
salt_mock['cmd.run_all'].assert_called_with(
['kubeadm', 'token', 'generate'])
def test_token_generate(self):
"""
Test kuebadm.token_generate without parameters
"""
result = {"retcode": 0, "stdout": "token"}
salt_mock = {
"cmd.run_all": MagicMock(return_value=result),
}
with patch.dict(kubeadm.__salt__, salt_mock):
assert kubeadm.token_generate() == "token"
salt_mock["cmd.run_all"].assert_called_with(
["kubeadm", "token", "generate"])
def test_token_generate_params(self):
'''
Test kuebadm.token_generate with parameters
'''
result = {'retcode': 0, 'stdout': 'token'}
salt_mock = {
'cmd.run_all': MagicMock(return_value=result),
}
with patch.dict(kubeadm.__salt__, salt_mock):
assert kubeadm.token_generate(kubeconfig='/kube.cfg',
rootfs='/mnt') == 'token'
salt_mock['cmd.run_all'].assert_called_with([
'kubeadm', 'token', 'generate', '--kubeconfig', '/kube.cfg',
'--rootfs', '/mnt'
])
def test_token_generate_params(self):
"""
Test kuebadm.token_generate with parameters
"""
result = {"retcode": 0, "stdout": "token"}
salt_mock = {
"cmd.run_all": MagicMock(return_value=result),
}
with patch.dict(kubeadm.__salt__, salt_mock):
assert (kubeadm.token_generate(kubeconfig="/kube.cfg",
rootfs="/mnt") == "token")
salt_mock["cmd.run_all"].assert_called_with([
"kubeadm",
"token",
"generate",
"--kubeconfig",
"/kube.cfg",
"--rootfs",
"/mnt",
])
