fix!: client should not throw on 404 (#751)

web3-storage · Dec 6, 2021 · a4cce7e · a4cce7e
1 parent 84f55e1
commit a4cce7e
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 22 deletions.
diff --git a/packages/api/test/mocks/pgrest/get_user.json b/packages/api/test/mocks/pgrest/get_user.json
@@ -1,4 +1,4 @@
-{
+[{
   "id": 1,
   "issuer": "issuer-str",
   "keys": [
@@ -8,4 +8,4 @@
       "secret": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiJ0ZXN0IiwiaXNzIjoid2ViMy1zdG9yYWdlIiwiaWF0IjoxNjMzOTU3Mzg5ODcyLCJuYW1lIjoidGVzdCJ9.KEH0jHUfJls44YWsj8uex_zj0dUIvdyqGalv2rhWnx8"
     }
   ]
-}
+}]
diff --git a/packages/db/index.js b/packages/db/index.js
@@ -70,10 +70,10 @@ export class DBClient {
    * Get user by its issuer.
    *
    * @param {string} issuer
-   * @return {Promise<import('./db-client-types').UserOutput>}
+   * @return {Promise<import('./db-client-types').UserOutput | undefined>}
    */
   async getUser (issuer) {
-    /** @type {{ data: import('./db-client-types').UserOutput, error: PostgrestError }} */
+    /** @type {{ data: import('./db-client-types').UserOutput[], error: PostgrestError }} */
     const { data, error } = await this._client
       .from('user')
       .select(`
@@ -87,13 +87,12 @@ export class DBClient {
         updated:updated_at
       `)
       .eq('issuer', issuer)
-      .single()
 
     if (error) {
       throw new DBError(error)
     }
 
-    return data
+    return data.length ? data[0] : undefined
   }
 
   /**
@@ -614,7 +613,7 @@ export class DBClient {
    *
    * @param {string} issuer
    * @param {string} secret
-   * @return {Promise<import('./db-client-types').AuthKey>}
+   * @return {Promise<import('./db-client-types').AuthKey | undefined>}
    */
   async getKey (issuer, secret) {
     /** @type {{ data, error: PostgrestError } */
@@ -630,22 +629,26 @@ export class DBClient {
       })
       .filter('keys.deleted_at', 'is', null)
       .eq('keys.secret', secret)
-      .single()
 
     if (error) {
       throw new DBError(error)
     }
 
-    if (!data.keys.length) {
-      throw new Error('user has no key with given secret')
+    if (!data.length) {
+      return undefined
+    }
+
+    const keyData = data[0]
+    if (!keyData.keys.length) {
+      return undefined
     }
 
     return {
-      _id: data.keys[0]._id,
-      name: data.keys[0].name,
+      _id: keyData.keys[0]._id,
+      name: keyData.keys[0].name,
       user: {
-        _id: data._id,
-        issuer: data.issuer
+        _id: keyData._id,
+        issuer: keyData.issuer
       }
     }
   }

diff --git a/packages/db/test/user.spec.js b/packages/db/test/user.spec.js
@@ -42,14 +42,9 @@ describe('user operations', () => {
     assert.strictEqual(user.publicAddress, publicAddress, 'user has correct public address')
   })
 
-  it('should fail to get a non existing user', async () => {
-    try {
-      await client.getUser('fake-issuer')
-    } catch (err) {
-      assert(err, 'errored to get a non existing user')
-      return
-    }
-    throw new Error('should fail to get a non existing user')
+  it('should return undefined to get a non existing user', async () => {
+    const user = await client.getUser('fake-issuer')
+    assert.strictEqual(user, undefined)
   })
 
   it('should update user with same issuer (login)', async () => {
@@ -88,6 +83,13 @@ describe('user operations', () => {
     assert.strictEqual(updatedUser.created, user.created, 'user has same created timestamp')
   })
 
+  it('should return undefined to get non existing key for user', async () => {
+    const secret = 'test-secret-fail'
+    const fetchedKey = await client.getKey(user.issuer, secret)
+
+    assert.strictEqual(fetchedKey, undefined)
+  })
+
   it('can create auth keys for user', async () => {
     const name = 'test-key-name'
     const secret = 'test-secret'