Browse Source

add shareLink object

itizawa 5 years ago
parent
commit
0f3c1a93a2
1 changed files with 17 additions and 16 deletions
  1. 17 16
      src/test/models/shareLink.test.js

+ 17 - 16
src/test/models/shareLink.test.js

@@ -38,11 +38,18 @@ describe('ShareLink', () => {
       render: (page) => { return page },
     };
 
+    const shareLink = {
+      relatedPage: 'relatedPageId',
+    };
+
     test('share link is not found', async() => {
 
+      shareLink.populate = () => { return null };
+
       jest.spyOn(ShareLink, 'findOne').mockImplementation(() => {
-        return { populate: () => { return null } };
+        return shareLink;
       });
+
       const response = await Page.showSharedPage(req, res);
 
       expect(response).toEqual('layout-growi/not_found_shared_page');
@@ -50,8 +57,10 @@ describe('ShareLink', () => {
 
     test('share link is found, but it does not have Page', async() => {
 
+      shareLink.populate = () => { return { _id: 'somePageId' } };
+
       jest.spyOn(ShareLink, 'findOne').mockImplementation(() => {
-        return { populate: () => { return { _id: 'somePageId' } } };
+        return shareLink;
       });
       const response = await Page.showSharedPage(req, res);
 
@@ -61,14 +70,10 @@ describe('ShareLink', () => {
 
     test('share link is found, but it is expired', async() => {
 
+      shareLink.populate = () => { return { _id: 'somePageId', relatedPage, isExpired: () => { return true } } };
+
       jest.spyOn(ShareLink, 'findOne').mockImplementation(() => {
-        return {
-          populate: () => {
-            return {
-              _id: 'somePageId', relatedPage, isExpired: () => { return true },
-            };
-          },
-        };
+        return shareLink;
       });
 
       const response = await Page.showSharedPage(req, res);
@@ -78,14 +83,10 @@ describe('ShareLink', () => {
 
     test('share link is found, and it has the page you can see', async() => {
 
+      shareLink.populate = () => { return { _id: 'somePageId', relatedPage, isExpired: () => { return false } } };
+
       jest.spyOn(ShareLink, 'findOne').mockImplementation(() => {
-        return {
-          populate: () => {
-            return {
-              _id: 'somePageId', relatedPage, isExpired: () => { return false },
-            };
-          },
-        };
+        return shareLink;
       });
       const response = await Page.showSharedPage(req, res);