CloudShell-Terraform-Shell/package/tests/integration_tests/helper_objects/mock_tests_data.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
import os
from unittest.mock import Mock

from package.tests.integration_tests.helper_objects.integration_context import MockAPIIntegrationData
from package.tests.integration_tests.helper_services.service_attributes_factory import ServiceAttributesMockBuilder

from cloudshell.api.cloudshell_api import NameValuePair


class MockTestsData(object):
    def __init__(self):
        self.mock_api = Mock()
        self.service1 = Mock()
        self.service2 = Mock()
        self._mocked_tf_working_dir = ''
        self.service1.Alias = ''
        self.service1.Attributes = None
        self.service2.Alias = ''
        self.service2.Attributes = None

        self._prepare_mock_services()
        self._prepare_mock_api()

    def prepare_integration_data(self):
        self.integration_data1 = MockAPIIntegrationData(self.service1.Alias, self.mock_api)
        self.integration_data2 = MockAPIIntegrationData(self.service2.Alias, self.mock_api)

    def _prepare_mock_services(self):
        self.service1 = Mock()
        self.service1.Alias = "MOCK_ALIAS1"
        self.service1.Attributes = ServiceAttributesMockBuilder.create_empty_attributes()
        self.service2 = Mock()
        self.service2.Alias = "MOCK_ALIAS2"
        self.service2.Attributes = ServiceAttributesMockBuilder.create_empty_attributes()

    def _prepare_mock_api(self):
        self.mock_api = Mock()
        self.mock_api.DecryptPassword = _decrypt_password
        self.mock_api.GetReservationDetails.return_value.ReservationDescription.Services = [self.service1,
                                                                                            self.service2]

        # self.mock_api.SetServiceAttributesValues = self._set_service_attributes_values

        self.mock_api.GetResourceDetails.return_value.ResourceFamilyName = 'Cloud Provider'
        self.mock_api.GetResourceDetails.return_value.ResourceModelName = 'Microsoft Azure'
        self.mock_api.GetResourceDetails.return_value.ResourceAttributes = [
            NameValuePair(Name="Azure Subscription ID", Value=os.environ.get("AZURE_SUBSCRIPTION_ID")),
            NameValuePair(Name="Azure Tenant ID", Value=os.environ.get("AZURE_TENANT_ID")),
            NameValuePair(Name="Azure Application ID", Value=os.environ.get("AZURE_APPLICATION_ID")),
            NameValuePair(Name="Azure Application Key", Value=os.environ.get("AZURE_APPLICATION_KEY_DEC"))
        ]

    def _get_mocked_tf_working_dir(self):
        return self._mocked_tf_working_dir

    def _set_mocked_tf_working_dir(self, tf_working_dir: str):
        self._mocked_tf_working_dir = tf_working_dir


def _decrypt_password(x):
    result = Mock()
    result.Value = x
    return result