From e0525dbfebc524b50bf0f734fbb79b7778b29184 Mon Sep 17 00:00:00 2001
From: LFC <990479+MichaelScofield@users.noreply.github.com>
Date: Fri, 5 Jan 2024 14:54:01 +0800
Subject: [PATCH] chore: expose some codes to let other projects use them
 (#3102)

---
 src/datanode/Cargo.toml                 | 3 +++
 src/datanode/src/lib.rs                 | 4 ++--
 src/frontend/src/instance/standalone.rs | 2 +-
 src/meta-srv/src/procedure.rs           | 2 +-
 src/meta-srv/src/procedure/utils.rs     | 2 +-
 5 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/src/datanode/Cargo.toml b/src/datanode/Cargo.toml
index 85afd709e4..3bed5bf9c5 100644
--- a/src/datanode/Cargo.toml
+++ b/src/datanode/Cargo.toml
@@ -4,6 +4,9 @@ version.workspace = true
 edition.workspace = true
 license.workspace = true
 
+[features]
+testing = []
+
 [dependencies]
 api.workspace = true
 arrow-flight.workspace = true
diff --git a/src/datanode/src/lib.rs b/src/datanode/src/lib.rs
index fc7126c2e5..0e036f2bd5 100644
--- a/src/datanode/src/lib.rs
+++ b/src/datanode/src/lib.rs
@@ -24,5 +24,5 @@ pub mod heartbeat;
 pub mod metrics;
 pub mod region_server;
 mod store;
-#[cfg(test)]
-mod tests;
+#[cfg(any(test, feature = "testing"))]
+pub mod tests;
diff --git a/src/frontend/src/instance/standalone.rs b/src/frontend/src/instance/standalone.rs
index b7a067d49c..4ff928fe6d 100644
--- a/src/frontend/src/instance/standalone.rs
+++ b/src/frontend/src/instance/standalone.rs
@@ -40,7 +40,7 @@ impl DatanodeManager for StandaloneDatanodeManager {
 }
 
 /// Relative to [client::region::RegionRequester]
-struct RegionInvoker {
+pub struct RegionInvoker {
     region_server: RegionServer,
 }
 
diff --git a/src/meta-srv/src/procedure.rs b/src/meta-srv/src/procedure.rs
index 45aaf061db..009915f7d9 100644
--- a/src/meta-srv/src/procedure.rs
+++ b/src/meta-srv/src/procedure.rs
@@ -16,4 +16,4 @@ pub mod region_failover;
 pub mod region_migration;
 #[cfg(test)]
 mod tests;
-mod utils;
+pub mod utils;
diff --git a/src/meta-srv/src/procedure/utils.rs b/src/meta-srv/src/procedure/utils.rs
index f0f0ba72bb..1441aa5968 100644
--- a/src/meta-srv/src/procedure/utils.rs
+++ b/src/meta-srv/src/procedure/utils.rs
@@ -18,7 +18,7 @@ pub fn region_lock_key(table_id: TableId, region_number: RegionNumber) -> String
     format!("{}/region-{}", table_id, region_number)
 }
 
-#[cfg(test)]
+#[cfg(feature = "mock")]
 pub mod mock {
     use std::io::Error;
     use std::sync::Arc;