From 2d3cccabfc84123e69169c0694aac37bd3ee30c3 Mon Sep 17 00:00:00 2001
From: Pranav C <pranavxc@gmail.com>
Date: Fri, 21 Oct 2022 22:03:58 +0530
Subject: [PATCH] test(api): add tests for new apis

Signed-off-by: Pranav C <pranavxc@gmail.com>
---
 packages/nc-gui/components/admin/Token.vue    |   2 +-
 packages/nc-gui/components/admin/User.vue     |  17 +-
 packages/nocodb/tests/unit/rest/index.test.ts |   4 +-
 .../nocodb/tests/unit/rest/tests/auth.test.ts |   4 +-
 .../nocodb/tests/unit/rest/tests/org.test.ts  | 173 ++++++++++++++++++
 5 files changed, 188 insertions(+), 12 deletions(-)
 create mode 100644 packages/nocodb/tests/unit/rest/tests/org.test.ts

diff --git a/packages/nc-gui/components/admin/Token.vue b/packages/nc-gui/components/admin/Token.vue
index 7dd2b95f6d..751ebe3430 100644
--- a/packages/nc-gui/components/admin/Token.vue
+++ b/packages/nc-gui/components/admin/Token.vue
@@ -139,7 +139,7 @@ const copyToken = (token: string | undefined) => {
         <!-- Token -->
         <a-table-column key="token" :title="$t('labels.token')" data-index="token">
           <template #default="{ text, record }">
-            <div>
+            <div class="w-[320px]">
               <span v-if="record.show">{{ text }}</span>
               <span v-else>****************************************</span>
             </div>
diff --git a/packages/nc-gui/components/admin/User.vue b/packages/nc-gui/components/admin/User.vue
index 2184413f17..fd323934b8 100644
--- a/packages/nc-gui/components/admin/User.vue
+++ b/packages/nc-gui/components/admin/User.vue
@@ -118,8 +118,8 @@ const copyInviteUrl = (user: User) => {
       </a-tab-pane>
     </a-tabs>
     <template v-if="selectedTabKey === 0">
-      <!--    <div class="text-xl mt-4">User Management</div> -->
-      <!--    <a-divider class="!my-3" /> -->
+      <div class="text-xl mt-4">User Management</div>
+      <a-divider class="!my-3" />
       <div class="max-w-[900px] mx-auto p-4">
         <div class="py-2 flex gap-4 items-center">
           <a-input-search
@@ -132,7 +132,7 @@ const copyInviteUrl = (user: User) => {
           >
           </a-input-search>
           <div class="flex-grow"></div>
-            <MdiReload @click="loadUsers" class="cursor-pointer"/>
+          <MdiReload @click="loadUsers" class="cursor-pointer" />
           <a-button size="small" @click="showUserModal = true">
             <div class="flex items-center gap-1">
               <MdiAdd />
@@ -176,14 +176,14 @@ const copyInviteUrl = (user: User) => {
                   <a-select-option :value="Role.OrgLevelCreator" :label="$t(`objects.roleType.orgLevelCreator`)">
                     <div>{{ $t(`objects.roleType.orgLevelCreator`) }}</div>
                     <span class="text-gray-500 text-xs whitespace-normal"
-                      >Creator can create new projects and access any invited project.</span
+                    >Creator can create new projects and access any invited project.</span
                     >
                   </a-select-option>
 
                   <a-select-option :value="Role.OrgLevelViewer" :label="$t(`objects.roleType.orgLevelViewer`)">
                     <div>{{ $t(`objects.roleType.orgLevelViewer`) }}</div>
                     <span class="text-gray-500 text-xs whitespace-normal"
-                      >Viewer is not allowed to create new projects but they can access any invited project.</span
+                    >Viewer is not allowed to create new projects but they can access any invited project.</span
                     >
                   </a-select-option>
                 </a-select>
@@ -207,7 +207,8 @@ const copyInviteUrl = (user: User) => {
               <div v-if="!record.roles.includes('super')" class="flex items-center gap-2">
                 <MdiDeleteOutline class="nc-action-btn cursor-pointer" @click="deleteUser(text)" />
 
-                <a-dropdown :trigger="['click']" class="flex" placement="bottomRight" overlay-class-name="nc-dropdown-user-mgmt">
+                <a-dropdown :trigger="['click']" class="flex" placement="bottomRight"
+                            overlay-class-name="nc-dropdown-user-mgmt">
                   <div class="flex flex-row items-center">
                     <a-button type="text" class="!px-0">
                       <div class="flex flex-row items-center h-[1.2rem]">
@@ -250,8 +251,8 @@ const copyInviteUrl = (user: User) => {
       </div>
     </template>
     <template v-else>
-      <!--      <div class="text-xl mt-4">Settings</div> -->
-      <!--      <a-divider class="!my-3" /> -->
+      <div class="text-xl mt-4">Settings</div>
+      <a-divider class="!my-3" />
       <a-form-item>
         <a-checkbox name="virtual">Enable user signup</a-checkbox>
       </a-form-item>
diff --git a/packages/nocodb/tests/unit/rest/index.test.ts b/packages/nocodb/tests/unit/rest/index.test.ts
index 2944f6395a..05ef09c84a 100644
--- a/packages/nocodb/tests/unit/rest/index.test.ts
+++ b/packages/nocodb/tests/unit/rest/index.test.ts
@@ -1,5 +1,6 @@
 import 'mocha';
 import authTests from './tests/auth.test';
+import orgTests from './tests/org.test';
 import projectTests from './tests/project.test';
 import tableTests from './tests/table.test';
 import tableRowTests from './tests/tableRow.test';
@@ -7,6 +8,7 @@ import viewRowTests from './tests/viewRow.test';
 
 function restTests() {
   authTests();
+  orgTests();
   projectTests();
   tableTests();
   tableRowTests();
@@ -15,4 +17,4 @@ function restTests() {
 
 export default function () {
   describe('Rest', restTests);
-}
\ No newline at end of file
+}
diff --git a/packages/nocodb/tests/unit/rest/tests/auth.test.ts b/packages/nocodb/tests/unit/rest/tests/auth.test.ts
index 4b0ae080b0..c809d829c5 100644
--- a/packages/nocodb/tests/unit/rest/tests/auth.test.ts
+++ b/packages/nocodb/tests/unit/rest/tests/auth.test.ts
@@ -4,7 +4,7 @@ import request from 'supertest';
 import init from '../../init';
 import { defaultUserArgs } from '../../factory/user';
 
-function authTests() {
+function orgTests() {
   let context;
 
   beforeEach(async function () {
@@ -74,7 +74,7 @@ function authTests() {
       .get('/api/v1/auth/user/me')
       .unset('xc-auth')
       .expect(200);
-      
+
       if (!response.body?.roles?.guest) {
         return new Error('User should be guest');
       }
diff --git a/packages/nocodb/tests/unit/rest/tests/org.test.ts b/packages/nocodb/tests/unit/rest/tests/org.test.ts
new file mode 100644
index 0000000000..fb45b20bf9
--- /dev/null
+++ b/packages/nocodb/tests/unit/rest/tests/org.test.ts
@@ -0,0 +1,173 @@
+import { expect } from 'chai'
+import 'mocha'
+import request from 'supertest'
+import { OrgUserRoles } from '../../../../src/enums/OrgUserRoles'
+import init from '../../init'
+
+function authTests() {
+  let context
+
+  beforeEach(async function() {
+    context = await init()
+  })
+
+  it('Get users list', async () => {
+    const response = await request(context.app)
+      .get('/api/v1/users')
+      .set('xc-auth', context.token)
+      .expect(200)
+
+    expect(response.body).to.have.keys(['list', 'pageInfo'])
+    expect(response.body.list).to.have.length(1)
+
+  })
+
+  it('Invite a new user', async () => {
+
+    const response = await request(context.app)
+      .post('/api/v1/users')
+      .set('xc-auth', context.token).send({ email: 'a@nocodb.com' })
+      .expect(200)
+
+    console.log(response.body)
+
+    expect(response.body).to.have.property('invite_token').to.be.a('string')
+// todo: verify invite token
+  })
+
+  it('Update user role', async () => {
+    const email = 'a@nocodb.com'
+    // invite a user
+    await request(context.app)
+      .post('/api/v1/users')
+      .set('xc-auth', context.token).send({ email })
+      .expect(200)
+    const response = await request(context.app)
+      .get('/api/v1/users')
+      .set('xc-auth', context.token)
+      .expect(200)
+    expect(response.body.list).to.have.length(2)
+
+    const user = response.body.list.find(u => u.email === email)
+
+    expect(user).to.have.property('roles').to.be.equal(OrgUserRoles.VIEWER)
+
+
+    await request(context.app)
+      .patch('/api/v1/users/' + user.id)
+      .set('xc-auth', context.token)
+      .send({ roles: OrgUserRoles.CREATOR })
+      .expect(200)
+
+
+    const response2 = await request(context.app)
+      .get('/api/v1/users')
+      .set('xc-auth', context.token)
+      .expect(200)
+    expect(response2.body.list).to.have.length(2)
+
+    const user2 = response2.body.list.find(u => u.email === email)
+
+    expect(user2).to.have.property('roles').to.be.equal(OrgUserRoles.CREATOR)
+  })
+
+  it('Remove user', async () => {
+    const email = 'a@nocodb.com'
+    // invite a user
+    await request(context.app)
+      .post('/api/v1/users')
+      .set('xc-auth', context.token).send({ email })
+      .expect(200)
+
+    const response = await request(context.app)
+      .get('/api/v1/users')
+      .set('xc-auth', context.token)
+      .expect(200)
+    expect(response.body.list).to.have.length(2)
+
+    const user = response.body.list.find(u => u.email === email)
+
+    expect(user).to.have.property('roles').to.be.equal(OrgUserRoles.VIEWER)
+
+
+    await request(context.app)
+      .delete('/api/v1/users/' + user.id)
+      .set('xc-auth', context.token)
+      .expect(200)
+
+
+    const response2 = await request(context.app)
+      .get('/api/v1/users')
+      .set('xc-auth', context.token)
+      .expect(200)
+    expect(response2.body.list).to.have.length(1)
+
+  })
+
+
+  it('Get token list', async () => {
+    const response = await request(context.app)
+      .get('/api/v1/tokens')
+      .set('xc-auth', context.token)
+      .expect(200)
+
+    expect(response.body).to.have.keys(['list', 'pageInfo'])
+    expect(response.body.list).to.have.length(0)
+
+  })
+
+  it('Generate token', async () => {
+    const r = await request(context.app)
+      .post('/api/v1/tokens')
+      .set('xc-auth', context.token)
+      .send({ description: 'test' })
+      .expect(200)
+
+    const response = await request(context.app)
+      .get('/api/v1/tokens')
+      .set('xc-auth', context.token)
+      .expect(200)
+
+    expect(response.body).to.have.keys(['list', 'pageInfo'])
+    expect(response.body.list).to.have.length(1)
+    expect(response.body.list[0]).to.have.property('token').to.be.a('string')
+    expect(response.body.list[0]).to.have.property('description').to.be.a('string').to.be.eq('test')
+
+  })
+
+  it.only('Delete token', async () => {
+    const r = await request(context.app)
+      .post('/api/v1/tokens')
+      .set('xc-auth', context.token)
+      .send({ description: 'test' })
+      .expect(200)
+
+    let response = await request(context.app)
+      .get('/api/v1/tokens')
+      .set('xc-auth', context.token)
+      .expect(200)
+
+    expect(response.body).to.have.keys(['list', 'pageInfo'])
+    expect(response.body.list).to.have.length(1)
+
+    await request(context.app)
+      .delete('/api/v1/tokens/' + r.body.token)
+      .set('xc-auth', context.token)
+      .expect(200)
+
+
+    response = await request(context.app)
+      .get('/api/v1/tokens')
+      .set('xc-auth', context.token)
+      .expect(200)
+
+    expect(response.body).to.have.keys(['list', 'pageInfo'])
+    expect(response.body.list).to.have.length(0)
+
+  })
+
+}
+
+export default function() {
+  describe('Organisation', authTests)
+}