additional tests

Author: tijmenbruggeman

src/class-tiny-picture.php

@@ -379,7 +379,7 @@ protected function create_alternative_sources( $original_source_html ) {
 				}
 
 				$source_attr_parts['type'] = $mimetype;
-				$source_parts[] = '<source';
+				$source_parts = array( '<source' );
 				foreach ( $source_attr_parts as $source_attr_name => $source_attr_val ) {
 					$source_parts[] = $source_attr_name . '="' . $source_attr_val . '"';
 				}

test/unit/TinyPictureTest.php

@@ -191,6 +191,28 @@ public function test_picture_with_attributes() {
         $output = $this->tiny_picture->replace_sources($input);
 
         $this->assertEquals($expected, $output);
+    }
+
+    public function test_adds_both_avif_and_webp()
+    {
+        $this->wp->createImage(1000, '2025/01', 'test.webp');
+        $this->wp->createImage(1000, '2025/01', 'test.avif');
+
+        $input = '<img src="/wp-content/uploads/2025/01/test.png">';
+        $expected = '<picture><source srcset="/wp-content/uploads/2025/01/test.webp" type="image/webp" /><source srcset="/wp-content/uploads/2025/01/test.avif" type="image/avif" /><img src="/wp-content/uploads/2025/01/test.png"></picture>';
+        $output = $this->tiny_picture->replace_sources($input);
+
+        $this->assertEquals($expected, $output);
+    }
+
+    public function test_img_with_query_and_fragment_keeps_both()
+    {
+        $this->wp->createImage(37857, '2025/09', 'test.avif');
 
+        $input = '<img src="/wp-content/uploads/2025/09/test.png?v=123#top">';
+        $expected = '<picture><source srcset="/wp-content/uploads/2025/09/test.avif" type="image/avif" /><img src="/wp-content/uploads/2025/09/test.png?v=123#top"></picture>';
+        $output = $this->tiny_picture->replace_sources($input);
+
+        $this->assertEquals($expected, $output);
     }
 }