43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
#
|
|
# Copyright (c) 2020 Wind River Systems, Inc.
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
|
|
from dcmanager.common import consts
|
|
from dcmanager.db.sqlalchemy import api as db_api
|
|
|
|
|
|
def create_fake_strategy(
|
|
ctxt,
|
|
strategy_type,
|
|
subcloud_apply_type=consts.SUBCLOUD_APPLY_TYPE_PARALLEL,
|
|
state=consts.SW_UPDATE_STATE_INITIAL,
|
|
max_parallel_subclouds=2,
|
|
stop_on_failure=True,
|
|
extra_args=None):
|
|
values = {
|
|
"type": strategy_type,
|
|
"subcloud_apply_type": subcloud_apply_type,
|
|
"max_parallel_subclouds": max_parallel_subclouds,
|
|
"stop_on_failure": stop_on_failure,
|
|
"state": state,
|
|
"extra_args": extra_args
|
|
}
|
|
return db_api.sw_update_strategy_create(ctxt, **values)
|
|
|
|
|
|
def create_fake_strategy_step(
|
|
ctxt,
|
|
state=consts.STRATEGY_STATE_INITIAL,
|
|
subcloud_id=1,
|
|
stage=1,
|
|
details="Dummy details"):
|
|
values = {
|
|
"subcloud_id": subcloud_id,
|
|
"stage": stage,
|
|
"state": state,
|
|
"details": details,
|
|
}
|
|
return db_api.strategy_step_create(ctxt, **values)
|