Added unit tests for extractDomainResource

Author: rakillen

core/src/test/python/base_test.py

@@ -70,6 +70,9 @@ def _match(self, value, dictionary, *args):
             key = '/'.join(list(args))
             self.fail(key + ' equals ' + str(dictionary_value) + ', should equal ' + str(value))
 
+    def _match_values(self, description, actual, expected):
+        self.assertEqual(actual, expected, description + " equals " + str(actual) + ", should equal " + str(expected))
+
     def _no_dictionary_key(self, dictionary, key):
         if key in dictionary:
             self.fail('Dictionary should not contain ' + key)

core/src/test/python/wlsdeploy/tool/extract/extract_test.py

@@ -0,0 +1,85 @@
+"""
+Copyright (c) 2021, Oracle and/or its affiliates.
+Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl.
+"""
+import os
+
+from base_test import BaseTestCase
+from wlsdeploy.logging.platform_logger import PlatformLogger
+from wlsdeploy.tool.extract.domain_resource_extractor import DomainResourceExtractor
+from wlsdeploy.util.model import Model
+from wlsdeploy.util.model_context import ModelContext
+from wlsdeploy.util.model_translator import FileToPython
+
+
+class ExtractTest(BaseTestCase):
+    __logger = PlatformLogger('wlsdeploy.extract')
+
+    def __init__(self, *args):
+        BaseTestCase.__init__(self, *args)
+        self.MODELS_DIR = os.path.join(self.TEST_CLASSES_DIR, 'extract')
+        self.EXTRACT_OUTPUT_DIR = os.path.join(self.TEST_OUTPUT_DIR, 'extract')
+
+    def setUp(self):
+        BaseTestCase.setUp(self)
+        self._establish_directory(self.EXTRACT_OUTPUT_DIR)
+
+    def testDefaultModel(self):
+        """
+        Test that default values and information from the model
+        are incorporated into the resulting domain resource file.
+        """
+        resource = self._extract_domain_resource('1')
+
+        # clusters from topology should be in the domain resource file
+        cluster_list = self._traverse(resource, 'spec', 'clusters')
+        self._match_values("Cluster count", len(cluster_list), 2)
+        self._match_values("Cluster 0 clusterName", cluster_list[0]['clusterName'], 'cluster1')
+        self._match_values("Cluster 0 replicas", cluster_list[0]['replicas'], 3)
+        self._match_values("Cluster 1 clusterName", cluster_list[1]['clusterName'], 'cluster2')
+        self._match_values("Cluster 1 replicas", cluster_list[1]['replicas'], 12)
+
+        # secrets from the model should be in the domain resource file
+        secret_list = self._traverse(resource, 'spec', 'configuration', 'secrets')
+        self._match_values("Secret count", len(secret_list), 2)
+        self._match_values("Secret 0", secret_list[0], 'demodomain-m1-system')
+        self._match_values("Secret 1", secret_list[1], 'demodomain-m2-system')
+
+    def testKubernetesModel(self):
+        """
+        Test that fields from the kubernetes section of the model
+        are transferred to the resulting domain resource file
+        """
+        resource = self._extract_domain_resource('2')
+
+        # clusters from kubernetes section should be in the domain resource file
+        cluster_list = self._traverse(resource, 'spec', 'clusters')
+        self._match_values("Cluster count", len(cluster_list), 2)
+        self._match_values("Cluster 0 clusterName", cluster_list[0]['clusterName'], 'CLUSTER_1')
+        self._match_values("Cluster 1 clusterName", cluster_list[1]['clusterName'], 'CLUSTER_2')
+
+        # secrets from the model should be in the domain resource file
+        secret_list = self._traverse(resource, 'spec', 'configuration', 'secrets')
+        self._match_values("Secret count", len(secret_list), 2)
+        self._match_values("Secret 0", secret_list[0], 'secret-1')
+        self._match_values("Secret 1", secret_list[1], 'secret-2')
+
+    def _extract_domain_resource(self, suffix):
+        model_file = os.path.join(self.MODELS_DIR, 'model-' + suffix + '.yaml')
+        translator = FileToPython(model_file, use_ordering=True)
+        model_dict = translator.parse()
+        model = Model(model_dict)
+
+        resource_file = os.path.join(self.EXTRACT_OUTPUT_DIR, 'domain-resource-' + suffix + '.yaml')
+        args_map = {
+            '-domain_home': '/u01/domain',
+            '-oracle_home': '/oracle',
+            '-domain_resource_file': resource_file
+        }
+        model_context = ModelContext('ExtractTest', args_map)
+
+        extractor = DomainResourceExtractor(model, model_context, self.__logger)
+        extractor.extract()
+
+        translator = FileToPython(resource_file, use_ordering=True)
+        return translator.parse()

core/src/test/resources/extract/model-1.yaml

@@ -0,0 +1,27 @@
+# Copyright (c) 2021, Oracle Corporation and/or its affiliates.
+# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl.
+
+# Test model for extractDomainResource.
+# This will test that default values and information from the model
+# are incorporated into the resulting domain resource file.
+
+topology:
+    Name: DemoDomain
+    Cluster:
+        cluster1:
+            # should get a replicas value of 3, since 3 servers are assigned
+        cluster2:
+            # should get a replicas value of 12, based on dynamic cluster size
+            DynamicServers:
+                DynamicClusterSize: 12
+    Server:
+        AdminServer:
+            Cluster: cluster1
+        m1:
+            Cluster: cluster1
+            # secrets should be listed in domain resource
+            SystemPasswordEncrypted: '@@SECRET:@@ENV:DOMAIN_UID@@-m1-system:password@@'
+        m2:
+            Cluster: cluster1
+            # secrets should be listed in domain resource
+            SystemPasswordEncrypted: '@@SECRET:@@ENV:DOMAIN_UID@@-m2-system:password@@'

core/src/test/resources/extract/model-2.yaml

@@ -0,0 +1,77 @@
+# Copyright (c) 2021, Oracle Corporation and/or its affiliates.
+# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl.
+
+# Test model for extractDomainResource.
+# This will test that fields from the kubernetes section of the model
+# are transferred to the resulting domain resource file
+
+kubernetes:
+    apiVersion: weblogic.oracle/v6
+    kind: Domain
+    metadata:
+        name: DOMAIN_UID
+        namespace: NAMESPACE
+        labels:
+            weblogic.resourceVersion: domain-v2
+            weblogic.domainUID: DOMAIN_UID
+    spec:
+        domainHome: DOMAIN_HOME
+        domainHomeInImage: true
+        image: phx.ocir.io/weblogick8s/onprem-domain-image:2.0
+        imagePullPolicy: Always
+        imagePullSecrets:
+            - name: WEBLOGIC_IMAGE_PULL_SECRET_NAME
+        webLogicCredentialsSecret:
+            name: WEBLOGIC_CREDENTIALS_SECRET_NAME
+        includeServerOutInPodLog: 1
+        logHomeEnabled: 0
+        logHome: /scratch/logs
+        dataHome: /data
+        serverStartPolicy: IF_NEEDED
+        serverPod:
+            env:
+                - name: JAVA_OPTIONS
+                  value: -good_java
+                - name: USER_MEM_ARGS
+                  value: '-XX:+UseContainerSupport -Djava.security.egd=file:/dev/./urandom '
+            volumes:
+                - name: weblogic-domain-storage-volume
+                  persistentVolumeClaim:
+                      claimName: DOMAIN_PVC_NAME
+            volumeMounts:
+                - name: weblogic-domain-storage-volume
+                  mountPath: DOMAIN_ROOT_DIR
+        adminServer:
+            serverStartState: RUNNING
+            serverStartPolicy: ADMIN_ONLY
+            restartVersion: xyz
+            adminService:
+                annotations:
+                    abc: xyz
+                    lmn: opq
+                labels:
+                    abc: xyz
+                    lmn: opq
+                channels:
+                    - channelName: default
+                      nodePort: 5001
+                    - channelName: T3Channel
+            serverPod:
+                env:
+                    - name: JAVA_OPTIONS
+                      value: -im_admin_pod
+                    - name: USER_MEM_ARGS
+                      value: '-XX:+UseContainerSupport -Djava.security.egd=file:/dev/./urandom '
+        clusters:
+            - clusterName: CLUSTER_1
+              serverStartState: RUNNING
+              replicas: 1
+            - clusterName: CLUSTER_2
+              serverStartState: RUNNING
+              replicas: 4
+        configuration:
+            model:
+                domainType: WLS
+            secrets:
+                - secret-1
+                - secret-2