From 7d2d2ff4102de0a5ed4d651c8bfd398a3d589f3c Mon Sep 17 00:00:00 2001
From: Hongbo Wu <hongbo@omnivore.app>
Date: Thu, 14 Jul 2022 17:47:48 +0800
Subject: [PATCH] fix api key test

---
 packages/api/test/resolvers/api_key.test.ts | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/packages/api/test/resolvers/api_key.test.ts b/packages/api/test/resolvers/api_key.test.ts
index 18e12e775..11452fdf4 100644
--- a/packages/api/test/resolvers/api_key.test.ts
+++ b/packages/api/test/resolvers/api_key.test.ts
@@ -3,6 +3,8 @@ import { createTestUser, deleteTestUser } from '../db'
 import { graphqlRequest, request } from '../util'
 import { expect } from 'chai'
 import supertest from 'supertest'
+import { getRepository } from '../../src/entity/utils'
+import { ApiKey } from '../../src/entity/api_key'
 
 const testAPIKey = (apiKey: string): supertest.Test => {
   const query = `
@@ -155,6 +157,8 @@ describe('Api Key resolver', () => {
   })
 
   describe('get api keys', () => {
+    let apiKeys = [] as ApiKey[]
+
     before(async () => {
       name = 'test-get-api-keys'
       query = `
@@ -178,8 +182,10 @@ describe('Api Key resolver', () => {
       }
     `
 
-      const response = await graphqlRequest(query, authToken)
-      apiKeyId = response.body.data.generateApiKey.apiKey.id
+      apiKeys = await getRepository(ApiKey).find({
+        select: ['id', 'name'],
+        where: { user: { id: user.id } },
+      })
     })
 
     it('should get api keys', async () => {
@@ -190,8 +196,6 @@ describe('Api Key resolver', () => {
             apiKeys {
               id
               name
-              expiresAt
-              usedAt
             }
           }
           ... on ApiKeysError {
@@ -203,9 +207,7 @@ describe('Api Key resolver', () => {
 
       const response = await graphqlRequest(query, authToken).expect(200)
       expect(response.body.data.apiKeys.apiKeys).to.be.an('array')
-      expect(response.body.data.apiKeys.apiKeys[0].id).to.eql(apiKeyId)
-      expect(response.body.data.apiKeys.apiKeys[0].name).to.eql(name)
-      expect(response.body.data.apiKeys.apiKeys[0].usedAt).to.be.null
+      expect(response.body.data.apiKeys.apiKeys).to.eql(apiKeys)
     })
   })
 })