001 /**
002 * Copyright 2004-2013 The Kuali Foundation
003 *
004 * Licensed under the Educational Community License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 * http://www.opensource.org/licenses/ecl2.php
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016 package org.kuali.hr.time.util;
017
018 import java.math.BigDecimal;
019 import java.sql.Date;
020 import java.sql.Timestamp;
021 import java.text.SimpleDateFormat;
022 import java.util.List;
023
024 import org.joda.time.DateTime;
025 import org.joda.time.Interval;
026 import org.junit.Assert;
027 import org.junit.Test;
028 import org.kuali.hr.time.calendar.CalendarEntries;
029
030 public class TKUtilsTest extends Assert {
031
032 @Test
033 public void testGetWeekIntervals() throws Exception {
034 Date beginDate = new Date((new DateTime(2010, 1, 1, 12, 0, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis());
035 Date endDate = new Date((new DateTime(2010, 3, 1, 12, 0, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis());
036 List<Interval> intervals = TKUtils.getWeekIntervals(beginDate, endDate);
037 assertEquals("Missing partial week", 9, intervals.size());
038
039 beginDate = new Date((new DateTime(2010, 1, 1, 12, 0, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis());
040 endDate = new Date((new DateTime(2010, 1, 3, 12, 0, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis());
041 intervals = TKUtils.getWeekIntervals(beginDate, endDate);
042 assertEquals("Missing partial week", 1, intervals.size());
043
044 }
045
046 @Test
047 public void testgetHoursBetween() throws Exception {
048 Timestamp beginTime = new Timestamp((new DateTime(2010, 10, 16, 12, 3, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis());
049 Timestamp endTime = new Timestamp((new DateTime(2010, 10, 17, 12, 3, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis());
050 BigDecimal hours = TKUtils.getHoursBetween(beginTime.getTime(), endTime.getTime());
051 assertEquals("Wrong hours", 24, hours.intValue());
052
053 endTime = new Timestamp((new DateTime(2010, 10, 16, 18, 3, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis());
054 hours = TKUtils.getHoursBetween(beginTime.getTime(), endTime.getTime());
055 assertEquals("Wrong hours", 6, hours.intValue());
056
057 endTime = new Timestamp((new DateTime(2010, 10, 16, 18, 0, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis());
058 hours = TKUtils.getHoursBetween(beginTime.getTime(), endTime.getTime());
059 assertEquals("Wrong hours", 5, hours.intValue());
060
061 }
062
063 @Test
064 public void testGetFullWeekDaySpanForPayCalendarEntry() {
065 CalendarEntries payCalendarEntry = new CalendarEntries();
066 // begin date is a Monday
067 payCalendarEntry.setBeginPeriodDateTime(new Date((new DateTime(2011, 8, 8, 12, 0, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis()));
068 // end date is a Thursday
069 payCalendarEntry.setEndPeriodDateTime(new Date((new DateTime(2011, 8, 25, 12, 0, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis()));
070 List<Interval> intervals = TKUtils.getFullWeekDaySpanForCalendarEntry(payCalendarEntry, TKUtils.getSystemDateTimeZone() );
071 SimpleDateFormat format = new SimpleDateFormat("MM/dd/yyyy");
072 assertEquals("First Interval should be 08/07/2011", "08/07/2011", format.format(intervals.get(0).getStart().toDate()));
073 assertEquals("Last Interval should be 08/26/2011", "08/26/2011", format.format(intervals.get(intervals.size()-1).getStart().toDate()));
074
075 // begin date is a Sunday
076 payCalendarEntry.setBeginPeriodDateTime(new Date((new DateTime(2011, 8, 14, 12, 0, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis()));
077 intervals = TKUtils.getFullWeekDaySpanForCalendarEntry(payCalendarEntry, TKUtils.getSystemDateTimeZone() );
078 assertEquals("First Interval should be 08/14/2011", "08/14/2011", format.format(intervals.get(0).getStart().toDate()));
079 assertEquals("Last Interval should be 08/26/2011", "08/26/2011", format.format(intervals.get(intervals.size()-1).getStart().toDate()));
080
081 // end date is a Sunday
082 payCalendarEntry.setEndPeriodDateTime(new Date((new DateTime(2011, 8, 28, 12, 0, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis()));
083 intervals = TKUtils.getFullWeekDaySpanForCalendarEntry(payCalendarEntry, TKUtils.getSystemDateTimeZone() );
084 assertEquals("First Interval should be 08/14/2011", "08/14/2011",format.format(intervals.get(0).getStart().toDate()));
085 assertEquals("Last Interval should be 09/02/2011", "09/02/2011", format.format(intervals.get(intervals.size()-1).getStart().toDate()));
086
087 // end date is a Sunday and end time is 0:00
088 payCalendarEntry.setEndPeriodDateTime(new Date((new DateTime(2011, 8, 28, 0, 0, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis()));
089 intervals = TKUtils.getFullWeekDaySpanForCalendarEntry(payCalendarEntry, TKUtils.getSystemDateTimeZone() );
090 assertEquals("First Interval should be 08/14/2011", "08/14/2011",format.format(intervals.get(0).getStart().toDate()));
091 assertEquals("Last Interval should be 08/27/2011", "08/27/2011", format.format(intervals.get(intervals.size()-1).getStart().toDate()));
092
093 // end date is a Monday and end time is 0:00
094 payCalendarEntry.setEndPeriodDateTime(new Date((new DateTime(2011, 8, 29, 0, 0, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis()));
095 intervals = TKUtils.getFullWeekDaySpanForCalendarEntry(payCalendarEntry, TKUtils.getSystemDateTimeZone() );
096 assertEquals("First Interval should be 08/14/2011", "08/14/2011",format.format(intervals.get(0).getStart().toDate()));
097 assertEquals("Last Interval should be 09/03/2011", "09/03/2011", format.format(intervals.get(intervals.size()-1).getStart().toDate()));
098
099 }
100
101 @Test
102 public void testConvertMillisToMinutes() {
103 BigDecimal mins = TKUtils.convertMillisToMinutes(new Long(380000));
104 assertTrue("Minutes should be between 6 and 7", mins.compareTo(new BigDecimal(6)) > 0 && mins.compareTo(new BigDecimal(7)) < 0);
105 mins = TKUtils.convertMillisToMinutes(new Long(240000));
106 assertTrue("Minutes should be 4", mins.compareTo(new BigDecimal(4)) == 0);
107 }
108
109 @Test
110 public void testFromDateString() {
111 String dateString = "01/01/2012..12/31/2012";
112 String fromDateString = TKUtils.getFromDateString(dateString);
113 assertTrue("fromDateString should be 01/01/2012, not " + fromDateString, fromDateString.equals("01/01/2012"));
114 assertNotNull(TKUtils.formatDateString(fromDateString));
115
116 dateString = ">=2/01/2012";
117 fromDateString = TKUtils.getFromDateString(dateString);
118 assertTrue("fromDateString should be 2/01/2012, not " + fromDateString, fromDateString.equals("2/01/2012"));
119 assertNotNull(TKUtils.formatDateString(fromDateString));
120 }
121
122 @Test
123 public void testToDateString() {
124 String dateString = "01/01/2012..12/31/2012";
125 String toDateString = TKUtils.getToDateString(dateString);
126 assertTrue("toDateString should be 12/31/2012, not " + toDateString, toDateString.equals("12/31/2012"));
127 assertNotNull(TKUtils.formatDateString(toDateString));
128
129 dateString = "<=2/01/2012";
130 toDateString = TKUtils.getToDateString(dateString);
131 assertTrue("toDateString should be 2/01/2012, not " + toDateString, toDateString.equals("2/01/2012"));
132 assertNotNull(TKUtils.formatDateString(toDateString));
133 }
134
135 }