EAGLESIX-2892 added DepsTracker unit tests

lib/pipeline/DepsTracker.js

@@ -14,7 +14,7 @@ var DepsTracker = module.exports = function DepsTracker() {
     this.needTextScore = false;
 }, klass = DepsTracker, base = Object, proto = klass.prototype = Object.create(base.prototype, {constructor:{value:klass}});
 
-var ParsedDeps = require("./ParsedDeps.js");
+var ParsedDeps = require("./ParsedDeps");
 
 /**
  * Returns a projection object covering the dependencies tracked by this class.
@@ -78,7 +78,7 @@ proto.toProjection = function toProjection() {
  * @return {ParsedDeps}
  */
 proto.toParsedDeps = function toParsedDeps() {
-    var doc;
+    var doc = {};
 
     if(this.needWholeDocument || this.needTextScore) {
         // can't use ParsedDeps in this case

test/lib/pipeline/DepsTracker.js

@@ -0,0 +1,90 @@
+"use strict";
+var assert = require("assert"),
+    DepsTracker = require("../../../lib/pipeline/DepsTracker");
+
+module.exports = {
+    "DepsTracker": {
+        "toProjection()": {
+            "should be able to convert dependencies to a projection": function(){
+                var deps = new DepsTracker(),
+                    expected = {"_id":0,"a":1,"b":1};
+                deps.fields = {'a':1,'b':1};
+
+                assert.deepEqual(expected, deps.toProjection());
+            },
+            "should be able to convert dependencies with subfields to a projection": function(){
+                var deps = new DepsTracker(),
+                    expected = {"_id":0,"a":1};
+                deps.fields = {'a':1,'a.b':1};
+
+                assert.deepEqual(expected, deps.toProjection());
+            },
+            "should be able to convert dependencies with _id to a projection": function(){
+                var deps = new DepsTracker(),
+                    expected = {"a":1,"b":1,"_id":1};
+                deps.fields = {"_id":1,'a':1,'b':1};
+
+                assert.deepEqual(expected, deps.toProjection());
+            },
+            "should be able to convert dependencies with id and subfields to a projection": function(){
+                var deps = new DepsTracker(),
+                    expected = {"_id":1,"b":1};
+                deps.fields = {'_id.a':1,'b':1};
+
+                assert.deepEqual(expected, deps.toProjection());
+            },
+            "should return empty object if needWholeDocument is true": function() {
+                var deps = new DepsTracker(),
+                    expected = {};
+                deps.needWholeDocument = true;
+
+                assert.deepEqual(expected, deps.toProjection());
+            },
+            "should return $noFieldsNeeded if there are no dependencies": function() {
+                var deps = new DepsTracker(),
+                    expected = {"_id":0,"$noFieldsNeeded":1};
+
+                assert.deepEqual(expected, deps.toProjection());
+            }
+        },
+        "toParsedDeps()": {
+            "should not parse if needWholeDocument is true": function() {
+                var deps = new DepsTracker(),
+                    expected;
+                deps.needWholeDocument = true;
+
+                assert.deepEqual(expected, deps.toParsedDeps());
+            },
+            "should not parse if needTextScore is true": function() {
+                var deps = new DepsTracker(),
+                    expected;
+                deps.needTextScore = true;
+
+                assert.deepEqual(expected, deps.toParsedDeps());
+            },
+            "should be able to parse dependencies": function() {
+                var deps = new DepsTracker(),
+                    expected = {"_fields":{"a":true,"b":true}};
+                deps.fields = {'a':1,'b':1};
+
+                assert.deepEqual(expected, deps.toParsedDeps());
+            },
+            "should be able to parse dependencies with subfields": function() {
+                var deps = new DepsTracker(),
+                    expected = {"_fields":{"a":true}};
+                deps.fields = {'a':1,'a.b':1};
+
+                assert.deepEqual(expected, deps.toParsedDeps());
+            },
+            "should be able to parse dependencies with arrays": function() {
+                var deps = new DepsTracker(),
+                    expected = {"_fields":{"a":true,"b":true}};
+                deps.fields = {'a':1,'b':[{'c':1},{'d':'d.e'}]};
+
+                assert.deepEqual(expected, deps.toParsedDeps());
+            }
+        }
+    }
+};
+
+if (!module.parent)(new(require("mocha"))()).ui("exports").reporter("spec").addFile(__filename).run();