@dhyasama/totem-models 2.3.0 → 2.3.2

package/lib/Financials.js

@@ -10,7 +10,6 @@
 module.exports = function(mongoose, config) {
 
   var Schema = mongoose.Schema;
-  var Round = mongoose.model('Round');
   var async = require('async');
   var _ = require('underscore');
 
@@ -93,31 +92,7 @@ module.exports = function(mongoose, config) {
       'customer': config.CUSTOMER_ID
     });
 
-    query.exec(function(err, rounds) {
-
-      if (err) return cb(err, null);
-      if (!rounds || rounds.length == 0) return cb(null, []);
-
-      var error = checkForUnpopulatedData(rounds, 'Round.getByOrg');
-      if (error) return cb(error, null);
-
-      // reverse chronological
-      rounds = _.sortBy(rounds, function(r) {
-
-        var date = r.closingDate;
-
-        // use investment date in absence of closing date
-        if (!date && r.vehicles.length && r.vehicles[0].investments.length) {
-          date = r.vehicles[0].investments[0].investmentDate;
-        }
-
-        return date;
-
-      }).reverse();
-
-      return cb(null, rounds);
-
-    });
+    query.exec(cb);
 
   };
 

package/lib/Organization.js

@@ -1209,15 +1209,15 @@ module.exports = function(mongoose, config) {
 
   };
 
-  Organization.statics.modify = function(orgId, update, cb) {
+  Organization.statics.modify = function(filter, update, cb) {
 
     if (!cb) throw new Error('cb  is required');
-    if (!orgId) { return cb(new Error('orgId is required'), null); }
+    if (!filter) { return cb(new Error('filter is required'), null); }
     if (!update) { return cb(new Error('update is required'), null); }
 
     var self = this;
 
-    self.findOneAndUpdate({ _id: orgId }, update, { upsert: false, new: true }, cb);
+    self.findOneAndUpdate(filter, update, { upsert: false, new: true }, cb);
 
   };
 

package/package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "@dhyasama/totem-models",
-  "version": "2.2.3",
+  "version": "2.3.1",
   "lockfileVersion": 1,
   "requires": true,
   "dependencies": {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dhyasama/totem-models",
-  "version": "2.3.0",
+  "version": "2.3.2",
   "author": "Jason Reynolds",
   "license": "UNLICENSED",
   "description": "Models for Totem platform",

package/test/Financials.js

@@ -0,0 +1,151 @@
+"use strict";
+
+var
+
+  should = require("should"),
+  _ = require('underscore'),
+  config = require('./config')['test'],
+  mongoose = require('mongoose'),
+  clearDB = require('mocha-mongoose')(config.db.uri, {noClear: true}),
+  Financials = mongoose.model('Financials');
+
+var customerA = new mongoose.Types.ObjectId();
+var customerB = new mongoose.Types.ObjectId();
+var orgA = new mongoose.Types.ObjectId();
+var orgB = new mongoose.Types.ObjectId();
+
+describe('Financials', function() {
+
+  before(function(done) {
+    if (mongoose.connection.db) return done();
+    mongoose.connect(config.db.uri, done);
+  });
+
+  before(function(done) {
+    clearDB(done);
+  });
+
+  it('creates a financial document', function(done) {
+
+    var f = new Financials({
+      organization: orgA,
+      customer: customerA,
+      asOfDate: new Date(),
+      runway: 1
+    });
+
+    Financials.upsert(f, function(err, result) {
+      should.not.exist(err);
+      should.exist(result);
+      done();
+    });
+
+  });
+
+  it('creates more financial documents', function(done) {
+
+    var f = new Financials({
+      organization: orgB,
+      customer: customerA,
+      asOfDate: new Date(),
+      runway: 2
+    });
+
+    Financials.upsert(f, function(err, result) {
+
+      should.not.exist(err);
+      should.exist(result);
+
+      var f = new Financials({
+        organization: orgB,
+        customer: customerB,
+        asOfDate: new Date(),
+        runway: 3
+      });
+
+      Financials.upsert(f, function(err, result) {
+
+        should.not.exist(err);
+        should.exist(result);
+
+        var f = new Financials({
+          organization: orgA,
+          customer: customerB,
+          asOfDate: new Date(),
+          runway: 4
+        });
+
+        Financials.upsert(f, function(err, result) {
+          should.not.exist(err);
+          should.exist(result);
+          done();
+        });
+
+      });
+
+    });
+
+  });
+
+  it('gets financials for an org', function(done) {
+
+    config.CUSTOMER_ID = customerA;
+
+    Financials.getByOrg(orgA, function(err, result) {
+      should.not.exist(err);
+      should.exist(result);
+      result.length.should.equal(1);
+      result[0].runway.should.equal(1);
+      done();
+    });
+
+  });
+
+  it('gets financials for the same org but different customer', function(done) {
+
+    config.CUSTOMER_ID = customerB;
+
+    Financials.getByOrg(orgA, function(err, result) {
+      should.not.exist(err);
+      should.exist(result);
+      result.length.should.equal(1);
+      result[0].runway.should.equal(4);
+      done();
+    });
+
+  });
+
+  it('removes financials for a customer', function(done) {
+
+    config.CUSTOMER_ID = customerA;
+
+    Financials.find({}, function(err, result) {
+
+      should.not.exist(err);
+      should.exist(result);
+      result.length.should.equal(4);
+
+      Financials.removeCustomerFinancials(config.CUSTOMER_ID, function(err, result) {
+
+        should.not.exist(err);
+        should.exist(result);
+
+        Financials.find({}, function(err, result) {
+
+          should.not.exist(err);
+          should.exist(result);
+          result.length.should.equal(2);
+          result[0].runway.should.equal(3);
+          result[1].runway.should.equal(4);
+
+          done();
+
+        });
+
+      });
+
+    });
+
+  });
+
+});