disable runtime only for omnipy tests, not all tests

Author: sveinugu

src/omnipy/hub/runtime.py

@@ -1,5 +1,4 @@
 from dataclasses import dataclass, field
-import sys
 from typing import Any
 
 from omnipy.api.enums import EngineChoice
@@ -24,6 +23,7 @@
 from omnipy.hub.root_log import RootLogConfigEntryPublisher, RootLogObjects
 from omnipy.log.registry import RunStateRegistry
 from omnipy.modules.prefect.engine.prefect import PrefectEngine
+from omnipy.util.helpers import called_from_omnipy_tests
 
 
 def _job_creator_factory():
@@ -116,16 +116,4 @@ def _update_job_creator_engine(self, _item_changed: Any):
         self.objects.job_creator.set_engine(self._get_engine(self.config.engine))
 
 
-# TODO: The check disabling runtime for tests also trigger for tests that are run outside of Omnipy,
-#  breaking tests on the user side.
-#  Find a better way to disable the global runtime object for Omnipy tests
-
-
-def _get_runtime() -> 'Runtime | None':
-    if 'pytest' not in sys.modules:
-        return Runtime()
-    else:
-        return None
-
-
-runtime: 'Runtime | None' = _get_runtime()
+runtime: 'Runtime | None' = None if called_from_omnipy_tests() else Runtime()

src/omnipy/util/helpers.py

@@ -249,3 +249,16 @@ def get_calling_module_name() -> str | None:
         if module is not None:
             return module.__name__
         start_frame_index += 1
+
+
+def called_from_omnipy_tests() -> bool:
+    stack = inspect.stack()
+    for index in range(len(stack)):
+        frame = stack[index][0]
+        module = inspect.getmodule(frame)
+        if module is not None \
+                and module.__name__.startswith('tests') \
+                and module.__file__ is not None \
+                and 'omnipy/tests' in module.__file__:
+            return True
+    return False

tests/util/test_helpers.py

@@ -8,7 +8,8 @@
 from typing_inspect import get_generic_type
 
 from omnipy import Dataset, Model
-from omnipy.util.helpers import (ensure_non_str_byte_iterable,
+from omnipy.util.helpers import (called_from_omnipy_tests,
+                                 ensure_non_str_byte_iterable,
                                  ensure_plain_type,
                                  get_calling_module_name,
                                  get_first_item,
@@ -414,3 +415,7 @@ def local_call_get_calling_module_name() -> str:
     assert other_module_call_get_calling_module_name() == 'tests.util.test_helpers'
     assert calling_module_name_when_importing_other_module == 'tests.util.test_helpers'
 
+
+def test_called_from_omnipy_tests() -> None:
+    # Negative test is left as an exercise to the reader
+    assert called_from_omnipy_tests()