test(iframe-plugin): implement tests for provideEmbeddedHeaders

jsamr · jsamr · commit 30613d34daad · 2021-10-12T11:03:43.000-03:00
diff --git a/packages/iframe-plugin/src/__tests__/IframeRenderer.test.tsx b/packages/iframe-plugin/src/__tests__/IframeRenderer.test.tsx
@@ -1,19 +1,25 @@
 import React from 'react';
 import HTML, { RenderHTMLProps } from 'react-native-render-html';
 import renderer from 'react-test-renderer';
-import WebView from 'react-native-webview';
-import IframeRenderer from '../IframeRenderer';
+import WebView from '@formidable-webview/ersatz';
+import IframeRenderer, { iframeModel } from '../IframeRenderer';
 
 describe('iframe renderer', () => {
+  const defaultConfig: Partial<RenderHTMLProps> = {
+    WebView,
+    renderers: {
+      iframe: IframeRenderer
+    },
+    customHTMLElementModels: {
+      iframe: iframeModel
+    },
+    contentWidth: 10
+  };
   it('should render without errors', () => {
     expect(() => {
       renderer.create(
         <HTML
-          WebView={WebView}
-          renderers={{
-            iframe: IframeRenderer
-          }}
-          contentWidth={10}
+          {...defaultConfig}
           source={{
             html: '<iframe width="300" height="300" src="https://google.com/" />'
           }}
@@ -27,6 +33,7 @@ describe('iframe renderer', () => {
         html: '<iframe width="300" height="300" src="https://google.com/" />'
       },
       provideEmbeddedHeaders: (uri, tagName, params) => {
+        //@ts-ignore
         if (tagName === 'iframe') {
           params;
           return {
@@ -35,16 +42,12 @@ describe('iframe renderer', () => {
         }
       }
     };
-    const rendered = renderer.create(
-      <HTML
-        WebView={WebView}
-        renderers={{
-          iframe: IframeRenderer
-        }}
-        contentWidth={10}
-        {...props}
-      />
-    );
-    expect(rendered.toJSON()).toMatchSnapshot();
+    const rendered = renderer.create(<HTML {...defaultConfig} {...props} />);
+
+    expect(rendered.root.findByType(WebView).props.source).toMatchObject({
+      headers: {
+        'X-Frame-Options': 'ALLOW-FROM https://google.com'
+      }
+    });
   });
 });
