added .revenue to Apartment

abarnhard · Jul 25, 2014 · e36f4ca · e36f4ca
1 parent 5f2135c
commit e36f4ca
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/app/models/apartment.js b/app/models/apartment.js
@@ -123,6 +123,18 @@ Apartment.tenants = function(cb){
   });
 };
 
+Apartment.revenue = function(cb){
+  Apartment.find({}, function(apts){
+    var sum = 0;
+    for(var i = 0; i < apts.length; i++){
+      if(apts[i].renters.length > 0){
+        sum += apts[i].cost();
+      }
+    }
+    cb(sum);
+  });
+};
+
 // HELPER FUNCTIONS //
 
 function reProto(apt){

diff --git a/test/unit/apartment.js b/test/unit/apartment.js
@@ -247,6 +247,14 @@ describe('Apartment', function(){
       });
     });
   });
+  describe('.revenue', function(){
+    it('should return the total revenue from all occupied apartments', function(done){
+      Apartment.revenue(function(revenue){
+        expect(revenue).to.equal(4500);
+        done();
+      });
+    });
+  });
 
 });