@blaze-cms/nextjs-tools 0.124.0-alpha.16 → 0.124.0-alpha.19

package/tests/unit/src/application/query/index.test.js

@@ -0,0 +1,19 @@
+const {
+  checkUrlQuery,
+  getPageQuery,
+  GET_ENTITY_SCHEMA
+} = require('../../../../../src/application/query');
+
+describe('Application query', () => {
+  it('should get check url query', () => {
+    expect(checkUrlQuery).toMatchSnapshot();
+  });
+
+  it('should get page query', () => {
+    expect(getPageQuery({})).toMatchSnapshot();
+  });
+
+  it('should match get entity schema', () => {
+    expect(GET_ENTITY_SCHEMA).toMatchSnapshot();
+  });
+});

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

@@ -74,6 +74,8 @@ describe('Check url', () => {
   const envs = {
     BLAZE_ROOT_SELECTOR_CLASSES_LIMIT: rootSelectorEnvValue
   };
+  const mockHandlerResponse = () => jest.fn(() => true);
+  const cacheControlHeader = 'Cache-Control';
 
   beforeAll(() => {
     setEnvs(envs);
@@ -105,7 +107,7 @@ describe('Check url', () => {
   });
 
   it('should handle static route', async () => {
-    handleStaticRoutes.getHandler.mockImplementationOnce(() => jest.fn(() => true));
+    handleStaticRoutes.getHandler.mockImplementationOnce(mockHandlerResponse);
     const result = await checkUrl({ ...props, asPath: ROUTE_PATTERN_SITEMAP });
     expect(handleStaticRoutes.getHandler).toHaveBeenCalledWith(ROUTE_PATTERN_SITEMAP);
     expect(result).toEqual(true);
@@ -117,18 +119,20 @@ describe('Check url', () => {
   });
 
   it('should set empty Cache-Control header', async () => {
+    handleStaticRoutes.getHandler.mockImplementationOnce(mockHandlerResponse);
     await checkUrl(props);
-    expect(res.setHeader).toHaveBeenCalledWith('Cache-Control', '');
+    expect(res.setHeader).toHaveBeenCalledWith(cacheControlHeader, '');
   });
 
   it('should set Cache-Control header from env', async () => {
+    handleStaticRoutes.getHandler.mockImplementationOnce(mockHandlerResponse);
     const cacheEnvs = {
       BLAZE_CACHE_CONTROL_HEADER: 'max-age=0'
     };
     setEnvs(cacheEnvs);
     await checkUrl(props);
     expect(res.setHeader).toHaveBeenCalledWith(
-      'Cache-Control',
+      cacheControlHeader,
       cacheEnvs.BLAZE_CACHE_CONTROL_HEADER
     );
     deleteEnvs(cacheEnvs);
@@ -205,6 +209,31 @@ describe('Check url', () => {
     expect(blazeApp.events.emit).toHaveBeenCalled();
     apolloQueryCheck(apolloClient, rootPage);
   });
+
+  it('should set max age in cache control header from entity', async () => {
+    const cacheControlMaxAge = 1000;
+    apolloClient.query.mockImplementationOnce(async () => ({
+      data: {
+        checkUrl: { urlTo, pageData: { cacheControlMaxAge } }
+      }
+    }));
+    await checkUrl(props);
+    expect(res.setHeader).toHaveBeenCalledWith(cacheControlHeader, `max-age=${cacheControlMaxAge}`);
+  });
+
+  it('should not set cache control header if cacheControlMaxAge=null', async () => {
+    const cacheControlMaxAge = null;
+    apolloClient.query.mockImplementationOnce(async () => ({
+      data: {
+        checkUrl: { urlTo, pageData: { cacheControlMaxAge } }
+      }
+    }));
+    await checkUrl(props);
+    expect(res.setHeader).not.toHaveBeenCalledWith(
+      cacheControlHeader,
+      `max-age=${cacheControlMaxAge}`
+    );
+  });
 });
 
 function apolloQueryCheck(apolloClient, asPath) {