From 7f5f8749dae7b4beeccfb743b2c870e39abb71fa Mon Sep 17 00:00:00 2001
From: Ning Sun <sunng@protonmail.com>
Date: Wed, 9 Aug 2023 14:10:54 +0800
Subject: [PATCH] test: add conditional compilation flag for datanode mock
 module (#2130)

---
 src/datanode/src/lib.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/datanode/src/lib.rs b/src/datanode/src/lib.rs
index 7661a7133e..c5c36e531d 100644
--- a/src/datanode/src/lib.rs
+++ b/src/datanode/src/lib.rs
@@ -21,7 +21,7 @@ mod greptimedb_telemetry;
 pub mod heartbeat;
 pub mod instance;
 pub mod metrics;
-#[cfg(feature = "testing")]
+#[cfg(any(test, feature = "testing"))]
 mod mock;
 pub mod server;
 pub mod sql;