CloudShell-Terraform-Shell/package/tests/integration_tests/helper_objects/integration_context.py at 3b912a7b5c9cfedc6d938a8c125c057fcf1979c5 · QualiSystemsLab/CloudShell-Terraform-Shell · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
from unittest import mock
from unittest.mock import Mock

from cloudshell.api.cloudshell_api import CloudShellAPISession
from cloudshell.iac.terraform import TerraformShell, TerraformShellConfig
from cloudshell.logging.qs_logger import get_qs_logger
from cloudshell.shell.core.driver_context import ResourceCommandContext

# from tests.integration_tests.helper_objects.env_vars import EnvVars
from package.tests.integration_tests.helper_objects.env_vars import EnvVars
# from tests.integration_tests.helper_services.service_attributes_factory import ServiceAttributesMockBuilder
from package.tests.integration_tests.helper_services.service_attributes_factory import ServiceAttributesMockBuilder

from abc import ABCMeta, abstractmethod


class BaseIntegrationData(metaclass=ABCMeta):
    def __init__(self, service_name: str):
        self._sb_service_alias = service_name
        self._logger = get_qs_logger(log_group=service_name)

        self._build_context()
        self._set_general_context_resource_data()
        self._set_context()
        self.create_tf_shell()

    @abstractmethod
    def _set_context(self):
        raise NotImplementedError()

    def _build_context(self):
        self.context = mock.create_autospec(ResourceCommandContext)
        self.context.connectivity = Mock()

    def _set_general_context_resource_data(self):
        self.context.resource = Mock()
        self.context.resource.attributes = dict()
        self.context.resource.name = self._sb_service_alias
        self.context.resource.model = 'Generic Terraform Service'

    def create_tf_shell(self):
        _config = TerraformShellConfig(write_sandbox_messages=True, update_live_status=True)
        self.tf_shell = TerraformShell(self.context, self._logger, _config)


class RealAPIIntegrationData(BaseIntegrationData):
    def __init__(self, service_name: str):
        self._env_vars = EnvVars()
        self.api = CloudShellAPISession(
            self._env_vars.cs_server,
            self._env_vars.cs_user,
            self._env_vars.cs_pass,
            self._env_vars.cs_domain
        )
        BaseIntegrationData.__init__(self, service_name)

    def _set_context(self):
        self.context.connectivity.server_address = self._env_vars.cs_server
        self.context.connectivity.admin_auth_token = self.api.authentication.xmlrpc_token
        self.set_context_resource_attributes_from_cs()
        self.context.reservation = Mock()
        self.context.reservation.reservation_id = self._env_vars.cs_res_id
        self.context.reservation.domain = self._env_vars.cs_domain

    def set_context_resource_attributes_from_cs(self, the_only_attribute_to_update: str = ""):
        services = self.api.GetReservationDetails(self._env_vars.cs_res_id, disableCache=True) \
            .ReservationDescription.Services
        for service in services:
            if service.Alias == self._sb_service_alias:
                for attribute in service.Attributes:
                    if the_only_attribute_to_update and attribute.Name == the_only_attribute_to_update:
                        self.context.resource.attributes[attribute.Name] = attribute.Value
                        return
                    elif not the_only_attribute_to_update:
                        self.context.resource.attributes[attribute.Name] = attribute.Value


class MockAPIIntegrationData(BaseIntegrationData):
    def __init__(self, service_name: str, mock_api: Mock):
        self.api = mock_api
        self.api.authentication.xmlrpc_token = Mock()

        BaseIntegrationData.__init__(self, service_name)

    def _set_context(self):
        self.context.resource.attributes = ServiceAttributesMockBuilder.create_empty_attributes()
        self.context.reservation = Mock()
        self.context.reservation.reservation_id = "mock_reservation_id"