test: Even more forking tests

Bluefinger · Oct 18, 2023 · d9ba2d8 · d9ba2d8
1 parent 39516ab
commit d9ba2d8
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 2 deletions.
diff --git a/src/component.rs b/src/component.rs
@@ -233,7 +233,7 @@ where
 #[cfg(test)]
 mod tests {
     use bevy::reflect::TypePath;
-    use bevy_prng::ChaCha8Rng;
+    use bevy_prng::{ChaCha8Rng, ChaCha12Rng};
 
     use super::*;
 
@@ -249,6 +249,18 @@ mod tests {
         );
     }
 
+    #[test]
+    fn forking_as() {
+        let mut rng1 = EntropyComponent::<ChaCha12Rng>::default();
+
+        let rng2 = rng1.fork_as::<ChaCha8Rng>();
+
+        let rng1 = format!("{:?}", rng1);
+        let rng2 = format!("{:?}", rng2);
+
+        assert_ne!(&rng1, &rng2, "forked EntropyComponents should not match each other");
+    }
+
     #[test]
     fn forking_inner() {
         let mut rng1 = EntropyComponent::<ChaCha8Rng>::default();

diff --git a/src/resource.rs b/src/resource.rs
@@ -162,7 +162,7 @@ where
 #[cfg(test)]
 mod tests {
     use bevy::reflect::TypePath;
-    use bevy_prng::ChaCha8Rng;
+    use bevy_prng::{ChaCha8Rng, ChaCha12Rng};
 
     use super::*;
 
@@ -192,6 +192,18 @@ mod tests {
         assert_ne!(source_val, forked_val);
     }
 
+    #[test]
+    fn forking_as() {
+        let mut rng1 = GlobalEntropy::<ChaCha12Rng>::default();
+
+        let rng2 = rng1.fork_as::<ChaCha8Rng>();
+
+        let rng1 = format!("{:?}", rng1);
+        let rng2 = format!("{:?}", rng2);
+
+        assert_ne!(&rng1, &rng2, "GlobalEntropy should not match the forked component");
+    }
+
     #[test]
     fn forking_inner() {
         let mut rng1 = GlobalEntropy::<ChaCha8Rng>::default();