110 lines
3.0 KiB
Rust
110 lines
3.0 KiB
Rust
use bevy::ecs::system::RunSystemOnce;
|
|
use pomomon_garden::features::grid::components::{Grid, TileState};
|
|
use pomomon_garden::features::inventory::components::ItemType;
|
|
use pomomon_garden::features::notification::components::Notifications;
|
|
use pomomon_garden::features::pom::actions::InteractionAction;
|
|
use pomomon_garden::features::pom::components::{GridPosition, InteractionTarget, PathQueue, Pom};
|
|
use pomomon_garden::features::pom::perform_interaction;
|
|
use pomomon_garden::prelude::*;
|
|
|
|
mod common;
|
|
use common::setup_app;
|
|
|
|
#[test]
|
|
fn test_prevent_trapping_plant() {
|
|
let occupied = TileState::Occupied {
|
|
seed: ItemType::BerrySeed {
|
|
name: "Wall".into(),
|
|
},
|
|
watered: false,
|
|
growth_stage: 0,
|
|
withered: false,
|
|
dry_counter: 0,
|
|
};
|
|
|
|
let mut app = setup_app(
|
|
2,
|
|
2,
|
|
&[
|
|
(0, 0, TileState::Empty), // Pom
|
|
(1, 0, TileState::Empty), // Target
|
|
(0, 1, occupied.clone()),
|
|
(1, 1, occupied.clone()),
|
|
],
|
|
vec![(
|
|
ItemType::BerrySeed {
|
|
name: "Test".into(),
|
|
},
|
|
10,
|
|
)],
|
|
None,
|
|
);
|
|
|
|
app.init_resource::<Notifications>();
|
|
|
|
app.world_mut().spawn((
|
|
Pom,
|
|
GridPosition { x: 0, y: 0 },
|
|
InteractionTarget {
|
|
target: Some((1, 0)),
|
|
action: Some(InteractionAction::Plant(ItemType::BerrySeed {
|
|
name: "Test".into(),
|
|
})),
|
|
},
|
|
PathQueue::default(),
|
|
));
|
|
|
|
let _ = app.world_mut().run_system_once(perform_interaction);
|
|
|
|
let grid = app.world().resource::<Grid>();
|
|
let tile_entity = grid.get_tile((1, 0)).unwrap();
|
|
let tile_state = app.world().entity(tile_entity).get::<TileState>().unwrap();
|
|
|
|
// Should remain Empty because planting was blocked
|
|
match tile_state {
|
|
TileState::Empty => (),
|
|
_ => panic!("Should have prevented planting! State is {:?}", tile_state),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_allow_safe_plant() {
|
|
let mut app = setup_app(
|
|
10,
|
|
10,
|
|
&[(0, 0, TileState::Empty), (1, 0, TileState::Empty)],
|
|
vec![(
|
|
ItemType::BerrySeed {
|
|
name: "Test".into(),
|
|
},
|
|
10,
|
|
)],
|
|
None,
|
|
);
|
|
app.init_resource::<Notifications>();
|
|
|
|
app.world_mut().spawn((
|
|
Pom,
|
|
GridPosition { x: 0, y: 0 },
|
|
InteractionTarget {
|
|
target: Some((1, 0)),
|
|
action: Some(InteractionAction::Plant(ItemType::BerrySeed {
|
|
name: "Test".into(),
|
|
})),
|
|
},
|
|
PathQueue::default(),
|
|
));
|
|
|
|
let _ = app.world_mut().run_system_once(perform_interaction);
|
|
|
|
let grid = app.world().resource::<Grid>();
|
|
let tile_entity = grid.get_tile((1, 0)).unwrap();
|
|
let tile_state = app.world().entity(tile_entity).get::<TileState>().unwrap();
|
|
|
|
// Should be Occupied
|
|
match tile_state {
|
|
TileState::Occupied { .. } => (),
|
|
_ => panic!("Should have allowed planting! State is {:?}", tile_state),
|
|
}
|
|
}
|