update/software-client/software_client/tests/v1/test_deploy.py

# Copyright (c) 2019-2024 Wind River Systems, Inc.
# Copyright 2013 Hewlett-Packard Development Company, L.P.
#
#    Licensed under the Apache License, Version 2.0 (the "License"); you may
#    not use this file except in compliance with the License. You may obtain
#    a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#    License for the specific language governing permissions and limitations
#    under the License.

import testtools

from software_client.tests import utils
import software_client.v1.deploy
import software_client.v1.deploy_shell


HOST_LIST = {
    'data': [
        {
            'ip': '192.168.204.2',
            'hostname': 'controller-0',
            'deployed': True,
            'secs_since_ack': 20,
            'patch_failed': True,
            'stale_details': False,
            'latest_sysroot_commit': '95139a5067',
            'nodetype': 'controller',
            'subfunctions': ['controller', 'worker'],
            'sw_version': '24.03',
            'state': 'install-failed',
            'allow_insvc_patching': True,
            'interim_state': False,
            'reboot_required': False
        }
    ]
}


fixtures = {
    '/v1/deploy_host':
    {
        'GET': (
            {},
            HOST_LIST,
        ),
    },
    '/v1/deploy':
    {
        'GET': (
            {},
            {},
        ),
    },
    '/v1/deploy/precheck/1':
    {
        'GET': (
            {},
            {},
        ),
    },
    '/v1/deploy/1/precheck':
    {
        'POST': (
            {},
            {},
        ),
    },
    '/v1/deploy/1/start':
    {
        'POST': (
            {},
            {'force': 'true'},
        ),
    },
    '/v1/deploy_host/controller-1':
    {
        'POST': (
            {},
            {"error": True},
        ),
    },
    '/v1/deploy_host/controller-1/force':
    {
        'POST': (
            {},
            None,
        ),
    },
    '/v1/deploy/activate':
    {
        'POST': (
            {},
            {},
        ),
    },
    '/v1/deploy/complete':
    {
        'POST': (
            {},
            {},
        ),
    },
}


class Args:
    def __init__(self, **kwargs):
        for key, value in kwargs.items():
            if isinstance(value, dict):
                self.__dict__[key] = Args(**value)
            else:
                self.__dict__[key] = value


class DeployManagerTest(testtools.TestCase):

    def setUp(self):
        super(DeployManagerTest, self).setUp()
        self.api = utils.FakeAPI(fixtures)
        self.mgr = software_client.v1.deploy.DeployManager(self.api)

    def test_host_list(self):
        hosts = self.mgr.host_list()
        expect = [
            ('GET', '/v1/deploy_host', {}, None),
        ]
        self.assertEqual(self.api.calls, expect)
        self.assertEqual(len(hosts), 2)
        self.assertEqual(hosts[1]['data'][0]['hostname'],
                         HOST_LIST['data'][0]['hostname'])

    def test_show(self):
        self.mgr.show()
        expect = [
            ('GET', '/v1/deploy', {}, None),
        ]
        self.assertEqual(self.api.calls, expect)

    def test_precheck(self):
        input = {'deployment': '1', 'region_name': 'RegionOne', 'force': 1}
        args = Args(**input)
        check = self.mgr.precheck(args)
        expect = [
            ('POST', '/v1/deploy/1/precheck', {}, {'force': 'true', 'region_name': 'RegionOne'}),
        ]
        self.assertEqual(self.api.calls, expect)
        self.assertEqual(len(check), 2)

    def test_start(self):
        input = {'deployment': '1', 'force': 'True'}
        args = Args(**input)
        resp = self.mgr.start(args)
        expect = [
            ('POST', '/v1/deploy/1/start', {}, {'force': 'true'}),
        ]
        self.assertEqual(self.api.calls, expect)
        self.assertEqual(len(resp), 2)

    def test_host(self):
        input = {'host': 'controller-1', 'force': False}
        args = Args(**input)
        self.mgr.host(args)
        expect = [
            ('POST', '/v1/deploy_host/controller-1', {}, None),
        ]
        self.assertEqual(self.api.calls, expect)

    def test_activate(self):
        input = {}
        args = Args(**input)
        resp = self.mgr.activate(args)
        expect = [
            ('POST', '/v1/deploy/activate', {}, {}),
        ]
        self.assertEqual(self.api.calls, expect)
        self.assertEqual(len(resp), 2)

    def test_complete(self):
        input = {}
        args = Args(**input)
        resp = self.mgr.complete(args)
        expect = [
            ('POST', '/v1/deploy/complete', {}, {}),
        ]
        self.assertEqual(self.api.calls, expect)
        self.assertEqual(len(resp), 2)