fix: tests to reference new names from model

Greenstand · Feb 3, 2022 · fbf0b01 · fbf0b01
1 parent bdd8a20
commit fbf0b01
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 7 deletions.
diff --git a/server/models/Stakeholder.js b/server/models/Stakeholder.js
@@ -596,4 +596,6 @@ module.exports = {
   getNonRelations,
   createRelation,
   deleteRelation,
+  StakeholderTree,
+  FilterCriteria
 };
diff --git a/server/models/Stakeholder.spec.js b/server/models/Stakeholder.spec.js
@@ -1,14 +1,14 @@
 const { expect } = require('chai');
 const sinon = require('sinon');
 const {
-  getStakeholders,
-  Stakeholder,
+  getAllStakeholders,
+  StakeholderTree,
   FilterCriteria,
 } = require('./Stakeholder');
 
 describe('Stakeholder Model', () => {
   it('Stakeholder Model should return defined fields', () => {
-    const stakeholder = Stakeholder({});
+    const stakeholder = StakeholderTree({});
     expect(stakeholder).to.have.keys([
       'id',
       'type',
@@ -29,6 +29,8 @@ describe('Stakeholder Model', () => {
       'map',
       'owner_id',
       'organization_id',
+      'parents',
+      'children'
     ]);
   });
 
@@ -57,11 +59,11 @@ describe('Stakeholder Model', () => {
     });
   });
 
-  describe('getStakeholders', () => {
+  describe('getAllStakeholders', () => {
     it('should get stakeholders with filter --id', async () => {
       const getFilterById = sinon.mock();
       const getStakeholderByOrganizationId = sinon.mock();
-      const executeGetStakeholders = getStakeholders({
+      const executeGetStakeholders = getAllStakeholders({
         getFilterById,
         getStakeholderByOrganizationId,
       });
@@ -86,7 +88,7 @@ describe('Stakeholder Model', () => {
     it('should get stakeholders with filter --organization_id', async () => {
       const getStakeholderByOrganizationId = sinon.mock();
       const getFilterById = sinon.mock();
-      const executeGetStakeholders = getStakeholders({
+      const executeGetStakeholders = getAllStakeholders({
         getStakeholderByOrganizationId,
         getFilterById,
       });

diff --git a/server/repositories/StakeholderRepository.js b/server/repositories/StakeholderRepository.js
@@ -92,7 +92,7 @@ class StakeholderRepository extends BaseRepository {
     return { stakeholders: results, count: +count[0].count };
   }
 
-  // not currently being used
+  // not currently being used but may be useful later
   async getStakeholderTreeById(id, options) {
     const stakeholder = await this._session
       .getDB()(this._tableName)