From c84186916e916e4ce9db5e5774e5ae8117f89259 Mon Sep 17 00:00:00 2001
From: Andrew <andrew@digerati.co.nz>
Date: Tue, 29 Nov 2022 17:14:46 +1300
Subject: [PATCH] Verify bKGD in tests

---
 tests/regression.rs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/tests/regression.rs b/tests/regression.rs
index de63039f..69b11463 100644
--- a/tests/regression.rs
+++ b/tests/regression.rs
@@ -62,7 +62,13 @@ fn test_it_converts(
         "optimized to wrong bit depth"
     );
     if let Some(palette) = png.raw.palette.as_ref() {
-        assert!(palette.len() <= 1 << (png.raw.ihdr.bit_depth.as_u8() as usize));
+        let mut max_palette_size = 1 << (png.raw.ihdr.bit_depth.as_u8() as usize);
+        // Ensure bKGD color is valid
+        if let Some(&idx) = png.raw.aux_headers.get(b"bKGD").and_then(|b| b.first()) {
+            assert!(palette.len() > idx as usize);
+            max_palette_size = max_palette_size.max(idx as usize + 1);
+        }
+        assert!(palette.len() <= max_palette_size);
     } else {
         assert_ne!(png.raw.ihdr.color_type, ColorType::Indexed);
     }