Create kesuses for rate limiting in YQ

Fill config in kikimr runner

Create coordination nodes

Pass to config

Protos

1 files changed, 20 insertions, 12 deletions

diff --git a/library/python/testing/yatest_common/yatest/common/runtime.py b/library/python/testing/yatest_common/yatest/common/runtime.py
index 6e6c3c8759..b8d5964d1d 100644
--- a/library/python/testing/yatest_common/yatest/common/runtime.py
+++ b/library/python/testing/yatest_common/yatest/common/runtime.py
@@ -9,22 +9,30 @@ import six
 
 _lock = threading.Lock()
 
+_config = None
+
+
+def _set_ya_config(config=None, ya=None):
+    global _config
+    if config:
+        _config = config
+    elif ya:
+        class Config:
+            def __init__(self):
+                self.ya = None
+        _config = Config()
+        _config.ya = ya
+
 
 def _get_ya_config():
-    try:
-        import library.python.pytest.plugins.ya as ya_plugin
-        if ya_plugin.pytest_config is not None:
-            return ya_plugin.pytest_config
-        import pytest
-        return pytest.config
-    except (ImportError, AttributeError):
+    if _config:
+        return _config
+    else:
         try:
-            import library.python.testing.recipe
-            if library.python.testing.recipe.ya:
-                return library.python.testing.recipe
+            import pytest
+            return pytest.config
         except (ImportError, AttributeError):
-            pass
-        raise NotImplementedError("yatest.common.* is only available from the testing runtime")
+            raise NotImplementedError("yatest.common.* is only available from the testing runtime")
 
 
 def _get_ya_plugin_instance():