FIx strict context tests

2 files changed, 7 insertions, 0 deletions

diff --git a/library/python/testing/yatest_common/yatest/common/runtime.py b/library/python/testing/yatest_common/yatest/common/runtime.py
index 75c932743f..8ab51f3de7 100644
--- a/library/python/testing/yatest_common/yatest/common/runtime.py
+++ b/library/python/testing/yatest_common/yatest/common/runtime.py
@@ -424,6 +424,11 @@ class Context(object):
         return _get_ya_plugin_instance().get_context("test_tool_path")
 
     @property
+    @default_value(None)
+    def retry_index(self):
+        return _get_ya_plugin_instance().get_context("retry_index")
+
+    @property
     @default_value(False)
     def sanitize(self):
         """
diff --git a/library/python/testing/yatest_lib/ya.py b/library/python/testing/yatest_lib/ya.py
index ab1a5366f2..27ea2e89b2 100644
--- a/library/python/testing/yatest_lib/ya.py
+++ b/library/python/testing/yatest_lib/ya.py
@@ -94,6 +94,8 @@ class Ya(object):
         self._context["work_path"] = to_str(context_runtime.get("work_path"))
         self._context["test_tool_path"] = context_runtime.get("test_tool_path")
         self._context["test_output_ram_drive_path"] = to_str(context_runtime.get("test_output_ram_drive_path"))
+        self._context["retry_index"] = context_runtime.get("retry_index")
+
         self._context["ya_global_resources"] = context_resources.get("global")
 
         self._context["sanitize"] = context_build.get("sanitizer")