From 2cc154e09b724135148cc75b0a2e7d076208776a Mon Sep 17 00:00:00 2001
From: Alexey Katsman <alexey.katsman@jetbrains.com>
Date: Tue, 9 Sep 2025 19:49:24 +0200
Subject: [PATCH] Fix some of the PyBehaveEnvTest tests

GitOrigin-RevId: 2adbf7ee64e690b6cf6bb4763c92fc025203a9aa
---
 python/helpers/pycharm/behave_runner.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/python/helpers/pycharm/behave_runner.py b/python/helpers/pycharm/behave_runner.py
index 293005332e3d..2edaf6515a95 100644
--- a/python/helpers/pycharm/behave_runner.py
+++ b/python/helpers/pycharm/behave_runner.py
@@ -17,6 +17,7 @@ from behave.formatter.base import Formatter
 from behave.model import Step, ScenarioOutline, Feature, Scenario
 from behave.formatter import _registry
 from behave.tag_expression import TagExpression
+from behave.step_registry import registry as the_step_registry
 from _jb_django_behave import run_as_django_behave
 import _bdd_utils
 import tcmessages
@@ -120,6 +121,10 @@ class _RunnerWrapper(runner.Runner):
         self.dry_run = False
         self.hooks.clear()
         self.features = []
+        if self.step_registry is None:
+            the_step_registry.clear()
+        else:
+            self.step_registry.clear()
 
 
 class _BehaveRunner(_bdd_utils.BddRunner):