From 8fd9522f2de0362f7c19126658c6875c8d7564de Mon Sep 17 00:00:00 2001 From: Alexey Kudravtsev Date: Tue, 22 Apr 2025 12:13:03 +0200 Subject: [PATCH] utility method TestModeFlags.runWithFlag, reset flags at the test end GitOrigin-RevId: 0614d45dd075aa9f0edb7d436df4bfd42074ddba --- .../tests/targets/java/CommonJavaTargetTestBase.kt | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/platform/remote-servers/target-integration-tests/testSrc/com/intellij/tests/targets/java/CommonJavaTargetTestBase.kt b/platform/remote-servers/target-integration-tests/testSrc/com/intellij/tests/targets/java/CommonJavaTargetTestBase.kt index de8dbef6fb13..9943c50be886 100644 --- a/platform/remote-servers/target-integration-tests/testSrc/com/intellij/tests/targets/java/CommonJavaTargetTestBase.kt +++ b/platform/remote-servers/target-integration-tests/testSrc/com/intellij/tests/targets/java/CommonJavaTargetTestBase.kt @@ -304,14 +304,7 @@ abstract class CommonJavaTargetTestBase(protected val executionMode: ExecutionMo } private fun runWithConnectInUnitTestMode(block: () -> Unit) { - val connectInUnitTestModeValue = TestModeFlags.get(SearchForTestsTask.CONNECT_IN_UNIT_TEST_MODE_PROPERTY_KEY) - TestModeFlags.set(SearchForTestsTask.CONNECT_IN_UNIT_TEST_MODE_PROPERTY_KEY, true) - try { - block() - } - finally { - TestModeFlags.set(SearchForTestsTask.CONNECT_IN_UNIT_TEST_MODE_PROPERTY_KEY, connectInUnitTestModeValue) - } + TestModeFlags.runWithFlag(SearchForTestsTask.CONNECT_IN_UNIT_TEST_MODE_PROPERTY_KEY, true, block) } /**