Add unit tests for bundle packing

Author: yungcomputerchair

src/bundle.rs

@@ -73,7 +73,7 @@ fn align<T: Into<usize> + From<usize>>(value: T, alignment: T) -> T {
     aligned.into()
 }
 
-#[derive(Debug)]
+#[derive(Debug, PartialEq, Eq)]
 struct LevelEnds {
     compressed_end: u32,
     uncompressed_end: u32,
@@ -85,7 +85,7 @@ const EXPECTED_PLAYER_VERSION: &str = "fusion-2.x.x";
 const EXPECTED_ENGINE_VERSION_BASE: &str = "2";
 const DEFAULT_ENGINE_VERSION: &str = "2.5.4b5";
 
-#[derive(Debug)]
+#[derive(Debug, PartialEq, Eq)]
 pub struct AssetBundleHeader {
     signature: String,
     stream_version: u32,
@@ -335,6 +335,12 @@ impl std::fmt::Display for LevelFile {
         }
     }
 }
+impl PartialEq for LevelFile {
+    fn eq(&self, other: &Self) -> bool {
+        self.name == other.name && self.data == other.data
+    }
+}
+impl Eq for LevelFile {}
 impl LevelFile {
     fn new(name: String, data: Vec<u8>) -> Self {
         Self {
@@ -349,6 +355,19 @@ impl LevelFile {
 struct Level {
     files: Vec<LevelFile>,
 }
+impl PartialEq for Level {
+    fn eq(&self, other: &Self) -> bool {
+        // Files may be out of order, that's fine
+        for file in &self.files {
+            let other_file = other.files.iter().find(|f| f.name == file.name);
+            if other_file.is_none_or(|f| f.data != file.data) {
+                return false;
+            }
+        }
+        true
+    }
+}
+impl Eq for Level {}
 impl Level {
     fn read<R: Read + BufRead>(reader: &mut R) -> Result<Self, Error> {
         let mut reader = Counter::new(BufReader::new(get_lzma_decoder(reader)?));
@@ -435,7 +454,7 @@ impl Level {
     }
 }
 
-#[derive(Debug)]
+#[derive(Debug, PartialEq, Eq)]
 pub struct AssetBundle {
     levels: Vec<Level>,
 }

src/tests.rs

@@ -86,10 +86,12 @@ async fn test_generate_manifest() {
 #[cfg(feature = "lzma")]
 #[tokio::test]
 async fn test_extract_bundle() {
+    use crate::bundle::AssetBundle;
+
     let bundle_path = "example_builds/compressed/good/Map_00_00.unity3d";
     let output_dir = TempDir::new();
 
-    let (_, bundle) = crate::bundle::AssetBundle::from_file(bundle_path).unwrap();
+    let (_, bundle) = AssetBundle::from_file(bundle_path).unwrap();
     bundle.extract_files(output_dir.path()).unwrap();
 
     let version = Version::from_manifest_file("example_manifest.json").unwrap();
@@ -100,3 +102,31 @@ async fn test_extract_bundle() {
         .unwrap();
     assert!(corrupted.is_empty());
 }
+
+#[cfg(feature = "lzma")]
+#[tokio::test]
+async fn test_repack_bundle() {
+    use crate::bundle::AssetBundle;
+
+    let bundle_path = "example_builds/compressed/good/Map_00_00.unity3d";
+    let output_dir = TempDir::new();
+
+    let (_, og_bundle) = AssetBundle::from_file(bundle_path).unwrap();
+    og_bundle.extract_files(output_dir.path()).unwrap();
+
+    let repacked_bundle = AssetBundle::from_directory(output_dir.path()).unwrap();
+    assert!(og_bundle == repacked_bundle);
+}
+
+#[cfg(feature = "lzma")]
+#[tokio::test]
+async fn test_pack_bundle() {
+    use crate::bundle::AssetBundle;
+
+    let bundle_path = "example_builds/compressed/good/Map_00_00.unity3d";
+    let unpacked_path = "example_builds/uncompressed/good/map_5f00_5f00_2eunity3d";
+
+    let (_, og_bundle) = AssetBundle::from_file(bundle_path).unwrap();
+    let packed_bundle = AssetBundle::from_directory(unpacked_path).unwrap();
+    assert!(og_bundle == packed_bundle);
+}