shu 0.5.0

use crate::aesthetics::{AesPlugin, Aesthetics, Distribution, Gy, Point, Unscale};
use crate::geom::{AesFilter, GeomHist, HistTag, Xaxis};
use crate::gui::{file_drop, UiState};
use crate::{data, escher, geom};
use bevy::prelude::*;
use bevy::window::WindowId;
use bevy_prototype_lyon::prelude::{DrawMode, GeometryBuilder, Path, PathBuilder, StrokeMode};

use bevy::asset::FileAssetIo;
use bevy::tasks::IoTaskPool;

/// Setup to test systems that require [`AsserServer`] as an argument.
/// Adapted form bevy source code.
fn setup(asset_path: impl AsRef<std::path::Path>) -> AssetServer {
    IoTaskPool::init(Default::default);

    AssetServer::new(FileAssetIo::new(asset_path, false))
}

#[test]
fn gy_dist_aes_spaws_xaxis_spawns_hist() {
    // Setup app
    let mut app = App::new();
    // build_axes queries for aesthetics
    app.world
        .spawn(Aesthetics {
            identifiers: vec!["a".to_string(), "b".to_string(), "c".to_string()],
            condition: None,
        })
        .insert(Gy {})
        .insert(Distribution(vec![
            vec![1f32, 2., 2.],
            vec![1f32, 2., 1.],
            vec![6f32, 2., 6.],
        ]))
        .insert(AesFilter {
            met: false,
            pbox: false,
        })
        .insert(GeomHist::right(geom::HistPlot::Kde));
    // and for Paths with ArrowTag
    let path_builder = PathBuilder::new();
    let line = path_builder.build();
    app.world.spawn((
        GeometryBuilder::build_as(
            &line,
            DrawMode::Stroke(StrokeMode::new(
                Color::rgb(51. / 255., 78. / 255., 101. / 255.),
                10.0,
            )),
            Transform::from_xyz(1., 1., 1.),
        ),
        escher::ArrowTag {
            id: String::from("a"),
            hists: None,
            node_id: 9,
            direction: Vec2::new(0., 1.),
        },
        AesFilter {
            met: false,
            pbox: false,
        },
    ));

    let asset_server = setup("assets");
    app.insert_resource(asset_server);
    app.insert_resource(UiState::default());
    app.add_plugin(AesPlugin);
    app.update();

    // one update for xaxis creation
    assert!(app
        .world
        .query::<&Xaxis>()
        .iter(&app.world)
        .next()
        .is_some());

    // another update for HistTag creation
    app.update();
    assert!(app
        .world
        .query::<(&HistTag, &Path)>()
        .iter(&app.world)
        .next()
        .is_some());
}

#[test]
fn point_dist_aes_spaws_box_axis_spawns_box() {
    // Setup app
    let mut app = App::new();
    // build_axes queries for aesthetics
    app.world
        .spawn(Aesthetics {
            identifiers: vec!["a".to_string(), "b".to_string(), "c".to_string()],
            condition: None,
        })
        .insert(Gy {})
        .insert(Point(vec![1f32, 2., 2.]))
        .insert(AesFilter {
            met: false,
            pbox: true,
        })
        .insert(GeomHist::right(geom::HistPlot::Kde));
    // and for Paths with ArrowTag
    let path_builder = PathBuilder::new();
    let line = path_builder.build();
    app.world.spawn((
        GeometryBuilder::build_as(
            &line,
            DrawMode::Stroke(StrokeMode::new(
                Color::rgb(51. / 255., 78. / 255., 101. / 255.),
                10.0,
            )),
            Transform::from_xyz(1., 1., 1.),
        ),
        escher::ArrowTag {
            id: String::from("a"),
            hists: None,
            node_id: 9,
            direction: Vec2::new(0., 1.),
        },
        AesFilter {
            met: false,
            pbox: true,
        },
    ));

    app.insert_resource(UiState::default());
    app.insert_resource(AssetServer::new(FileAssetIo::new("asset1", false)));
    app.add_plugin(AesPlugin);
    app.update();

    assert!(app
        .world
        .query::<(&Xaxis, &Unscale)>()
        .iter(&app.world)
        .next()
        .is_some());

    // another update for HistTag creation
    app.update();

    assert!(app
        .world
        .query::<(&HistTag, &Unscale, &Path)>()
        .iter(&app.world)
        .next()
        .is_some());
}

#[test]
fn loading_file_drop_does_not_crash() {
    // Setup app
    let mut app = App::new();
    app.insert_resource(UiState::default());
    let asset_server = setup("assets");
    let escher_handle: Handle<escher::EscherMap> = asset_server.load("ecoli_core_map.json");
    app.insert_resource(data::ReactionState {
        reaction_data: None,
        reac_loaded: false,
        met_loaded: false,
    });
    app.insert_resource(asset_server);
    app.insert_resource(escher::MapState {
        escher_map: escher_handle,
        loaded: false,
    });
    app.add_stage_before(
        bevy::app::CoreStage::PreUpdate,
        bevy::asset::AssetStage::LoadAssets,
        SystemStage::parallel(),
    );
    app.add_stage_after(
        bevy::app::CoreStage::PostUpdate,
        bevy::asset::AssetStage::AssetEvents,
        SystemStage::parallel(),
    );
    app.add_event::<FileDragAndDrop>();
    app.add_plugin(data::DataPlugin);
    app.add_plugin(escher::EscherPlugin);
    app.add_system(file_drop);

    app.update();
    app.world.send_event(FileDragAndDrop::DroppedFile {
        id: WindowId::new(),
        path_buf: "assets/ecoli_core_map.json".into(),
    });
    app.update();
}