Add test case for exaslct generate-alter-session

Author: tkilias

exaslct_src/test/resources/test-flavor/flavor_base/language_definition

@@ -1 +1 @@
-language_definition=PYTHON=localzmq+protobuf:///{{ bucketfs_name }}/{{ bucket_name }}/{{ path_in_bucket }}/{{ release_name }}?lang=python#buckets/{{ bucketfs_name }}/{{ bucket_name }}/{{ path_in_bucket }}/{{ release_name }}/exaudf/exaudfclient JAVA=builtin_java R=builtin_r
+PYTHON=localzmq+protobuf:///{{ bucketfs_name }}/{{ bucket_name }}/{{ path_in_bucket }}{{ release_name }}?lang=python#buckets/{{ bucketfs_name }}/{{ bucket_name }}/{{ path_in_bucket }}{{ release_name }}/exaudf/exaudfclient

exaslct_src/test/test_generate_alter_session.py

@@ -0,0 +1,27 @@
+import unittest
+
+import docker
+
+from exaslct_src.lib.utils.docker_utils import find_images_by_tag
+from exaslct_src.test import utils
+
+
+class GenerateAlterSessionTest(unittest.TestCase):
+
+    def setUp(self):
+        print(f"SetUp {self.__class__.__name__}")
+        self.test_environment = utils.ExaslctTestEnvironment(self)
+
+    def test_generate_alter_session_with_path_in_bucket(self):
+        command = f"./exaslct generate-alter-session --bucketfs-name bfsdefault --bucket-name default --path-in-bucket path --container-name container"
+        completed_process=self.test_environment.run_command(command,use_docker_repository=False, use_output_directory=False, capture_output=True)
+        self.assertIn("ALTER SESSION SET SCRIPT_LANGUAGES='PYTHON=localzmq+protobuf:///bfsdefault/default/path/container?lang=python#buckets/bfsdefault/default/path/container/exaudf/exaudfclient';",completed_process.stdout.decode("UTF-8"))
+
+
+    def test_generate_alter_session_without_path_in_bucket(self):
+        command = f"./exaslct generate-alter-session --bucketfs-name bfsdefault --bucket-name default --container-name container"
+        completed_process=self.test_environment.run_command(command,use_docker_repository=False, use_output_directory=False, capture_output=True)
+        self.assertIn("ALTER SESSION SET SCRIPT_LANGUAGES='PYTHON=localzmq+protobuf:///bfsdefault/default/container?lang=python#buckets/bfsdefault/default/container/exaudf/exaudfclient';",completed_process.stdout.decode("UTF-8"))
+
+if __name__ == '__main__':
+    unittest.main()

exaslct_src/test/utils.py

@@ -71,11 +71,14 @@ def clean_images(self):
         self.run_command(f"./exaslct clean-flavor-images", clean=True)
 
     def run_command(self, command: str,
+                    use_output_directory: bool = True,
                     use_flavor_path: bool = True,
                     use_docker_repository: bool = True,
                     track_task_dependencies: bool = False,
-                    clean: bool = False):
-        command = f"{command} {self.output_directory_arguments}"
+                    clean: bool = False,
+                    capture_output: bool = False):
+        if use_output_directory:
+            command = f"{command} {self.output_directory_arguments}"
         if track_task_dependencies:
             command = f"{command} {self.task_dependencies_argument}"
         if use_flavor_path:
@@ -86,8 +89,16 @@ def run_command(self, command: str,
             command = f"{command} {self.clean_docker_repository_arguments}"
         print()
         print(f"command: {command}")
-        completed_process = subprocess.run(shlex.split(command))
-        completed_process.check_returncode()
+        if capture_output:
+            completed_process = subprocess.run(shlex.split(command),stdout=subprocess.PIPE,stderr=subprocess.STDOUT)
+        else:
+            completed_process = subprocess.run(shlex.split(command))
+        try:
+            completed_process.check_returncode()
+        except subprocess.CalledProcessError as e:
+            print(e.stdout.decode("UTF-8"))
+            raise e
+        return completed_process
 
     def close(self):
         try:

google-cloud-build/ci-scripts/test_branch_exaslct.yaml

@@ -20,6 +20,20 @@ steps:
           'run_exaslct_test.sh',
           'exaslct_src/test/test_base_task.py'
         ]
+- id: 'Test exaslct test_hash_temp_dir.py'
+  name: 'gcr.io/script-languages/build-container'
+  args: [
+          'bash', 
+          'run_exaslct_test.sh',
+          'exaslct_src/test/test_hash_temp_dir.py'
+        ]
+- id: 'Test exaslct test_generate_alter_session.py'
+  name: 'gcr.io/script-languages/build-container'
+  args: [
+          'bash', 
+          'run_exaslct_test.sh',
+          'exaslct_src/test/test_generate_alter_session.py'
+        ]
 - id: 'Test exaslct test_docker_build.py'
   name: 'gcr.io/script-languages/build-container'
   args: [
@@ -55,13 +69,6 @@ steps:
           'run_exaslct_test.sh',
           'exaslct_src/test/test_docker_save.py'
         ]
-- id: 'Test exaslct test_hash_temp_dir.py'
-  name: 'gcr.io/script-languages/build-container'
-  args: [
-          'bash', 
-          'run_exaslct_test.sh',
-          'exaslct_src/test/test_hash_temp_dir.py'
-        ]
 - id: 'Test exaslct test_run_db_test_docker_db.py'
   name: 'gcr.io/script-languages/build-container'
   args: [