From 8533f2df7d8945b1c66e3c985906d0f9be21bca9 Mon Sep 17 00:00:00 2001
From: Peter Velkov <kidroca@gmail.com>
Date: Fri, 3 Feb 2023 15:26:33 +0200
Subject: [PATCH] [test] Image: test headers related functionality

---
 .../__snapshots__/index-test.js.snap          |   8 +-
 .../src/exports/Image/__tests__/index-test.js | 103 +++++++++++++++---
 2 files changed, 90 insertions(+), 21 deletions(-)

diff --git a/packages/react-native-web/src/exports/Image/__tests__/__snapshots__/index-test.js.snap b/packages/react-native-web/src/exports/Image/__tests__/__snapshots__/index-test.js.snap
index 04b41e3efa..e0fe801e7e 100644
--- a/packages/react-native-web/src/exports/Image/__tests__/__snapshots__/index-test.js.snap
+++ b/packages/react-native-web/src/exports/Image/__tests__/__snapshots__/index-test.js.snap
@@ -329,14 +329,14 @@ exports[`components/Image prop "style" removes other unsupported View styles 1`]
 >
   <div
     class="css-view-175oi2r r-backgroundColor-1niwhzg r-backgroundPosition-vvn4in r-backgroundRepeat-u6sd8q r-bottom-1p0dtai r-height-1pi2tsx r-left-1d2f490 r-position-u8s1d r-right-zchlnj r-top-ipm5af r-width-13qz1uu r-zIndex-1wyyakw r-backgroundSize-4gszlv"
-    style="filter: url(#tint-57);"
+    style="filter: url(#tint-72);"
   />
   <svg
     style="position: absolute; height: 0px; visibility: hidden; width: 0px;"
   >
     <defs>
       <filter
-        id="tint-57"
+        id="tint-72"
       >
         <feflood
           flood-color="blue"
@@ -378,7 +378,7 @@ exports[`components/Image prop "style" supports "tintcolor" property (convert to
 >
   <div
     class="css-view-175oi2r r-backgroundColor-1niwhzg r-backgroundPosition-vvn4in r-backgroundRepeat-u6sd8q r-bottom-1p0dtai r-height-1pi2tsx r-left-1d2f490 r-position-u8s1d r-right-zchlnj r-top-ipm5af r-width-13qz1uu r-zIndex-1wyyakw r-backgroundSize-4gszlv"
-    style="background-image: url(https://google.com/favicon.ico); filter: url(#tint-56);"
+    style="background-image: url(https://google.com/favicon.ico); filter: url(#tint-71);"
   />
   <img
     alt=""
@@ -391,7 +391,7 @@ exports[`components/Image prop "style" supports "tintcolor" property (convert to
   >
     <defs>
       <filter
-        id="tint-56"
+        id="tint-71"
       >
         <feflood
           flood-color="red"
diff --git a/packages/react-native-web/src/exports/Image/__tests__/index-test.js b/packages/react-native-web/src/exports/Image/__tests__/index-test.js
index 3be0434e2e..c87cf1e7ee 100644
--- a/packages/react-native-web/src/exports/Image/__tests__/index-test.js
+++ b/packages/react-native-web/src/exports/Image/__tests__/index-test.js
@@ -13,7 +13,7 @@ import Image from '../';
 import ImageLoader, { ImageUriCache } from '../../../modules/ImageLoader';
 import PixelRatio from '../../PixelRatio';
 import React from 'react';
-import { render } from '@testing-library/react';
+import { render, waitFor } from '@testing-library/react';
 
 const originalImage = window.Image;
 
@@ -21,6 +21,16 @@ describe('components/Image', () => {
   beforeEach(() => {
     ImageUriCache._entries = {};
     window.Image = jest.fn(() => ({}));
+    ImageLoader.load = jest
+      .fn()
+      .mockImplementation((source, onLoad, onError) => {
+        act(() => onLoad({ source }));
+      });
+    ImageLoader.loadWithHeaders = jest.fn().mockImplementation((source) => ({
+      source,
+      promise: Promise.resolve(`blob:${Math.random()}`),
+      cancel: jest.fn()
+    }));
   });
 
   afterEach(() => {
@@ -106,10 +116,6 @@ describe('components/Image', () => {
 
   describe('prop "onLoad"', () => {
     test('is called after image is loaded from network', () => {
-      jest.useFakeTimers();
-      ImageLoader.load = jest.fn().mockImplementation((_, onLoad, onError) => {
-        onLoad();
-      });
       const onLoadStartStub = jest.fn();
       const onLoadStub = jest.fn();
       const onLoadEndStub = jest.fn();
@@ -121,15 +127,10 @@ describe('components/Image', () => {
           source="https://test.com/img.jpg"
         />
       );
-      jest.runOnlyPendingTimers();
       expect(onLoadStub).toBeCalled();
     });
 
     test('is called after image is loaded from cache', () => {
-      jest.useFakeTimers();
-      ImageLoader.load = jest.fn().mockImplementation((_, onLoad, onError) => {
-        onLoad();
-      });
       const onLoadStartStub = jest.fn();
       const onLoadStub = jest.fn();
       const onLoadEndStub = jest.fn();
@@ -143,7 +144,6 @@ describe('components/Image', () => {
           source={uri}
         />
       );
-      jest.runOnlyPendingTimers();
       expect(onLoadStub).toBeCalled();
       ImageUriCache.remove(uri);
     });
@@ -227,6 +227,34 @@ describe('components/Image', () => {
     });
   });
 
+  describe('prop "onLoadStart"', () => {
+    test('is called on update if "headers" are modified', () => {
+      const onLoadStartStub = jest.fn();
+      const { rerender } = render(
+        <Image
+          onLoadStart={onLoadStartStub}
+          source={{
+            uri: 'https://test.com/img.jpg',
+            headers: { 'x-custom-header': 'abc123' }
+          }}
+        />
+      );
+      act(() => {
+        rerender(
+          <Image
+            onLoadStart={onLoadStartStub}
+            source={{
+              uri: 'https://test.com/img.jpg',
+              headers: { 'x-custom-header': '123abc' }
+            }}
+          />
+        );
+      });
+
+      expect(onLoadStartStub.mock.calls.length).toBe(2);
+    });
+  });
+
   describe('prop "resizeMode"', () => {
     ['contain', 'cover', 'none', 'repeat', 'stretch', undefined].forEach(
       (resizeMode) => {
@@ -245,7 +273,8 @@ describe('components/Image', () => {
         '',
         {},
         { uri: '' },
-        { uri: 'https://google.com' }
+        { uri: 'https://google.com' },
+        { uri: 'https://google.com', headers: { 'x-custom-header': 'abc123' } }
       ];
       sources.forEach((source) => {
         expect(() => render(<Image source={source} />)).not.toThrow();
@@ -261,11 +290,6 @@ describe('components/Image', () => {
 
     test('is set immediately if the image was preloaded', () => {
       const uri = 'https://yahoo.com/favicon.ico';
-      ImageLoader.load = jest
-        .fn()
-        .mockImplementationOnce((_, onLoad, onError) => {
-          onLoad();
-        });
       return Image.prefetch(uri).then(() => {
         const source = { uri };
         const { container } = render(<Image source={source} />, {
@@ -346,6 +370,51 @@ describe('components/Image', () => {
         'http://localhost/static/img@2x.png'
       );
     });
+
+    test('it works with headers in 2 stages', async () => {
+      const uri = 'https://google.com/favicon.ico';
+      const headers = { 'x-custom-header': 'abc123' };
+      const source = { uri, headers };
+
+      // Stage 1
+      const loadRequest = {
+        promise: Promise.resolve('blob:123'),
+        cancel: jest.fn(),
+        source
+      };
+
+      ImageLoader.loadWithHeaders.mockReturnValue(loadRequest);
+
+      render(<Image source={source} />);
+
+      expect(ImageLoader.loadWithHeaders).toHaveBeenCalledWith(
+        expect.objectContaining(source)
+      );
+
+      // Stage 2
+      return waitFor(() => {
+        expect(ImageLoader.load).toHaveBeenCalledWith(
+          'blob:123',
+          expect.any(Function),
+          expect.any(Function)
+        );
+      });
+    });
+
+    // A common case is `source` declared as an inline object, which cause is to be a
+    // new object (with the same content) each time parent component renders
+    test('it still loads the image if source object is changed', () => {
+      const uri = 'https://google.com/favicon.ico';
+      const headers = { 'x-custom-header': 'abc123' };
+      const { rerender } = render(<Image source={{ uri, headers }} />);
+      rerender(<Image source={{ uri, headers }} />);
+
+      // when the underlying source didn't change we don't expect more than 1 load calls
+      return waitFor(() => {
+        expect(ImageLoader.loadWithHeaders).toHaveBeenCalledTimes(1);
+        expect(ImageLoader.load).toHaveBeenCalledTimes(1);
+      });
+    });
   });
 
   describe('prop "style"', () => {