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", ])