From 13d5230e5da650cf93e6dccb389c82d316d355c6 Mon Sep 17 00:00:00 2001
From: Peter Dillinger <peterd@meta.com>
Date: Mon, 26 Aug 2024 14:08:21 -0700
Subject: [PATCH] Fix flaky test DBTest2.VariousFileTemperatures (#12974)

Summary:
... apparently due to potentially not purging obsolete files after CompactRange

Example: https://github.com/facebook/rocksdb/actions/runs/10564621261/job/29267393711?pr=12959

Pull Request resolved: https://github.com/facebook/rocksdb/pull/12974

Test Plan: reproduced failure with USE_CLANG=1 COERCE_CONTEXT_SWITCH=1, now fixed

Reviewed By: cbi42

Differential Revision: D61812600

Pulled By: pdillinger

fbshipit-source-id: d4b23e1a179bb8ec39875ed7a8ce1649fa3344bd
---
 db/db_test2.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/db/db_test2.cc b/db/db_test2.cc
index f380144c6ab..92211ad42dd 100644
--- a/db/db_test2.cc
+++ b/db/db_test2.cc
@@ -6739,6 +6739,7 @@ TEST_F(DBTest2, VariousFileTemperatures) {
                 TCM({{options.default_write_temperature, 2}}));
 
       ASSERT_OK(db_->CompactRange({}, nullptr, nullptr));
+      ASSERT_OK(dbfull()->TEST_WaitForBackgroundWork());
 
       ASSERT_EQ(test_fs->CountCurrentSstFilesByTemp(),
                 TCM({{options.last_level_temperature, 1}}));