Fix tests

Author: haydenbleasel

packages/streamdown/__tests__/image.test.tsx

@@ -78,6 +78,9 @@ describe("ImageComponent", () => {
       />
     );
 
+    const img = container.querySelector('img[data-streamdown="image"]');
+    if (img) fireEvent.load(img);
+
     const button = container.querySelector('button[title="Download image"]');
     expect(button).toBeTruthy();
   });
@@ -112,6 +115,9 @@ describe("ImageComponent", () => {
       />
     );
 
+    const img = container.querySelector('img[data-streamdown="image"]');
+    if (img) fireEvent.load(img);
+
     const button = container.querySelector('button[title="Download image"]');
     expect(button).toBeTruthy();
     if (button) {
@@ -139,6 +145,9 @@ describe("ImageComponent", () => {
       />
     );
 
+    const img = container.querySelector('img[data-streamdown="image"]');
+    if (img) fireEvent.load(img);
+
     const button = container.querySelector('button[title="Download image"]');
     expect(button).toBeTruthy();
     if (button) {
@@ -168,6 +177,9 @@ describe("ImageComponent", () => {
       />
     );
 
+    const img = container.querySelector('img[data-streamdown="image"]');
+    if (img) fireEvent.load(img);
+
     const button = container.querySelector('button[title="Download image"]');
     expect(button).toBeTruthy();
     if (button) {
@@ -207,6 +219,9 @@ describe("ImageComponent", () => {
         />
       );
 
+      const img = container.querySelector('img[data-streamdown="image"]');
+      if (img) fireEvent.load(img);
+
       const button = container.querySelector('button[title="Download image"]');
       expect(button).toBeTruthy();
       if (button) {
@@ -238,6 +253,9 @@ describe("ImageComponent", () => {
       />
     );
 
+    const img = container.querySelector('img[data-streamdown="image"]');
+    if (img) fireEvent.load(img);
+
     const button = container.querySelector('button[title="Download image"]');
     expect(button).toBeTruthy();
     if (button) {
@@ -265,6 +283,9 @@ describe("ImageComponent", () => {
       <ImageComponent node={null as any} src="https://example.com/" />
     );
 
+    const img = container.querySelector('img[data-streamdown="image"]');
+    if (img) fireEvent.load(img);
+
     const button = container.querySelector('button[title="Download image"]');
     expect(button).toBeTruthy();
     if (button) {
@@ -290,6 +311,9 @@ describe("ImageComponent", () => {
       />
     );
 
+    const img = container.querySelector('img[data-streamdown="image"]');
+    if (img) fireEvent.load(img);
+
     const button = container.querySelector('button[title="Download image"]');
     expect(button).toBeTruthy();
     if (button) {
@@ -340,6 +364,9 @@ describe("ImageComponent", () => {
       />
     );
 
+    const img = container.querySelector('img[data-streamdown="image"]');
+    if (img) fireEvent.load(img);
+
     const button = container.querySelector('button[title="Download image"]');
     expect(button).toBeTruthy();
     if (button) {