Added implementation + integration tests for #25

Signed-off-by: Steffen Pankratz <steffen.pankratz@exasol.com>

Author: kratz00

exasol/pytest_slc/__init__.py

@@ -1,18 +1,23 @@
 from __future__ import annotations
 
+import contextlib
 import getpass
 from importlib.metadata import version
 from pathlib import Path
+from typing import Any
 
 import exasol.bucketfs as bfs
+import pyexasol
 import pytest
+from _pytest.fixtures import FixtureRequest
 from exasol.pytest_backend import paralleltask
 from exasol.python_extension_common.deployment.language_container_builder import (
     LanguageContainerBuilder,
 )
 from exasol.python_extension_common.deployment.language_container_deployer import (
     LanguageActivationLevel,
     LanguageContainerDeployer,
+    get_language_settings,
 )
 from exasol.slc.models.export_container_result import (  # noqa: F401
     ExportContainerResult,
@@ -24,6 +29,9 @@
 
 __version__ = version("pytest-exasol-slc")
 
+from pyexasol import ExaConnection
+
+SCRIPT_LANGUAGES_OPTION = "--script-languages"
 SKIP_SLC_OPTION = "--skip-slc"
 BFS_CONTAINER_DIRECTORY = "container"
 
@@ -32,6 +40,11 @@ def pytest_addoption(parser):
     parser.addoption(
         SKIP_SLC_OPTION, action="store_true", default=False, help="Skip SLC deployment"
     )
+    parser.addoption(
+        SCRIPT_LANGUAGES_OPTION,
+        default=None,
+        help="Script language definition",
+    )
 
 
 @pytest.fixture(scope="session")
@@ -62,7 +75,16 @@ def export_slc_async(
     """
     The fixture starts the export() function of the provided
     LanguageContainerBuilder object as an asynchronous task.
+    """
+
+    """
+    The operation will be skipped if SCRIPT_LANGUAGES_OPTION is defined.
+    """
+    if request.config.getoption(SCRIPT_LANGUAGES_OPTION):
+        yield None
+        return
 
+    """
     The operation will be skipped if none of the backends is in use or the
     container builder is not defined or the SLC deployment is skipped.
     """
@@ -173,3 +195,50 @@ def deployed_slc(deploy_slc, language_alias) -> str:
     """
     deploy_slc(language_alias)
     return language_alias
+
+
+def set_script_languages(
+    pyexasol_connection: ExaConnection, script_languages: Any | None
+):
+    query = "ALTER SESSION SET SCRIPT_LANGUAGES={script_languages}"
+    pyexasol_connection.execute(
+        query, query_params={"script_languages": script_languages}
+    )
+
+
+@pytest.fixture(scope="session")
+def script_languages(request: FixtureRequest) -> str:
+    script_languages = request.config.getoption(SCRIPT_LANGUAGES_OPTION)
+    if not script_languages or script_languages == "":
+        raise RuntimeError(f"Value for {SCRIPT_LANGUAGES_OPTION} missing")
+    return script_languages
+
+
+@contextlib.contextmanager
+def activate_script_languages(
+    pyexasol_connection_: ExaConnection, script_languages_: str
+):
+    # get script languages currently used
+    current_script_languages = get_language_settings(
+        pyexasol_connection_, LanguageActivationLevel.Session
+    )
+    set_script_languages(pyexasol_connection_, script_languages_)
+    yield
+    # reset script languages
+    set_script_languages(pyexasol_connection_, current_script_languages)
+
+
+@pytest.fixture(scope="session")
+def activate_script_languages_for_session(
+    script_languages, pyexasol_connection: pyexasol.ExaConnection
+):
+    with activate_script_languages(pyexasol_connection, script_languages):
+        yield
+
+
+@pytest.fixture(scope="module")
+def activate_script_languages_for_module(
+    script_languages, pyexasol_connection: pyexasol.ExaConnection
+):
+    with activate_script_languages(pyexasol_connection, script_languages):
+        yield

test/integration/script_languages_test.py

@@ -0,0 +1,48 @@
+import pytest
+from exasol.pytest_backend import (
+    BACKEND_ONPREM,
+    BACKEND_OPTION,
+)
+
+from exasol.pytest_slc import SCRIPT_LANGUAGES_OPTION
+
+pytest_plugins = ["pytester"]
+
+
+def test_empty_value_for_script_languages_raises_wrapper(pytester):
+    pytester.makepyfile("""
+import pytest
+
+def test_empty_value_for_script_languages_raises(script_languages):
+    assert True
+""")
+    result = pytester.runpytest(
+        BACKEND_OPTION, BACKEND_ONPREM, SCRIPT_LANGUAGES_OPTION, ""
+    )
+    assert result.ret == pytest.ExitCode.TESTS_FAILED
+    result.assert_outcomes(errors=1)
+
+
+def test_setting_script_languages_wrapper(pytester):
+    expected_script_languages = r"PYTHON3=PYTEST_SCL_INTEGRATION_TEST"
+    pytester.makepyfile(f"""
+    import pytest
+    from exasol.python_extension_common.deployment.language_container_deployer import (
+        LanguageActivationLevel,
+        get_language_settings,
+    )
+
+    def test_empty_value_for_script_languages_raises(activate_script_languages_for_module, pyexasol_connection):
+        current_script_languages = get_language_settings(
+            pyexasol_connection, LanguageActivationLevel.Session
+        )
+        assert current_script_languages == "{expected_script_languages}"
+    """)
+    result = pytester.runpytest(
+        BACKEND_OPTION,
+        BACKEND_ONPREM,
+        SCRIPT_LANGUAGES_OPTION,
+        expected_script_languages,
+    )
+    assert result.ret == pytest.ExitCode.OK
+    result.assert_outcomes(passed=1, skipped=1)