chore: apply review suggestions

Author: r0b1n

packages/pluggableWidgets/image-web/src/utils/__tests__/getImageProps.spec.ts

@@ -81,7 +81,15 @@ describe("getImageProps", () => {
             expect(getImageProps(input)).toEqual({ type: "image", image: "https://example.com/image.jpg" });
         });
 
-        it("returns undefined image when imageUrl is loading", () => {
+        it("returns stale URL when imageUrl is loading with a cached value", () => {
+            const input: GetImagePropsInput = {
+                datasource: "imageUrl",
+                imageUrl: dynamic.loading("https://example.com/image.jpg")
+            };
+            expect(getImageProps(input)).toEqual({ type: "image", image: "https://example.com/image.jpg" });
+        });
+
+        it("returns undefined image when imageUrl is loading with no cached value", () => {
             const input: GetImagePropsInput = {
                 datasource: "imageUrl",
                 imageUrl: dynamic.loading<string>()

packages/pluggableWidgets/image-web/src/utils/getImageProps.ts

@@ -18,29 +18,29 @@ export function getImageProps({
 }: GetImagePropsInput): ImageType {
     switch (datasource) {
         case "image": {
-            // if main image is available or loading
             if (imageObject?.status === ValueStatus.Available || imageObject?.status === ValueStatus.Loading) {
                 return {
                     type: "image",
                     image: imageObject?.value?.uri
                 };
             }
 
-            // if main image is not available, but fallback is available or loading
             if (defaultImage?.status === ValueStatus.Available || defaultImage?.status === ValueStatus.Loading) {
                 return {
                     type: "image",
                     image: defaultImage?.value?.uri
                 };
             }
 
-            // if main image and fallback are not available
             return { type: "image", image: undefined };
         }
         case "imageUrl":
             return {
                 type: "image",
-                image: imageUrl?.status === ValueStatus.Available ? imageUrl.value : undefined
+                image:
+                    imageUrl?.status === ValueStatus.Available || imageUrl?.status === ValueStatus.Loading
+                        ? imageUrl.value
+                        : undefined
             };
         case "icon": {
             if (imageIcon?.status === ValueStatus.Available && imageIcon.value) {