fix unit test

baoyachi · Jul 19, 2024 · d33c3ec · d33c3ec
1 parent 87bd316
commit d33c3ec
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/example_shadow/src/main.rs b/example_shadow/src/main.rs
@@ -49,6 +49,7 @@ pub fn print_build() {
     println!("rust_channel:{}", build::RUST_CHANNEL);
     println!("cargo_version:{}", build::CARGO_VERSION);
     println!("cargo_tree:{}", build::CARGO_TREE);
+    println!("cargo_tree:{}", build::CARGO_METADATA);
 
     println!("project_name:{}", build::PROJECT_NAME);
     println!("build_time:{}", build::BUILD_TIME);

diff --git a/src/lib.rs b/src/lib.rs
@@ -504,8 +504,15 @@ impl Shadow {
         let mut print_val = String::from("\n");
 
         // append gen const
-        for k in self.map.keys() {
-            let tmp = format!(r#"{}println!("{k}:{{{k}}}\n");{}"#, "\t", "\n");
+        for (k, v) in &self.map {
+            let tmp = match v.t {
+                ConstType::Str | ConstType::Bool => {
+                    format!(r#"{}println!("{k}:{{{k}}}\n");{}"#, "\t", "\n")
+                }
+                ConstType::Slice => {
+                    format!(r#"{}println!("{k}:{{:?}}\n",{});{}"#, "\t", k, "\n",)
+                }
+            };
             print_val.push_str(tmp.as_str());
         }