Merge pull request #712 from bytedance/feat-ut

fix unit test
bytedance · Nov 22, 2024 · 45e9b6a · 45e9b6a
2 parents 9e37d8c + 36636dc
commit 45e9b6a
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/plugins/lib/rust/src/lib.rs b/plugins/lib/rust/src/lib.rs
@@ -217,7 +217,7 @@ impl Client {
                 w.write_all(rec.timestamp.to_string().as_bytes())?;
                 w.write_all(b",\"data\":")?;
                 serde_json::to_writer(w.by_ref(), rec.get_data().get_fields())?;
-                w.write_all(b"}\n")
+                w.write_all(b"}\n")?
             }
             Ok(())
        }

diff --git a/rasp/librasp/src/manager.rs b/rasp/librasp/src/manager.rs
@@ -997,18 +997,20 @@ mod tests {
         fake_configs.push(PidMissingProbeConfig {
             message_type: 9,
             data: ProbeConfigData {
-                uuid: "fake".to_string(),
+                uuid: Some("fake".to_string()),
                 blocks: None,
                 filters: None,
                 limits: None,
                 patches: Some(fake_patches),
+               ..Default::default()
             },
         });
         let fake_manager = RASPManager {
             namespace_tracer: MntNamespaceTracer::new(),
             thread_comm: None,
             process_comm: None,
             runtime_dir: false,
+            ebpf_comm: None,
         };
         println!("{:?}", fake_configs);
         let _ = fake_manager

diff --git a/rasp/librasp/src/php.rs b/rasp/librasp/src/php.rs
@@ -371,6 +371,9 @@ mod php_test {
                     Ok(ProbeState::NotAttach) => {
                         println!("probe NotATTACH");
                     }
+                    Ok(ProbeState::AttachedVersionNotMatch) => {
+                        println!("probe AttachedVersionNotMatch");
+                    }
                     Err(e) => {
                         println!("{} check probe state failed: {}", process_info.pid, e);
                     }