Просмотр исходного кода

WIP: implement certifySharedFileMiddleware

Yuki Takei 2 лет назад
Родитель
Сommit
28b29b8a29

+ 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();
+    });
+
   });
 });