瀏覽代碼

WIP: implement certifySharedFileMiddleware

Yuki Takei 2 年之前
父節點
當前提交
28b29b8a29
共有 1 個文件被更改,包括 17 次插入1 次删除
  1. 17 1
      apps/app/src/server/middlewares/certify-shared-file/certify-shared-file.spec.ts

+ 17 - 1
apps/app/src/server/middlewares/certify-shared-file/certify-shared-file.spec.ts

@@ -29,8 +29,24 @@ describe('certifySharedFileMiddleware', () => {
       certifySharedFileMiddleware(req, res, next);
 
       // then
-      expect(next).toHaveBeenCalledOnce();
       expect(mocks.validateRefererMock).not.toHaveBeenCalled();
+      expect(next).toHaveBeenCalledOnce();
     });
+
+    it('when validateReferer returns null', () => {
+      // setup
+      const req = mock<RequestToAllowShareLink>();
+      req.params = { id: 'file id string' };
+      req.headers = { referer: 'referer string' };
+
+      // when
+      certifySharedFileMiddleware(req, res, next);
+
+      // then
+      expect(mocks.validateRefererMock).toHaveBeenCalledOnce();
+      expect(mocks.validateRefererMock).toHaveBeenCalledWith('referer string');
+      expect(next).toHaveBeenCalledOnce();
+    });
+
   });
 });