chore(tesseract): Join Graph mock (#10143)

Author: waralexrom

rust/cubesqlplanner/Cargo.lock

@@ -22,6 +22,9 @@ lazy_static = "1.4.0"
 regex = "1.3.9"
 typed-builder = "0.21.2"
 
+[dev-dependencies]
+petgraph = "0.6"
+
 [dependencies.neon]
 version = "=1"
 default-features = false

rust/cubesqlplanner/cubesqlplanner/Cargo.toml

@@ -3,28 +3,6 @@
 /// This macro generates a helper method that returns an owned StaticData struct.
 /// The helper is used by the trait's static_data() method which applies Box::leak.
 ///
-/// # Usage
-/// ```ignore
-/// impl_static_data!(
-///     MockDimensionDefinition,          // The mock type
-///     DimensionDefinitionStatic,         // The static data type
-///     dimension_type,                    // Fields to include
-///     owned_by_cube,
-///     multi_stage
-/// );
-/// ```
-///
-/// # Generated Code
-/// ```ignore
-/// impl MockDimensionDefinition {
-///     pub fn static_data(&self) -> DimensionDefinitionStatic {
-///         DimensionDefinitionStatic {
-///             dimension_type: self.dimension_type.clone(),
-///             owned_by_cube: self.owned_by_cube.clone(),
-///             multi_stage: self.multi_stage.clone(),
-///         }
-///     }
-/// }
 /// ```
 #[macro_export]
 macro_rules! impl_static_data {
@@ -54,24 +32,6 @@ macro_rules! impl_static_data {
 /// - The leaked memory is minimal and reclaimed when the test process exits
 /// - This approach significantly simplifies test code by avoiding complex lifetime management
 ///
-/// # Usage
-/// ```ignore
-/// impl DimensionDefinition for MockDimensionDefinition {
-///     impl_static_data_method!(DimensionDefinitionStatic);
-///
-///     fn sql(&self) -> Result<Option<Rc<dyn MemberSql>>, CubeError> {
-///         // ... other trait methods
-///     }
-/// }
-/// ```
-///
-/// # Generated Code
-/// ```ignore
-/// fn static_data(&self) -> &DimensionDefinitionStatic {
-///     // Intentional memory leak - acceptable for test mocks
-///     // The Box::leak pattern converts the owned value to a static reference
-///     Box::leak(Box::new(Self::static_data(self)))
-/// }
 /// ```
 #[macro_export]
 macro_rules! impl_static_data_method {

rust/cubesqlplanner/cubesqlplanner/src/test_fixtures/cube_bridge/macros.rs

@@ -20,21 +20,6 @@ use typed_builder::TypedBuilder;
 /// security_context_for_rust, and sql_utils_for_rust.
 /// Other methods throw todo!() errors.
 ///
-/// # Example
-///
-/// ```
-/// use cubesqlplanner::test_fixtures::cube_bridge::MockBaseTools;
-///
-/// // Use builder pattern
-/// let tools = MockBaseTools::builder().build();
-/// let driver_tools = tools.driver_tools(false).unwrap();
-/// let sql_templates = tools.sql_templates().unwrap();
-///
-/// // Or with custom components
-/// let custom_driver = MockDriverTools::with_timezone("Europe/London".to_string());
-/// let tools = MockBaseTools::builder()
-///     .driver_tools(custom_driver)
-///     .build();
 /// ```
 #[derive(Clone, TypedBuilder)]
 pub struct MockBaseTools {
@@ -62,27 +47,22 @@ impl BaseTools for MockBaseTools {
         self
     }
 
-    /// Returns driver tools - uses MockDriverTools
     fn driver_tools(&self, _external: bool) -> Result<Rc<dyn DriverTools>, CubeError> {
         Ok(self.driver_tools.clone())
     }
 
-    /// Returns SQL templates renderer - uses MockSqlTemplatesRender
     fn sql_templates(&self) -> Result<Rc<dyn SqlTemplatesRender>, CubeError> {
         Ok(self.sql_templates.clone())
     }
 
-    /// Returns security context - uses MockSecurityContext
     fn security_context_for_rust(&self) -> Result<Rc<dyn SecurityContext>, CubeError> {
         Ok(self.security_context.clone())
     }
 
-    /// Returns SQL utils - uses MockSqlUtils
     fn sql_utils_for_rust(&self) -> Result<Rc<dyn SqlUtils>, CubeError> {
         Ok(self.sql_utils.clone())
     }
 
-    /// Generate time series - not implemented in mock
     fn generate_time_series(
         &self,
         _granularity: String,
@@ -91,7 +71,6 @@ impl BaseTools for MockBaseTools {
         todo!("generate_time_series not implemented in mock")
     }
 
-    /// Generate custom time series - not implemented in mock
     fn generate_custom_time_series(
         &self,
         _granularity: String,
@@ -101,22 +80,18 @@ impl BaseTools for MockBaseTools {
         todo!("generate_custom_time_series not implemented in mock")
     }
 
-    /// Get allocated parameters - not implemented in mock
     fn get_allocated_params(&self) -> Result<Vec<String>, CubeError> {
         todo!("get_allocated_params not implemented in mock")
     }
 
-    /// Get all cube members - not implemented in mock
     fn all_cube_members(&self, _path: String) -> Result<Vec<String>, CubeError> {
         todo!("all_cube_members not implemented in mock")
     }
 
-    /// Get interval and minimal time unit - not implemented in mock
     fn interval_and_minimal_time_unit(&self, _interval: String) -> Result<Vec<String>, CubeError> {
         todo!("interval_and_minimal_time_unit not implemented in mock")
     }
 
-    /// Get pre-aggregation by name - not implemented in mock
     fn get_pre_aggregation_by_name(
         &self,
         _cube_name: String,
@@ -125,7 +100,6 @@ impl BaseTools for MockBaseTools {
         todo!("get_pre_aggregation_by_name not implemented in mock")
     }
 
-    /// Get pre-aggregation table name - not implemented in mock
     fn pre_aggregation_table_name(
         &self,
         _cube_name: String,
@@ -134,128 +108,10 @@ impl BaseTools for MockBaseTools {
         todo!("pre_aggregation_table_name not implemented in mock")
     }
 
-    /// Get join tree for hints - not implemented in mock
     fn join_tree_for_hints(
         &self,
         _hints: Vec<JoinHintItem>,
     ) -> Result<Rc<dyn JoinDefinition>, CubeError> {
         todo!("join_tree_for_hints not implemented in mock")
     }
 }
-
-#[cfg(test)]
-mod tests {
-    use super::*;
-
-    #[test]
-    fn test_builder_default() {
-        let tools = MockBaseTools::builder().build();
-        assert!(tools.driver_tools(false).is_ok());
-        assert!(tools.sql_templates().is_ok());
-        assert!(tools.security_context_for_rust().is_ok());
-        assert!(tools.sql_utils_for_rust().is_ok());
-    }
-
-    #[test]
-    fn test_default_trait() {
-        let tools = MockBaseTools::default();
-        assert!(tools.driver_tools(false).is_ok());
-        assert!(tools.sql_templates().is_ok());
-        assert!(tools.security_context_for_rust().is_ok());
-        assert!(tools.sql_utils_for_rust().is_ok());
-    }
-
-    #[test]
-    fn test_driver_tools() {
-        let tools = MockBaseTools::builder().build();
-        let driver_tools = tools.driver_tools(false).unwrap();
-
-        // Test that it returns a valid DriverTools implementation
-        let result = driver_tools
-            .time_grouped_column("day".to_string(), "created_at".to_string())
-            .unwrap();
-        assert_eq!(result, "date_trunc('day', created_at)");
-    }
-
-    #[test]
-    fn test_driver_tools_external_flag() {
-        let tools = MockBaseTools::builder().build();
-
-        // Both external true and false should work (mock ignores the flag)
-        assert!(tools.driver_tools(false).is_ok());
-        assert!(tools.driver_tools(true).is_ok());
-    }
-
-    #[test]
-    fn test_sql_templates() {
-        let tools = MockBaseTools::builder().build();
-        let templates = tools.sql_templates().unwrap();
-
-        // Test that it returns a valid SqlTemplatesRender implementation
-        assert!(templates.contains_template("filters/equals"));
-        assert!(templates.contains_template("functions/SUM"));
-    }
-
-    #[test]
-    fn test_security_context() {
-        let tools = MockBaseTools::builder().build();
-        // Just verify it returns without error
-        assert!(tools.security_context_for_rust().is_ok());
-    }
-
-    #[test]
-    fn test_sql_utils() {
-        let tools = MockBaseTools::builder().build();
-        // Just verify it returns without error
-        assert!(tools.sql_utils_for_rust().is_ok());
-    }
-
-    #[test]
-    fn test_builder_with_custom_driver_tools() {
-        let custom_driver = MockDriverTools::with_timezone("Europe/London".to_string());
-        let tools = MockBaseTools::builder()
-            .driver_tools(Rc::new(custom_driver))
-            .build();
-
-        let driver_tools = tools.driver_tools(false).unwrap();
-        let result = driver_tools.convert_tz("timestamp".to_string()).unwrap();
-        assert_eq!(
-            result,
-            "(timestamp::timestamptz AT TIME ZONE 'Europe/London')"
-        );
-    }
-
-    #[test]
-    fn test_builder_with_custom_sql_templates() {
-        let mut custom_templates = std::collections::HashMap::new();
-        custom_templates.insert("test/template".to_string(), "TEST {{value}}".to_string());
-        let sql_templates = MockSqlTemplatesRender::try_new(custom_templates).unwrap();
-
-        let tools = MockBaseTools::builder()
-            .sql_templates(Rc::new(sql_templates))
-            .build();
-
-        let templates = tools.sql_templates().unwrap();
-        assert!(templates.contains_template("test/template"));
-    }
-
-    #[test]
-    fn test_builder_with_all_custom_components() {
-        let driver_tools = MockDriverTools::with_timezone("Asia/Tokyo".to_string());
-        let sql_templates = MockSqlTemplatesRender::default_templates();
-        let security_context = MockSecurityContext;
-        let sql_utils = MockSqlUtils;
-
-        let tools = MockBaseTools::builder()
-            .driver_tools(Rc::new(driver_tools))
-            .sql_templates(Rc::new(sql_templates))
-            .security_context(Rc::new(security_context))
-            .sql_utils(Rc::new(sql_utils))
-            .build();
-
-        assert!(tools.driver_tools(false).is_ok());
-        assert!(tools.sql_templates().is_ok());
-        assert!(tools.security_context_for_rust().is_ok());
-        assert!(tools.sql_utils_for_rust().is_ok());
-    }
-}

rust/cubesqlplanner/cubesqlplanner/src/test_fixtures/cube_bridge/mock_base_tools.rs

@@ -7,7 +7,6 @@ use std::any::Any;
 use std::rc::Rc;
 use typed_builder::TypedBuilder;
 
-/// Mock implementation of CaseDefinition for testing
 #[derive(TypedBuilder)]
 pub struct MockCaseDefinition {
     when: Vec<Rc<MockCaseItem>>,

rust/cubesqlplanner/cubesqlplanner/src/test_fixtures/cube_bridge/mock_case_definition.rs

@@ -5,7 +5,6 @@ use std::any::Any;
 use std::rc::Rc;
 use typed_builder::TypedBuilder;
 
-/// Mock implementation of CaseElseItem for testing
 #[derive(Debug, Clone, TypedBuilder)]
 pub struct MockCaseElseItem {
     label: StringOrSql,

rust/cubesqlplanner/cubesqlplanner/src/test_fixtures/cube_bridge/mock_case_else_item.rs

@@ -7,7 +7,6 @@ use std::any::Any;
 use std::rc::Rc;
 use typed_builder::TypedBuilder;
 
-/// Mock implementation of CaseItem for testing
 #[derive(Debug, Clone, TypedBuilder)]
 pub struct MockCaseItem {
     sql: String,

rust/cubesqlplanner/cubesqlplanner/src/test_fixtures/cube_bridge/mock_case_item.rs

@@ -10,7 +10,6 @@ use std::any::Any;
 use std::rc::Rc;
 use typed_builder::TypedBuilder;
 
-/// Mock implementation of CaseSwitchDefinition for testing
 #[derive(TypedBuilder)]
 pub struct MockCaseSwitchDefinition {
     switch: String,