rust/tests/component.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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"
    );
}