Intermediate changes

commit_hash:71679cd60917305e9a5bf680434d352fcb48e90a

1 files changed, 20 insertions, 0 deletions

diff --git a/yt/yt/core/test_framework/framework.h b/yt/yt/core/test_framework/framework.h
index 9a01a6deae5..01ba4d37ae5 100644
--- a/yt/yt/core/test_framework/framework.h
+++ b/yt/yt/core/test_framework/framework.h
@@ -46,6 +46,26 @@ TString GenerateRandomFileName(const char* prefix);
 
 ////////////////////////////////////////////////////////////////////////////////
 
+#define _TEST_PI(TBaseClass, TestName, ParamValues) \
+    class TBaseClass##_##TestName \
+        : public TBaseClass \
+        , public ::testing::WithParamInterface<decltype(ParamValues)::iterator::value_type> \
+    {}; \
+    INSTANTIATE_TEST_SUITE_P(, TBaseClass##_##TestName, ParamValues); \
+    TEST_P(TBaseClass##_##TestName, Test)
+
+#define _TEST_PI_MANUAL(TBaseClass, TestName, ParamValues, ...) \
+    class TBaseClass##_##TestName \
+        : public TBaseClass \
+        , public ::testing::WithParamInterface<__VA_ARGS__> \
+    {}; \
+    INSTANTIATE_TEST_SUITE_P(, TBaseClass##_##TestName, ParamValues); \
+    TEST_P(TBaseClass##_##TestName, Test)
+
+#define TEST_PI(TBaseClass, TestName, ParamValues, ...) _TEST_PI##__VA_OPT__(_MANUAL) (TBaseClass, TestName, ParamValues __VA_OPT__(,) __VA_ARGS__)
+
+////////////////////////////////////////////////////////////////////////////////
+
 struct TWaitForPredicateOptions
 {
     int IterationCount = 300;