formatting

Author: milindsrivastava1997

asap-planner-rs/src/lib.rs

@@ -137,7 +137,9 @@ impl PlannerOutput {
                             if qs == query {
                                 if let Some(YamlValue::Sequence(aggs)) = qm.get("aggregations") {
                                     if let Some(YamlValue::Mapping(agg)) = aggs.first() {
-                                        for key in ["read_count_threshold", "num_aggregates_to_retain"] {
+                                        for key in
+                                            ["read_count_threshold", "num_aggregates_to_retain"]
+                                        {
                                             if let Some(v) = agg.get(key) {
                                                 if let YamlValue::Number(n) = v {
                                                     return n.as_u64();

asap-planner-rs/src/planner/logics.rs

@@ -295,9 +295,16 @@ mod tests {
         // t_lookback = t_repeat = 300 (OnlySpatial path)
         // effective_repeat = 300 (step=0)
         // ceil((300 + 0) / 300) = 1
-        let result =
-            get_cleanup_param(CleanupPolicy::CircularBuffer, pt, &mr, 300, "tumbling", 0, 0)
-                .unwrap();
+        let result = get_cleanup_param(
+            CleanupPolicy::CircularBuffer,
+            pt,
+            &mr,
+            300,
+            "tumbling",
+            0,
+            0,
+        )
+        .unwrap();
         assert_eq!(result, 1);
     }
 
@@ -306,9 +313,16 @@ mod tests {
         let (pt, mr) = match_query("sum(some_metric)");
         // t_lookback = t_repeat = 300, effective_repeat = min(300, 30) = 30
         // ceil((300 + 3600) / 30) = ceil(130) = 130
-        let result =
-            get_cleanup_param(CleanupPolicy::CircularBuffer, pt, &mr, 300, "tumbling", 3600, 30)
-                .unwrap();
+        let result = get_cleanup_param(
+            CleanupPolicy::CircularBuffer,
+            pt,
+            &mr,
+            300,
+            "tumbling",
+            3600,
+            30,
+        )
+        .unwrap();
         assert_eq!(result, 130);
     }
 
@@ -355,8 +369,15 @@ mod tests {
     fn cleanup_param_mismatched_range_and_step_returns_error() {
         let (pt, mr) = match_query("sum(some_metric)");
         // range_duration > 0 but step == 0 is invalid
-        let result =
-            get_cleanup_param(CleanupPolicy::CircularBuffer, pt, &mr, 300, "tumbling", 3600, 0);
+        let result = get_cleanup_param(
+            CleanupPolicy::CircularBuffer,
+            pt,
+            &mr,
+            300,
+            "tumbling",
+            3600,
+            0,
+        );
         assert!(result.is_err());
     }
 }

asap-planner-rs/src/planner/single_query.rs

@@ -447,7 +447,10 @@ mod tests {
 
     #[test]
     fn identical_configs_have_same_key() {
-        assert_eq!(base_config().identifying_key(), base_config().identifying_key());
+        assert_eq!(
+            base_config().identifying_key(),
+            base_config().identifying_key()
+        );
     }
 
     #[test]
@@ -478,10 +481,14 @@ mod tests {
     fn parameter_insertion_order_does_not_affect_key() {
         let mut cfg1 = base_config();
         let mut cfg2 = base_config();
-        cfg1.parameters.insert("depth".to_string(), Value::Number(3.into()));
-        cfg1.parameters.insert("width".to_string(), Value::Number(1024.into()));
-        cfg2.parameters.insert("width".to_string(), Value::Number(1024.into()));
-        cfg2.parameters.insert("depth".to_string(), Value::Number(3.into()));
+        cfg1.parameters
+            .insert("depth".to_string(), Value::Number(3.into()));
+        cfg1.parameters
+            .insert("width".to_string(), Value::Number(1024.into()));
+        cfg2.parameters
+            .insert("width".to_string(), Value::Number(1024.into()));
+        cfg2.parameters
+            .insert("depth".to_string(), Value::Number(3.into()));
         assert_eq!(cfg1.identifying_key(), cfg2.identifying_key());
     }
 }

asap-planner-rs/tests/integration.rs

@@ -217,7 +217,10 @@ aggregate_cleanup:
   policy: "not_a_real_policy"
 "#;
     let c = Controller::from_yaml(yaml, arroyo_opts()).unwrap();
-    assert!(matches!(c.generate(), Err(ControllerError::PlannerError(_))));
+    assert!(matches!(
+        c.generate(),
+        Err(ControllerError::PlannerError(_))
+    ));
 }
 
 #[test]
@@ -237,7 +240,10 @@ metrics:
     labels: ["instance"]
 "#;
     let c = Controller::from_yaml(yaml, arroyo_opts()).unwrap();
-    assert!(matches!(c.generate(), Err(ControllerError::DuplicateQuery(_))));
+    assert!(matches!(
+        c.generate(),
+        Err(ControllerError::DuplicateQuery(_))
+    ));
 }
 
 #[test]
@@ -263,7 +269,10 @@ metrics:
     labels: ["instance"]
 "#;
     let c = Controller::from_yaml(yaml, arroyo_opts()).unwrap();
-    assert!(matches!(c.generate(), Err(ControllerError::DuplicateQuery(_))));
+    assert!(matches!(
+        c.generate(),
+        Err(ControllerError::DuplicateQuery(_))
+    ));
 }
 
 #[test]
@@ -282,7 +291,10 @@ metrics:
     labels: ["instance"]
 "#;
     let c = Controller::from_yaml(yaml, arroyo_opts()).unwrap();
-    assert!(matches!(c.generate(), Err(ControllerError::UnknownMetric(_))));
+    assert!(matches!(
+        c.generate(),
+        Err(ControllerError::UnknownMetric(_))
+    ));
 }
 
 #[test]