Add archetype rechecking when adding an untyped archetype invariant

Author: sixfold-origami

crates/bevy_ecs/src/world/archetype_invariants.rs

@@ -402,6 +402,27 @@ mod tests {
         world.add_archetype_invariant(ArchetypeInvariant::<(A, B, C)>::atomic());
     }
 
+    #[test]
+    fn on_insert_untyped_happy() {
+        let mut world = World::new();
+
+        world.spawn().insert_bundle((A, B, C));
+        let archetype_invariant =
+            ArchetypeInvariant::<(A, B, C)>::atomic().into_untyped(&mut world);
+        world.add_untyped_archetype_invariant(archetype_invariant);
+    }
+
+    #[test]
+    #[should_panic]
+    fn on_insert_untyped_sad() {
+        let mut world = World::new();
+
+        world.spawn().insert_bundle((A, B));
+        let archetype_invariant =
+            ArchetypeInvariant::<(A, B, C)>::atomic().into_untyped(&mut world);
+        world.add_untyped_archetype_invariant(archetype_invariant);
+    }
+
     #[test]
     fn forbids_happy() {
         let mut world = World::new();

crates/bevy_ecs/src/world/mod.rs

@@ -679,6 +679,11 @@ impl World {
         &mut self,
         archetype_invariant: UntypedArchetypeInvariant,
     ) {
+        for archetype in &self.archetypes.archetypes {
+            let components = archetype.components.indices();
+            archetype_invariant.test_archetype(components);
+        }
+
         self.archetype_invariants.add(archetype_invariant);
     }