From 4b79c715818130dd0480fc2ecb72ff122e1adb0c Mon Sep 17 00:00:00 2001
From: mrambacher <mrambach@gmail.com>
Date: Thu, 3 Aug 2023 01:42:01 -0400
Subject: [PATCH] Unit tests: Enable DBIOFailureTest.DropWrites (#544)

---
 db/db_io_failure_test.cc | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/db/db_io_failure_test.cc b/db/db_io_failure_test.cc
index e229dca1d1..a3ae28d786 100644
--- a/db/db_io_failure_test.cc
+++ b/db/db_io_failure_test.cc
@@ -20,7 +20,7 @@ class DBIOFailureTest : public DBTestBase {
 };
 
 // Check that number of files does not grow when writes are dropped
-TEST_F(DBIOFailureTest, DISABLED_DropWrites) {
+TEST_F(DBIOFailureTest, DropWrites) {
   do {
     Options options = CurrentOptions();
     options.env = env_;
@@ -30,7 +30,7 @@ TEST_F(DBIOFailureTest, DISABLED_DropWrites) {
     ASSERT_OK(Put("foo", "v1"));
     ASSERT_EQ("v1", Get("foo"));
     Compact("a", "z");
-    const size_t num_files = CountFiles();
+    const auto num_files = GetSstFileCount(dbname_);
     // Force out-of-space errors
     env_->drop_writes_.store(true, std::memory_order_release);
     env_->sleep_counter_.Reset();
@@ -59,7 +59,7 @@ TEST_F(DBIOFailureTest, DISABLED_DropWrites) {
     ASSERT_EQ("5", property_value);
 
     env_->drop_writes_.store(false, std::memory_order_release);
-    const size_t count = CountFiles();
+    const auto count = GetSstFileCount(dbname_);
     ASSERT_LT(count, num_files + 3);
 
     // Check that compaction attempts slept after errors