add test case for limit = 1

vesoft-inc · Sep 23, 2021 · 23bf1ab · 23bf1ab
1 parent 4560209
commit 23bf1ab
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/src/storage/test/IndexScanLimitTest.cpp b/src/storage/test/IndexScanLimitTest.cpp
@@ -230,6 +230,17 @@ TEST_F(IndexScanLimitTest, LookupTagIndexLimit) {
     EXPECT_EQ(0, resp.get_data()->rows.size());
   }
 
+  // limit == 1
+  {
+    req.set_limit(1);
+    auto* processor = LookupProcessor::instance(storageEnv_.get(), nullptr, nullptr);
+    auto fut = processor->getFuture();
+    processor->process(req);
+    auto resp = std::move(fut).get();
+    EXPECT_EQ(0, resp.result.failed_parts.size());
+    EXPECT_EQ(1 * parts.size(), resp.get_data()->rows.size());
+  }
+
   // limit 5 by each part
   {
     req.set_limit(5);
@@ -324,6 +335,17 @@ TEST_F(IndexScanLimitTest, LookupEdgeIndexLimit) {
     EXPECT_EQ(0, resp.get_data()->rows.size());
   }
 
+  // limit == 1
+  {
+    req.set_limit(1);
+    auto* processor = LookupProcessor::instance(storageEnv_.get(), nullptr, nullptr);
+    auto fut = processor->getFuture();
+    processor->process(req);
+    auto resp = std::move(fut).get();
+    EXPECT_EQ(0, resp.result.failed_parts.size());
+    EXPECT_EQ(1 * parts.size(), resp.get_data()->rows.size());
+  }
+
   // limit 5 by each part
   {
     req.set_limit(5);