merge ready

center-for-threat-informed-defense · Dec 14, 2023 · 3f2d7dc · 3f2d7dc
1 parent 7da2b12
commit 3f2d7dc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/app/repository/references-repository.js b/app/repository/references-repository.js
@@ -73,7 +73,7 @@ class ReferenceRepository {
 
     async create(data) {
         // Create the document
-        const reference = new Reference(data);
+        const reference = new this.model(data);
 
         // Save the document in the database
         try {

diff --git a/app/services/references-service.js b/app/services/references-service.js
@@ -9,19 +9,23 @@ class ReferencesService {
     }
 
     async retrieveAll(options) {
-        return this.repository.retrieveAll(options);
+        const res = await this.repository.retrieveAll(options);
+        return res;
     }
 
     async create(data) {
-        return this.repository.create(data);
+        const res = await this.repository.create(data);
+        return res;
     }
 
     async update(data) {
-        return this.repository.update(data);
+        const res = await this.repository.update(data);
+        return res;
     }
 
     async deleteBySourceName(sourceName) {
-        return this.repository.deleteBySourceName(sourceName);
+        const res = await this.repository.deleteBySourceName(sourceName);
+        return res;
     }
 
 }