Exemplo n.º 1
0
    def test_not_missing_fun_calling_wfuncs(self):
        opts = salt.config.client_config(self.get_config_file_path("master"))
        opts["list_hosts"] = False
        opts["argv"] = ["state.show_highstate"]
        opts["selected_target_option"] = "glob"
        opts["tgt"] = "localhost"
        opts["arg"] = []
        roster = os.path.join(RUNTIME_VARS.TMP_CONF_DIR, "roster")
        handle_ssh_ret = [
            {
                "localhost": {}
            },
        ]

        expected = {"localhost": {}}
        display_output = MagicMock()
        with patch("salt.roster.get_roster_file",
                   MagicMock(return_value=roster)), patch(
                       "salt.client.ssh.SSH.handle_ssh",
                       MagicMock(return_value=handle_ssh_ret)), patch(
                           "salt.client.ssh.SSH.key_deploy",
                           MagicMock(return_value=expected)), patch(
                               "salt.output.display_output", display_output):
            client = ssh.SSH(opts)
            client.event = MagicMock()
            ret = next(client.run_iter())
            assert "localhost" in ret
            assert "fun" in ret["localhost"]
            client.run()
        display_output.assert_called_once_with(expected, "nested", opts)
        self.assertIs(ret, handle_ssh_ret[0])
        assert len(client.event.fire_event.call_args_list) == 2
        assert "fun" in client.event.fire_event.call_args_list[0][0][0]
        assert "fun" in client.event.fire_event.call_args_list[1][0][0]
Exemplo n.º 2
0
 def test_password_failure(self):
     """
     Check password failures when trying to deploy keys
     """
     opts = salt.config.client_config(self.get_config_file_path("master"))
     opts["list_hosts"] = False
     opts["argv"] = ["test.ping"]
     opts["selected_target_option"] = "glob"
     opts["tgt"] = "localhost"
     opts["arg"] = []
     roster = os.path.join(RUNTIME_VARS.TMP_CONF_DIR, "roster")
     handle_ssh_ret = [
         {
             "localhost": {
                 "retcode": 255,
                 "stderr": "Permission denied (publickey).\r\n",
                 "stdout": "",
             }
         },
     ]
     expected = {"localhost": "Permission denied (publickey)"}
     display_output = MagicMock()
     with patch("salt.roster.get_roster_file",
                MagicMock(return_value=roster)), patch(
                    "salt.client.ssh.SSH.handle_ssh",
                    MagicMock(return_value=handle_ssh_ret)), patch(
                        "salt.client.ssh.SSH.key_deploy",
                        MagicMock(return_value=expected)), patch(
                            "salt.output.display_output", display_output):
         client = ssh.SSH(opts)
         ret = next(client.run_iter())
         with self.assertRaises(SystemExit):
             client.run()
     display_output.assert_called_once_with(expected, "nested", opts)
     self.assertIs(ret, handle_ssh_ret[0])