50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
#
|
|
# Copyright (c) 2021, 2024 Wind River Systems, Inc.
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
|
|
from dcmanager.common import consts
|
|
from dcmanager.tests.unit.common import fake_strategy
|
|
from dcmanager.tests.unit.orchestrator.states.kube_rootca.test_base \
|
|
import TestKubeRootCaUpgradeState
|
|
from dcmanager.tests.unit.orchestrator.states.test_creating_vim_strategy \
|
|
import CreatingVIMStrategyStageMixin
|
|
|
|
|
|
class TestCreatingVIMKubeRootCAUpgradeStrategyStage(
|
|
CreatingVIMStrategyStageMixin, TestKubeRootCaUpgradeState
|
|
):
|
|
"""Test create vim kube rootca upgrade strategy during kube rootca orch"""
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
|
|
self.set_state(
|
|
consts.STRATEGY_STATE_CREATING_VIM_KUBE_ROOTCA_UPDATE_STRATEGY,
|
|
consts.STRATEGY_STATE_APPLYING_VIM_KUBE_ROOTCA_UPDATE_STRATEGY
|
|
)
|
|
|
|
def test_create_strategy_succeeds_with_extra_args(self):
|
|
"""Test create strategy succeeds with extra_args"""
|
|
|
|
# Create a strategy with extra_args
|
|
extra_args = {
|
|
"expiry-date": "2020:01:31",
|
|
"subject": "C=CA ST=ON L=OTT O=WR OU=STX CN=AL_RULES"
|
|
}
|
|
self.strategy = fake_strategy.create_fake_strategy(
|
|
self.ctx, self.DEFAULT_STRATEGY_TYPE, extra_args=extra_args
|
|
)
|
|
# Call the 'success' test
|
|
self.test_creating_vim_strategy_success()
|
|
|
|
def test_create_strategy_succeeds_without_extra_argst(self):
|
|
"""Test create strategy succeeds without extra_args"""
|
|
|
|
self.strategy = fake_strategy.create_fake_strategy(
|
|
self.ctx, self.DEFAULT_STRATEGY_TYPE, extra_args=None
|
|
)
|
|
|
|
self.test_creating_vim_strategy_success()
|