From 441a84b1f278bf3dd1a26ba91fdfd4020e26e567 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Te=C3=B3filo=20Monteiro?= <teofilo.monteiro@lego.com>
Date: Fri, 27 Sep 2024 12:15:48 +0200
Subject: [PATCH] update rbac tests

---
 src/tests/e2e/rbac.e2e.spec.ts | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/src/tests/e2e/rbac.e2e.spec.ts b/src/tests/e2e/rbac.e2e.spec.ts
index d9d46e1..d4d7826 100644
--- a/src/tests/e2e/rbac.e2e.spec.ts
+++ b/src/tests/e2e/rbac.e2e.spec.ts
@@ -47,6 +47,7 @@ test('Permission check e2e test', async (t) => {
 
     // verify list output
     const resources = await permit.api.resources.list();
+    t.true(Array.isArray(resources));
     t.is(resources.length, 1);
     t.is(resources[0].id, document.id);
     t.is(resources[0].key, document.key);
@@ -54,6 +55,13 @@ test('Permission check e2e test', async (t) => {
     t.is(resources[0].description, document.description);
     t.is(resources[0].urn, document.urn);
 
+    const resourcesWithTotalCount = await permit.api.resources.list({ includeTotalCount: true });
+    t.not(resourcesWithTotalCount, null);
+    t.not(resourcesWithTotalCount.data, null);
+    t.is(resourcesWithTotalCount.data.length, 1);
+    t.is(resourcesWithTotalCount.total_count, 1);
+    t.is(resourcesWithTotalCount.page_count, 1);
+
     // create admin role
     const admin = await permit.api.roles.create({
       key: 'admin',
@@ -84,6 +92,10 @@ test('Permission check e2e test', async (t) => {
     t.not(viewer.permissions, undefined);
     t.is(viewer.permissions?.length, 0);
 
+    const roles = await permit.api.roles.list();
+    t.true(Array.isArray(roles));
+    t.is(roles.length, 2);
+
     // assign permissions to roles
     const assignedViewer = await permit.api.roles.assignPermissions('viewer', ['document:read']);
 
@@ -234,7 +246,12 @@ test('Permission check e2e test', async (t) => {
       await permit.api.roles.delete('viewer');
       await permit.api.resources.delete('document');
       t.is((await permit.api.resources.list()).length, 0);
-      t.is((await permit.api.roles.list()).length, 0);
+
+      const roles = await permit.api.roles.list();
+
+      t.true(Array.isArray(roles));
+      t.is(roles.length, 0);
+
       t.is((await permit.api.tenants.list()).length, 1); // the default tenant
       t.is((await permit.api.users.list()).data.length, 0);
     } catch (error) {