Test for scale factor in tpch queries (#10990)

Author: iddqdex

ydb/library/workload/tpc_base/tpc_base.cpp

@@ -96,14 +96,15 @@ void TTpcBaseWorkloadGenerator::PatchQuery(TString& query) const {
 }
 
 void TTpcBaseWorkloadGenerator::FilterHeader(IOutputStream& result, TStringBuf header, const TString& query) const {
+    const TString scaleFactor = "$scale_factor = " + ToString(Params.GetScale()) + ";";
     for(TStringBuf line; header.ReadLine(line);) {
         const auto pos = line.find('=');
         if (pos == line.npos) {
             continue;
         }
         const auto name = StripString(line.SubString(0, pos));
         if (name == "$scale_factor") {
-            line = "$scale_factor = " + ToString(Params.GetScale()) + ";"; 
+            line = scaleFactor; 
         } 
         for(auto posInQ = query.find(name); posInQ != query.npos; posInQ = query.find(name, posInQ)) {
             posInQ += name.length();

ydb/library/workload/tpch/ut/queries_ut.cpp

@@ -0,0 +1,26 @@
+#include <ydb/library/workload/tpch/tpch.h>
+#include <library/cpp/testing/unittest/registar.h>
+#include <library/cpp/regex/pcre/regexp.h>
+
+using namespace NYdbWorkload;
+
+Y_UNIT_TEST_SUITE(TpchQueries) {
+    Y_UNIT_TEST(ScaleFactor) {
+        TTpchWorkloadParams params;
+        NLastGetopt::TOpts opts;
+        params.ConfigureOpts(opts, TWorkloadParams::ECommandType::Run, 0);
+        const char* args[] = {"ut", "--scale", "10"};
+        for (auto parser = NLastGetopt::TOptsParser(&opts, sizeof(args)/sizeof(*args), args); parser.Next();) {
+        }
+        auto generator = params.CreateGenerator();
+        ui32 query_num = 0;
+        for (const auto& query: generator->GetWorkload(0)) {
+            if (query_num == 11) {
+                UNIT_ASSERT_C(TRegExMatch("\\$scale_factor\\s*=\\s*10\\s*;").Match(query.Query.c_str()),
+                    "Scale factor is wrong for q11:" << Endl << query.Query);
+                break;
+            }
+            ++query_num;
+        }
+    }
+};

ydb/library/workload/tpch/ut/ya.make

@@ -0,0 +1,11 @@
+UNITTEST_FOR(ydb/library/workload/tpch)
+
+SIZE(SMALL)
+
+SRCS(queries_ut.cpp)
+
+PEERDIR(
+    library/cpp/regex/pcre
+)
+
+END()

ydb/library/workload/tpch/ya.make

@@ -100,3 +100,5 @@ PEERDIR(
 )
 
 END()
+
+RECURSE_FOR_TESTS(ut)