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