fix: dynamic variable in query

MikeIbberson · Aug 27, 2020 · daaaa66 · daaaa66
1 parent be8dc63
commit daaaa66
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/lib/__tests__/comparison.test.js b/lib/__tests__/comparison.test.js
@@ -58,6 +58,11 @@ describe('Comparison', () => {
   });
 
   describe('query', () => {
+    it('should return truthy on dynamic matching', () =>
+      expect(
+        new Comparison(['dynamic={{age}}']).query(stub),
+      ).toMatchObject({}));
+
     it('should return as mongo query', () =>
       expect(new Comparison(exp).query(stub)).toMatchObject({
         $and: [

diff --git a/lib/index.js b/lib/index.js
@@ -104,9 +104,21 @@ class Comparison {
       .filter(hasLength);
   }
 
-  query() {
+  query(obj = {}) {
     return {
-      $and: this.eligible.reduce((a, [key, value, fn]) => {
+      $and: this.eligible.reduce((a, params) => {
+        let key;
+        let value;
+        let fn;
+
+        if (Array.isArray(params)) {
+          [key, value, fn] = params;
+        } else if (isFn(params)) {
+          [key, value, fn] = params(obj);
+        } else {
+          return a;
+        }
+
         const v = isSerialized(value) ? value.split(',') : value;
         const name = Array.isArray(v) ? 'in' : fn.name;