1 files changed, 26 insertions, 0 deletions
diff --git a/rust/tests/component.rs b/rust/tests/component.rs
new file mode 100644
index 00000000..c57785ce
--- /dev/null
+++ b/rust/tests/component.rs
@@ -0,0 +1,26 @@
+use binaryninja::binary_view::BinaryViewExt;
+use binaryninja::component::ComponentBuilder;
+use binaryninja::headless::Session;
+use rstest::*;
+use std::path::PathBuf;
+
+#[fixture]
+#[once]
+fn session() -> Session {
+    Session::new().expect("Failed to initialize session")
+}
+
+#[rstest]
+fn test_component_creation(_session: &Session) {
+    let out_dir = env!("OUT_DIR").parse::<PathBuf>().unwrap();
+    let view = binaryninja::load(out_dir.join("atox.obj")).expect("Failed to create view");
+    let component = ComponentBuilder::new(view.clone()).name("test").finalize();
+    assert_eq!(component.name().as_str(), "test");
+    let root_component = view.root_component().unwrap();
+    // We should only have our component.
+    assert_eq!(root_component.components().len(), 1);
+    assert!(
+        root_component.contains_component(&component),
+        "Component not found in root component"
+    );
+}