@blaze-cms/nextjs-tools 0.118.0 → 0.119.0

package/tests/unit/src/components/DebugSidebar/DebugSidebar.test.js

@@ -1,6 +1,6 @@
 import '@testing-library/jest-dom/extend-expect';
 import React from 'react';
-import { MockedProvider } from '@apollo/react-testing';
+import { MockedProvider } from '@apollo/client/testing';
 import { render, act, fireEvent } from '@testing-library/react';
 import { DebugSidebar } from '../../../../../src/components';
 

package/tests/unit/src/containers/ContentContainer.test.js

@@ -1,7 +1,7 @@
 import '@testing-library/jest-dom/extend-expect';
 import React from 'react';
 
-import { MockedProvider } from '@apollo/react-testing';
+import { MockedProvider } from '@apollo/client/testing';
 import { render } from '@testing-library/react';
 import { buildPBComponents, getLightboxImages } from '@blaze-cms/plugin-page-builder-fe';
 import { ContentContainer } from '../../../../src/containers';

package/tests/unit/src/helpers/check-url.test.js

@@ -49,7 +49,8 @@ describe('Check url', () => {
     statusCode,
     writeHead: jest.fn(),
     write: jest.fn(),
-    end: jest.fn()
+    end: jest.fn(),
+    setHeader: jest.fn()
   };
   const asPath = '/page-slug';
   const urlTo = '/redirect-url';
@@ -115,6 +116,24 @@ describe('Check url', () => {
     expect(handleStaticRoutes.getHandler).not.toHaveBeenCalled();
   });
 
+  it('should set empty Cache-Control header', async () => {
+    await checkUrl(props);
+    expect(res.setHeader).toHaveBeenCalledWith('Cache-Control', '');
+  });
+
+  it('should set Cache-Control header from env', async () => {
+    const cacheEnvs = {
+      BLAZE_CACHE_CONTROL_HEADER: 'max-age=0'
+    };
+    setEnvs(cacheEnvs);
+    await checkUrl(props);
+    expect(res.setHeader).toHaveBeenCalledWith(
+      'Cache-Control',
+      cacheEnvs.BLAZE_CACHE_CONTROL_HEADER
+    );
+    deleteEnvs(cacheEnvs);
+  });
+
   it('should return disableSsr prop', async () => {
     getAccessToken.mockReturnValueOnce(true);
     const result = await checkUrl({ ...props });

package/tests/unit/src/helpers/static-route-handlers/generic-file-handler.test.js

@@ -1,4 +1,4 @@
-import gql from 'graphql-tag';
+import { gql } from '@apollo/client';
 import getHandler from '../../../../../src/helpers/static-route-handlers/generic-file-handler';
 import { getStaticFileHandlerTests } from '../../../../helpers/tests/static-file-handler-tests';
 

package/tests/unit/src/hoc/withBlaze.test.js

@@ -11,12 +11,14 @@ jest.mock('@blaze-cms/core-ui', () => {
       extract: jest.fn(() => ({ cache: true }))
     }
   };
+
   const mockBlaze = {
     ctx: { req: {}, res: {} },
     initComplete: jest.fn(() => true),
     init: jest.fn(async () => true),
     getClient: jest.fn(() => null),
     createClient: jest.fn(() => apolloClient),
+    getProcessedEntities: jest.fn(() => ({ typePolicyQueryFields: {}, possibleTypes: {} })),
     events: {
       emit: jest.fn()
     },
@@ -37,7 +39,15 @@ describe('withBlaze HOC', () => {
   const props = {
     test: true
   };
-  const expectedProps = [{ ...props, apolloClient: blazeMock.createClient(), apolloState: {} }, {}];
+  const expectedProps = [
+    {
+      ...props,
+      apolloClient: blazeMock.createClient(),
+      apolloState: {},
+      processedBlazeEntities: undefined
+    },
+    {}
+  ];
 
   const Component = 'Component';
   const AppTree = jest.fn(() => null);
@@ -67,6 +77,7 @@ describe('withBlaze HOC', () => {
   it('should wrap component and use buildConfig function', async () => {
     const AppWithBlaze = withBlaze(App);
     const { asFragment, queryByText } = render(<AppWithBlaze {...props} />);
+
     await waitFor(() => expect(queryByText(appString)).toBeDefined());
 
     expect(App).toHaveBeenCalledWith(...expectedProps);
@@ -100,7 +111,12 @@ describe('withBlaze HOC', () => {
     expect(buildConfig).toHaveBeenCalledWith(blazeConfig);
 
     const initialProps = await AppWithBlaze.getInitialProps(context);
-    expect(initialProps).toEqual({ ...appProps, pageProps: {}, apolloState: { cache: true } });
+    expect(initialProps).toEqual({
+      ...appProps,
+      pageProps: {},
+      apolloState: { cache: true },
+      processedBlazeEntities: blazeMock.getProcessedEntities()
+    });
     expect(App.getInitialProps).toHaveBeenCalledWith(context);
     const [firstApprTreeCall] = AppTree.mock.calls;
     expect(firstApprTreeCall[0]).toEqual({
@@ -110,6 +126,7 @@ describe('withBlaze HOC', () => {
       apolloClient: blazeMock.createClient()
     });
     expect(blazeMock.init).toHaveBeenCalledTimes(3);
+    expect(blazeMock.getProcessedEntities).toHaveBeenCalledTimes(2);
   });
 
   it('should add getInitialProps and not render in client', async () => {