Moved function to trim a date to the month only.

FlukeFan · Jan 9, 2012 · 09431a5 · 09431a5
1 parent 1d98d25
commit 09431a5
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 11 deletions.
diff --git a/Source/OfflineExample/Scripts/Offline/StorageService.js b/Source/OfflineExample/Scripts/Offline/StorageService.js
@@ -38,24 +38,15 @@ StorageService.prototype.execute = function (cmd, callback) {
     });
 }
 
-// TODO: add this to the Date prototype for cleaner code
-function toYearMonth(date) {
-    var yearMonth = new Date();
-    yearMonth.setTime(0);
-    yearMonth.setFullYear(date.getFullYear());
-    yearMonth.setMonth(date.getMonth());
-    return yearMonth;
-}
-
 StorageService.prototype.getAllByMonth = function () {
     var rawList = this.getRawList();
     var result = [];
 
     rawList.forEach(function (rawObj) {
         var app = new Appointment(rawObj);
 
-        if ((result.length == 0) || (result[result.length - 1].getMonth().valueOf() != toYearMonth(app.getVisitDate()).valueOf()))
-            result.push(new AppointmentMonth().setMonth(toYearMonth(app.getVisitDate())));
+        if ((result.length == 0) || (result[result.length - 1].getMonth().valueOf() != app.getVisitDate().getTrimToMonth().valueOf()))
+            result.push(new AppointmentMonth().setMonth(app.getVisitDate().getTrimToMonth()));
 
         var last = result[result.length - 1];
         last.setCount(last.getCount() + 1);

diff --git a/Source/OfflineExample/Scripts/Util/DateUtil.js b/Source/OfflineExample/Scripts/Util/DateUtil.js
@@ -18,4 +18,8 @@
         return shortMonth[this.getMonth()];
     }
 
+    Date.prototype.getTrimToMonth = function () {
+        return new Date(this.getFullYear(), this.getMonth(), 1);
+    }
+
 })();
diff --git a/Source/OfflineExample/Scripts/Util/spec/DateUtil.spec.js b/Source/OfflineExample/Scripts/Util/spec/DateUtil.spec.js
@@ -12,4 +12,12 @@ describe("DateUtil tests", function () {
         expect(shortName).toBe("Feb");
     });
 
+    it("should return the 1st of the month for a given date", function () {
+        var dte = new Date("08-Feb-2003 04:05:67");
+
+        var monthOnly = dte.getTrimToMonth();
+
+        expect(monthOnly.valueOf()).toBe(new Date("01 Feb 2003").valueOf());
+    });
+
 });