add testcase

sonic-net · Oct 8, 2024 · 0107fac · 0107fac
1 parent 2c75350
commit 0107fac
Showing 1 changed file with 21 additions and 13 deletions.
diff --git a/tests/redis_piped_state_ut.cpp b/tests/redis_piped_state_ut.cpp
@@ -106,29 +106,37 @@ static void consumerWorker(int index)
 {
     string tableName = "UT_REDIS_THREAD_" + to_string(index);
     DBConnector db(TEST_DB, 0, true);
-    ConsumerStateTable c(&db, tableName);
+    ConsumerStateTable c(&db, tableName, NUMBER_OF_OPS);
     Select cs;
     Selectable *selectcs;
     int numberOfKeysSet = 0;
     int numberOfKeyDeleted = 0;
     int ret, i = 0;
-    KeyOpFieldsValuesTuple kco;
+    std::deque<KeyOpFieldsValuesTuple> entries;
 
     cs.addSelectable(&c);
-    while ((ret = cs.select(&selectcs)) == Select::OBJECT)
+    while (true)
     {
-        c.pop(kco);
-        if (kfvOp(kco) == "SET")
-        {
-            numberOfKeysSet++;
-            validateFields(kfvKey(kco), kfvFieldsValues(kco));
-        } else if (kfvOp(kco) == "DEL")
+        ret = cs.select(&selectcs);
+        c.pops(entries);
+
+        for (auto& kco: entries)
         {
-            numberOfKeyDeleted++;
-        }
+            if (kfvOp(kco) == "SET")
+            {
+                numberOfKeysSet++;
+                validateFields(kfvKey(kco), kfvFieldsValues(kco));
+            } else if (kfvOp(kco) == "DEL")
+            {
+                numberOfKeyDeleted++;
+            }
+
+            if ((i++ % 100) == 0)
+                cout << "-" << flush;
 
-        if ((i++ % 100) == 0)
-            cout << "-" << flush;
+            if (numberOfKeyDeleted == NUMBER_OF_OPS)
+                break;
+        }
 
         if (numberOfKeyDeleted == NUMBER_OF_OPS)
             break;