Add count support to REPEAT

kuzudb · Jun 28, 2023 · 34d4c97 · 34d4c97
1 parent faada3f
commit 34d4c97
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/test_runner/test_parser.cpp b/test/test_runner/test_parser.cpp
@@ -201,7 +201,7 @@ void TestParser::extractStatementBlock() {
 }
 
 std::string TestParser::parseCommand() {
-    // REPEAT 3 "Alice " = "Alice Alice Alice "
+    // REPEAT 3 "col${count}, " = "col0, col1, col2, "
     if (currentToken.params[2] == "REPEAT") {
         checkMinimumParams(4);
         return parseCommandRepeat();
@@ -226,7 +226,9 @@ std::string TestParser::parseCommandRepeat() {
         throw TestException("Invalid DEFINE data type [" + path + ":" + line + "].");
     }
     for (auto i = 0; i < times; i++) {
-        result += repeatString;
+        auto stringToAppend = repeatString;
+        StringUtils::replaceAll(stringToAppend, "${count}", std::to_string(i));
+        result += stringToAppend;
     }
     return result;
 }