From 0fb7334220e681eb73582e348594ff1f2961a454 Mon Sep 17 00:00:00 2001
From: Xu Peng <xupeng3112@163.com>
Date: Tue, 7 May 2019 09:44:27 +0800
Subject: [PATCH] test(db): enhancement for unit test

Former-commit-id: 2fdacaf3f80a70c3319d9dc7b2756f042b5b5106
---
 cpp/unittest/db/db_tests.cpp | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/cpp/unittest/db/db_tests.cpp b/cpp/unittest/db/db_tests.cpp
index 30efd1313f..b85d2d01a1 100644
--- a/cpp/unittest/db/db_tests.cpp
+++ b/cpp/unittest/db/db_tests.cpp
@@ -39,7 +39,7 @@ TEST_F(DBTest, DB_TEST) {
         xb[d * i] += i / 2000.;
     }
 
-    int qb = 1;
+    int qb = 5;
     float *qxb = new float[d * qb];
     for(int i = 0; i < qb; i++) {
         for(int j = 0; j < d; j++) qxb[d * i + j] = drand48();
@@ -67,7 +67,15 @@ TEST_F(DBTest, DB_TEST) {
             STOP_TIMER(ss.str());
 
             ASSERT_STATS(stat);
-            ASSERT_EQ(results[0][0], target_ids[0]);
+            for (auto k=0; k<qb; ++k) {
+                ASSERT_EQ(results[k][0], target_ids[k]);
+                ss.str("");
+                ss << "Result [" << k << "]:";
+                for (auto result : results[k]) {
+                    ss << result << " ";
+                }
+                LOG(DEBUG) << ss.str();
+            }
             ASSERT_TRUE(count >= prev_count);
             std::this_thread::sleep_for(std::chrono::seconds(1));
         }