wehub-resource-sync
137 lines
4.5 KiB
Rust
137 lines
4.5 KiB
Rust
use re_chunk_store::RowId;
|
|
use re_log_types::{Timeline, TimelineName};
|
|
use re_sdk_types::archetypes::Scalars;
|
|
use re_test_context::TestContext;
|
|
use re_test_context::external::egui_kittest::SnapshotResults;
|
|
use re_test_viewport::TestContextExt as _;
|
|
use re_ui::UiExt as _;
|
|
use re_view_dataframe::DataframeView;
|
|
use re_viewer_context::{ViewClass as _, ViewId};
|
|
use re_viewport_blueprint::ViewBlueprint;
|
|
|
|
#[test]
|
|
pub fn test_null_timeline() {
|
|
let mut test_context = get_test_context();
|
|
|
|
let timeline_a = Timeline::new_sequence("timeline_a");
|
|
let timeline_b = Timeline::new_sequence("timeline_b");
|
|
|
|
test_context.log_entity("first", |builder| {
|
|
builder.with_archetype(RowId::new(), [(timeline_a, 0)], &Scalars::single(10.0))
|
|
});
|
|
|
|
test_context.log_entity("second", |builder| {
|
|
builder.with_archetype(
|
|
RowId::new(),
|
|
[(timeline_a, 1), (timeline_b, 10)],
|
|
&Scalars::single(12.0),
|
|
)
|
|
});
|
|
|
|
test_context.set_active_timeline(*timeline_a.name());
|
|
|
|
let view_id = setup_blueprint(&mut test_context, timeline_a.name());
|
|
test_context
|
|
.run_view_ui_and_save_snapshot(view_id, "null_timeline", egui::vec2(400.0, 200.0), None)
|
|
.unwrap();
|
|
}
|
|
|
|
#[test]
|
|
pub fn test_unknown_timeline() {
|
|
let mut snapshot_results = SnapshotResults::new();
|
|
let mut test_context = get_test_context();
|
|
|
|
let timeline = Timeline::new_sequence("existing_timeline");
|
|
|
|
test_context.log_entity("some_entity", |builder| {
|
|
builder
|
|
.with_archetype(RowId::new(), [(timeline, 0)], &Scalars::single(10.0))
|
|
.with_archetype(RowId::new(), [(timeline, 1)], &Scalars::single(20.0))
|
|
.with_archetype(RowId::new(), [(timeline, 2)], &Scalars::single(30.0))
|
|
});
|
|
|
|
let view_id = setup_blueprint(&mut test_context, &TimelineName::from("unknown_timeline"));
|
|
|
|
snapshot_results.add(test_context.run_view_ui_and_save_snapshot(
|
|
view_id,
|
|
"unknown_timeline_view_ui",
|
|
egui::vec2(300.0, 150.0),
|
|
None,
|
|
));
|
|
|
|
run_view_selection_panel_ui_and_save_snapshot(
|
|
&test_context,
|
|
view_id,
|
|
"unknown_timeline_selection_panel_ui",
|
|
egui::vec2(300.0, 450.0),
|
|
&mut snapshot_results,
|
|
);
|
|
}
|
|
|
|
fn get_test_context() -> TestContext {
|
|
let mut test_context = TestContext::new_with_view_class::<re_view_dataframe::DataframeView>();
|
|
|
|
// Make sure we can draw stuff in the table.
|
|
test_context.component_ui_registry = re_component_ui::create_component_ui_registry();
|
|
|
|
test_context
|
|
}
|
|
|
|
fn setup_blueprint(test_context: &mut TestContext, timeline_name: &TimelineName) -> ViewId {
|
|
test_context.setup_viewport_blueprint(|ctx, blueprint| {
|
|
let view_id = blueprint.add_view_at_root(ViewBlueprint::new_with_root_wildcard(
|
|
re_view_dataframe::DataframeView::identifier(),
|
|
));
|
|
|
|
// the dataframe view to the desired timeline
|
|
let query = re_view_dataframe::Query::from_blueprint(ctx, view_id);
|
|
query.save_timeline_name(ctx, timeline_name);
|
|
|
|
view_id
|
|
})
|
|
}
|
|
|
|
fn run_view_selection_panel_ui_and_save_snapshot(
|
|
test_context: &TestContext,
|
|
view_id: ViewId,
|
|
name: &str,
|
|
size: egui::Vec2,
|
|
snapshot_results: &mut SnapshotResults,
|
|
) {
|
|
let mut harness = test_context
|
|
.setup_kittest_for_rendering_ui(size)
|
|
.build_ui(|ui| {
|
|
test_context.run_ui(ui, |ctx, ui| {
|
|
let view_class = ctx
|
|
.view_class_registry()
|
|
.get_class_or_log_error(DataframeView::identifier());
|
|
|
|
let view_blueprint = ViewBlueprint::try_from_db(
|
|
view_id,
|
|
ctx.store_context.blueprint,
|
|
ctx.blueprint_query,
|
|
)
|
|
.expect("we just created that view");
|
|
|
|
let spacing = ui.spacing().item_spacing;
|
|
ui.list_item_scope("test_harness", |ui| {
|
|
ui.spacing_mut().item_spacing = spacing;
|
|
|
|
let mut view_states = test_context.view_states.lock();
|
|
let view_state =
|
|
view_states.get_mut_or_create(ctx.store_id(), view_id, view_class);
|
|
|
|
view_class
|
|
.selection_ui(ctx, ui, view_state, &view_blueprint.space_origin, view_id)
|
|
.expect("failed to run view selection panel ui");
|
|
});
|
|
});
|
|
|
|
test_context.handle_system_commands(ui.ctx());
|
|
});
|
|
|
|
harness.run();
|
|
harness.snapshot(name);
|
|
snapshot_results.extend_harness(&mut harness);
|
|
}
|