From e3f45deccfab043e9e09a276e87a71cdfa6891b0 Mon Sep 17 00:00:00 2001
From: Denis Smirnov <sd@picodata.io>
Date: Thu, 7 Apr 2022 13:43:46 +0700
Subject: [PATCH] fix: move back mock under test macros.

---
 src/executor/engine.rs   | 4 +++-
 src/ir/transformation.rs | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/executor/engine.rs b/src/executor/engine.rs
index 11c42c7566..f99a8cbde6 100644
--- a/src/executor/engine.rs
+++ b/src/executor/engine.rs
@@ -5,7 +5,6 @@ use crate::executor::result::BoxExecuteFormat;
 use crate::executor::vtable::VirtualTable;
 
 pub mod cartridge;
-pub mod mock;
 
 /// `Metadata` trait is interface for working with metadata storage, that was needed the query execute.
 pub trait Metadata {
@@ -70,3 +69,6 @@ pub trait Engine {
     /// Determine shard for query execution by sharding key value
     fn determine_bucket_id(&self, s: &str) -> u64;
 }
+
+#[cfg(test)]
+pub mod mock;
diff --git a/src/ir/transformation.rs b/src/ir/transformation.rs
index dbfed05bf6..509a1d9e6c 100644
--- a/src/ir/transformation.rs
+++ b/src/ir/transformation.rs
@@ -5,7 +5,6 @@
 pub mod bool_in;
 pub mod dnf;
 pub mod equality_propagation;
-mod helpers;
 pub mod merge_tuples;
 pub mod redistribution;
 pub mod split_columns;
@@ -182,3 +181,6 @@ impl Plan {
         Ok(new_top_id)
     }
 }
+
+#[cfg(test)]
+mod helpers;
-- 
GitLab