def test_get_flow_volume_flow_tasks(self):
"""Test that all expected parameter names exist for added tasks."""
mock_taskflow_engine = self.mock_object(taskflow.engines, 'load')
mock_taskflow_engine.side_effect = self._verify_volume_flow_tasks
manager.get_flow(mock.sentinel.context, mock.sentinel.db,
mock.sentinel.driver, mock.sentinel.host,
mock.sentinel.volume, mock.sentinel.ref)
def test_get_flow_volume_flow_tasks(self):
"""Test that all expected parameter names exist for added tasks."""
mock_taskflow_engine = self.mock_object(taskflow.engines, 'load')
mock_taskflow_engine.side_effect = self._verify_volume_flow_tasks
manager.get_flow(
mock.sentinel.context, mock.sentinel.db, mock.sentinel.driver,
mock.sentinel.host, mock.sentinel.volume, mock.sentinel.ref)
def test_get_flow(self):
mock_volume_flow = mock.Mock()
mock_linear_flow = self.mock_object(manager.linear_flow, 'Flow')
mock_linear_flow.return_value = mock_volume_flow
mock_taskflow_engine = self.mock_object(taskflow.engines, 'load')
expected_store = {
'context': mock.sentinel.context,
'volume': mock.sentinel.volume,
'manage_existing_ref': mock.sentinel.ref,
'optional_args': {'is_quota_committed': False},
}
manager.get_flow(
mock.sentinel.context, mock.sentinel.db, mock.sentinel.driver,
mock.sentinel.host, mock.sentinel.volume, mock.sentinel.ref)
mock_linear_flow.assert_called_once_with(
'volume_manage_existing_manager')
mock_taskflow_engine.assert_called_once_with(
mock_volume_flow, store=expected_store)
def test_get_flow(self):
mock_volume_flow = mock.Mock()
mock_linear_flow = self.mock_object(manager.linear_flow, 'Flow')
mock_linear_flow.return_value = mock_volume_flow
mock_taskflow_engine = self.mock_object(taskflow.engines, 'load')
expected_store = {
'context': mock.sentinel.context,
'volume': mock.sentinel.volume,
'manage_existing_ref': mock.sentinel.ref,
'group_snapshot': None,
'optional_args': {'is_quota_committed': False},
}
manager.get_flow(
mock.sentinel.context, mock.sentinel.db, mock.sentinel.driver,
mock.sentinel.host, mock.sentinel.volume, mock.sentinel.ref)
mock_linear_flow.assert_called_once_with(
'volume_manage_existing_manager')
mock_taskflow_engine.assert_called_once_with(
mock_volume_flow, store=expected_store)
