def test_config_images_pull_params(self):
'''
Test kuebadm.config_images_pull with parameters
'''
result = {
'retcode':
0,
'stdout':
'[config/images] Pulled image1\n'
'[config/images] Pulled image2\n'
}
salt_mock = {
'cmd.run_all': MagicMock(return_value=result),
}
with patch.dict(kubeadm.__salt__, salt_mock):
assert kubeadm.config_images_pull(
config='/kubeadm.cfg',
cri_socket='socket',
feature_gates='k=v',
kubernetes_version='version',
kubeconfig='/kube.cfg',
rootfs='/mnt') == ['image1', 'image2']
salt_mock['cmd.run_all'].assert_called_with([
'kubeadm', 'config', 'images', 'pull', '--config',
'/kubeadm.cfg', '--cri-socket', 'socket', '--feature-gates',
'k=v', '--kubernetes-version', 'version', '--kubeconfig',
'/kube.cfg', '--rootfs', '/mnt'
])
def test_config_images_pull_error(self):
'''
Test kuebadm.config_images_pull 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.config_images_pull()
def test_config_images_pull_error(self):
"""
Test kuebadm.config_images_pull 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.config_images_pull()
def test_config_images_pull(self):
'''
Test kuebadm.config_images_pull without parameters
'''
result = {'retcode': 0,
'stdout': '[config/images] Pulled image1\n'
'[config/images] Pulled image2\n'}
salt_mock = {
'cmd.run_all': MagicMock(return_value=result),
}
with patch.dict(kubeadm.__salt__, salt_mock):
assert kubeadm.config_images_pull() == ['image1', 'image2']
salt_mock['cmd.run_all'].assert_called_with(
['kubeadm', 'config', 'images', 'pull']
)
def test_config_images_pull(self):
"""
Test kuebadm.config_images_pull without parameters
"""
result = {
"retcode":
0,
"stdout":
"[config/images] Pulled image1\n[config/images] Pulled image2\n",
}
salt_mock = {
"cmd.run_all": MagicMock(return_value=result),
}
with patch.dict(kubeadm.__salt__, salt_mock):
assert kubeadm.config_images_pull() == ["image1", "image2"]
salt_mock["cmd.run_all"].assert_called_with(
["kubeadm", "config", "images", "pull"])
def test_config_images_pull_params(self):
"""
Test kuebadm.config_images_pull with parameters
"""
result = {
"retcode":
0,
"stdout":
"[config/images] Pulled image1\n"
"[config/images] Pulled image2\n",
}
salt_mock = {
"cmd.run_all": MagicMock(return_value=result),
}
with patch.dict(kubeadm.__salt__, salt_mock):
assert kubeadm.config_images_pull(
config="/kubeadm.cfg",
cri_socket="socket",
feature_gates="k=v",
kubernetes_version="version",
kubeconfig="/kube.cfg",
rootfs="/mnt",
) == ["image1", "image2"]
salt_mock["cmd.run_all"].assert_called_with([
"kubeadm",
"config",
"images",
"pull",
"--config",
"/kubeadm.cfg",
"--cri-socket",
"socket",
"--feature-gates",
"k=v",
"--kubernetes-version",
"version",
"--kubeconfig",
"/kube.cfg",
"--rootfs",
"/mnt",
])