def test__reboot_and_finish_deploy(self, get_pow_state_mock,
                                       power_action_mock):
        self.config(group='ansible',
                    post_deploy_get_power_state_retry_interval=0)

        with task_manager.acquire(self.context, self.node.uuid) as task:
            ansible_deploy._reboot_and_finish_deploy(task)
            get_pow_state_mock.assert_called_once_with(task)
            power_action_mock.assert_called_once_with(task, states.POWER_ON)
    def test__reboot_and_finish_deploy(self, get_pow_state_mock,
                                       power_action_mock):
        self.config(group='ansible',
                    post_deploy_get_power_state_retry_interval=0)

        with task_manager.acquire(self.context, self.node.uuid) as task:
            ansible_deploy._reboot_and_finish_deploy(task)
            get_pow_state_mock.assert_called_once_with(task)
            power_action_mock.assert_called_once_with(task, states.POWER_ON)
    def test__reboot_and_finish_deploy_retry(self, get_pow_state_mock,
                                             power_action_mock):
        self.config(group='ansible',
                    post_deploy_get_power_state_retry_interval=0)

        with task_manager.acquire(self.context, self.node.uuid) as task:
            ansible_deploy._reboot_and_finish_deploy(task)
            get_pow_state_mock.assert_called_with(task)
            self.assertEqual(
                CONF.ansible.post_deploy_get_power_state_retries + 1,
                len(get_pow_state_mock.mock_calls))
            expected_power_calls = [((task, states.POWER_OFF),),
                                    ((task, states.POWER_ON),)]
            self.assertEqual(expected_power_calls,
                             power_action_mock.call_args_list)
    def test__reboot_and_finish_deploy_retry(self, get_pow_state_mock,
                                             power_action_mock):
        self.config(group='ansible',
                    post_deploy_get_power_state_retry_interval=0)

        with task_manager.acquire(self.context, self.node.uuid) as task:
            ansible_deploy._reboot_and_finish_deploy(task)
            get_pow_state_mock.assert_called_with(task)
            self.assertEqual(
                CONF.ansible.post_deploy_get_power_state_retries + 1,
                len(get_pow_state_mock.mock_calls))
            expected_power_calls = [((task, states.POWER_OFF), ),
                                    ((task, states.POWER_ON), )]
            self.assertEqual(expected_power_calls,
                             power_action_mock.call_args_list)