@dhyasama/totem-models 2.3.1 → 2.3.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dhyasama/totem-models",
-  "version": "2.3.1",
+  "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();
+
+        });
+
+      });
+
+    });
+
+  });
+
+});