[Tests] Add a test for BZip2 truncated in the middle of a Huffman symbol

tsolomko · Jan 20, 2022 · 939f08a · 939f08a
1 parent f539a53
commit 939f08a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/Tests/BZip2Tests.swift b/Tests/BZip2Tests.swift
@@ -67,6 +67,13 @@ class BZip2Tests: XCTestCase {
         XCTAssertThrowsError(try BZip2.decompress(data: testData))
     }
 
+    func testBadFile_truncated() throws {
+        // This tests that encountering data truncated in the middle of a Huffman symbol correctly throws an error
+        // (and doesn't crash).
+        let testData = try Constants.data(forTest: "test1", withType: BZip2Tests.testType)[0...40]
+        XCTAssertThrowsError(try BZip2.decompress(data: testData))
+    }
+
     func testEmptyData() throws {
         XCTAssertThrowsError(try BZip2.decompress(data: Data()))
     }