test: throw no permission when download failed (#1742)

packages/workspace/src/affine/__tests__/api.spec.ts

@@ -101,19 +101,20 @@ describe('api', () => {
     document.removeEventListener('affine-error', listener);
   });
 
-  test('blob too large', async () => {
+  test('no permission', async () => {
-    let called = false;
+    await workspaceApis.downloadWorkspace('not-exist').catch(e => {
-    try {
-      await workspaceApis.uploadBlob(
-        'test',
-        new ArrayBuffer(1024 * 1024 * 1024 + 1),
-        'image/png'
-      );
-    } catch (e) {
-      called = true;
       expect(e).toBeInstanceOf(RequestError);
-    }
+      expect(e.code).toBe(MessageCode.noPermission);
-    expect(called, 'throw error').toBe(true);
+    });
+  });
+
+  test('blob too large', async () => {
+    await workspaceApis
+      .uploadBlob('test', new ArrayBuffer(1024 * 1024 * 1024 + 1), 'image/png')
+      .catch(e => {
+        expect(e).toBeInstanceOf(RequestError);
+        expect(e.code).toBe(MessageCode.blobTooLarge);
+      });
   });
 
   test('refresh token', async () => {

packages/workspace/src/affine/api/index.ts

@@ -5,9 +5,12 @@ import { z } from 'zod';
 import { getLoginStorage } from '../login';
 
 export class RequestError extends Error {
+  public readonly code: MessageCode;
+
   constructor(code: MessageCode, cause: unknown | null = null) {
     super(Messages[code].message);
     sendMessage(code);
+    this.code = code;
     this.name = 'RequestError';
     this.cause = cause;
   }
@@ -360,8 +363,16 @@ export function createWorkspaceApis(prefixUrl = '/') {
             Authorization: auth.token,
           },
         })
+          .then(r =>
+            r.status === 403
+              ? Promise.reject(new RequestError(MessageCode.noPermission))
+              : r
+          )
           .then(r => r.arrayBuffer())
           .catch(e => {
+            if (e instanceof RequestError) {
+              throw e;
+            }
             throw new RequestError(MessageCode.downloadWorkspaceFailed, e);
           });
       }