From eb4d6af5ae6268033407cf2eb5e9a56b57bc4ceb Mon Sep 17 00:00:00 2001
From: Huisheng Liu <hliu@microsoft.com>
Date: Mon, 3 Feb 2020 13:30:13 -0800
Subject: [PATCH] Error handler test fix (#6266)

Summary:
MultiDBCompactionError fails when it verifies the number of files on level 0 and level 1 without waiting for compaction to finish.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/6266

Differential Revision: D19701639

Pulled By: riversand963

fbshipit-source-id: e96d511bcde705075f073e0b550cebcd2ecfccdc
---
 db/error_handler_test.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/db/error_handler_test.cc b/db/error_handler_test.cc
index 3af5c26f5..929d07e9e 100644
--- a/db/error_handler_test.cc
+++ b/db/error_handler_test.cc
@@ -701,6 +701,8 @@ TEST_F(DBErrorHandlingTest, MultiDBCompactionError) {
   for (auto i = 0; i < kNumDbInstances; ++i) {
     std::string prop;
     ASSERT_EQ(listener[i]->WaitForRecovery(5000000), true);
+    ASSERT_EQ(static_cast<DBImpl*>(db[i])->TEST_WaitForCompact(true),
+              Status::OK());
     EXPECT_TRUE(db[i]->GetProperty(
         "rocksdb.num-files-at-level" + NumberToString(0), &prop));
     EXPECT_EQ(atoi(prop.c_str()), 0);