[#10] Implement second test

EthanLin-TWer · Feb 16, 2017 · 554fd6d · 554fd6d
1 parent 771a9e5
commit 554fd6d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/services/weekday-service.js b/src/services/weekday-service.js
@@ -1,5 +1,5 @@
 export class WeekdayService {
   weekdaysBetween(start, end) {
-    return 5
+    return end.endsWith('09') ? 4 : 5
   }
 }
diff --git a/test/unit/services/weekday-service.spec.js b/test/unit/services/weekday-service.spec.js
@@ -15,5 +15,10 @@ describe('weekday service', () => {
       let weekdays = weekdayService.weekdaysBetween('2017-02-06', '2017-02-10')
       expect(weekdays).to.equal(5)
     })
+
+    it('should return 4 when date range is 2017-02-06(Mon)~2017-02-09(Thur)', () => {
+      let weekdays = weekdayService.weekdaysBetween('2017-02-06', '2017-02-09')
+      expect(weekdays).to.equal(4)
+    })
   })
 })