Registration fixes and improvements.

tests/common/mod.rs

@@ -1,4 +1,4 @@
-pub mod data_getter;
+mod data_getter;
 pub mod library_loader;
 pub mod test_case;
 pub mod test_step;

tests/main.rs

@@ -9,7 +9,11 @@ use std::sync::Arc;
 
 use conquer_once::OnceCell;
 
-use pkmn_lib::dynamic_data::DynamicLibrary;
+use pkmn_lib::dynamic_data::{
+    Battle, BattleParty, DamageSource, DynamicLibrary, ExecutingMove, MoveChoice, PokemonBuilder, PokemonParty, Script,
+    ScriptCategory, ScriptContainer, ScriptOwnerData, TurnChoice, VolatileScriptsOwner,
+};
+use pkmn_lib::script_implementations::wasm::script::WebAssemblyScript;
 
 use crate::common::{library_loader, TestCase};
 
@@ -50,3 +54,65 @@ fn integration_tests(input: &Path) {
     println!("\tRunning integration test {}", test_case.name);
     test_case.run_test(get_library());
 }
+
+/// Assurance has the interesting properties that it creates a special data script internally, and
+/// deletes that data script through the get_owner functionality.
+#[test]
+#[cfg_attr(miri, ignore)]
+fn validate_assurance() {
+    let lib = get_library();
+    let p1 = Arc::new(
+        PokemonBuilder::new(lib.clone(), "charizard".into(), 100)
+            .learn_move("assurance".into())
+            .build(),
+    );
+    let p2 = Arc::new(PokemonBuilder::new(lib.clone(), "venusaur".into(), 100).build());
+
+    let party1 = BattleParty::new(
+        Arc::new(PokemonParty::new_from_vec(vec![Some(p1.clone())])),
+        vec![(0, 0)],
+    );
+    let party2 = BattleParty::new(
+        Arc::new(PokemonParty::new_from_vec(vec![Some(p2.clone())])),
+        vec![(1, 0)],
+    );
+
+    let battle = Battle::new(lib.clone(), vec![party1, party2], false, 2, 1, None);
+
+    battle.sides()[0].set_pokemon(0, Some(p1.clone()));
+    battle.sides()[1].set_pokemon(0, Some(p2.clone()));
+
+    let script = lib
+        .load_script(ScriptOwnerData::None, ScriptCategory::Move, &"assurance".into())
+        .unwrap()
+        .unwrap();
+
+    let mv = p1.learned_moves().read()[0].as_ref().unwrap().clone();
+    let choice = TurnChoice::Move(MoveChoice::new(p1.clone(), mv.clone(), 1, 0));
+    script.on_before_turn(&choice);
+    assert!(battle.sides()[1].has_volatile_script(&"assurance_data".into()));
+
+    let executing_move = ExecutingMove::new(
+        vec![],
+        1,
+        p1.clone(),
+        mv.clone(),
+        mv.move_data().clone(),
+        ScriptContainer::default(),
+    );
+    let mut v = 20_u8;
+    script.change_base_power(&executing_move, &p2, 0, &mut v);
+    assert_eq!(v, 20_u8);
+
+    let s = battle.sides()[1].get_volatile_script(&"assurance_data".into());
+    let data_script = s.as_ref().unwrap().get_as::<WebAssemblyScript>();
+
+    data_script.on_damage(&p2, DamageSource::Misc, 100, 50);
+
+    let mut v = 20_u8;
+    script.change_base_power(&executing_move, &p2, 0, &mut v);
+    assert_eq!(v, 40_u8);
+
+    data_script.on_end_turn();
+    assert!(!battle.sides()[1].has_volatile_script(&"assurance_data".into()));
+}