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.overtime.web;
017    
018    import java.sql.Date;
019    import java.util.ArrayList;
020    import java.util.HashMap;
021    import java.util.List;
022    import java.util.Map;
023    
024    import org.joda.time.DateTime;
025    import org.json.simple.JSONArray;
026    import org.json.simple.JSONObject;
027    import org.json.simple.JSONValue;
028    import org.junit.Assert;
029    import org.junit.Test;
030    import org.kuali.hr.time.assignment.Assignment;
031    import org.kuali.hr.time.assignment.AssignmentDescriptionKey;
032    import org.kuali.hr.time.calendar.CalendarEntries;
033    import org.kuali.hr.time.detail.web.TimeDetailActionFormBase;
034    import org.kuali.hr.time.earncode.EarnCode;
035    import org.kuali.hr.time.service.base.TkServiceLocator;
036    import org.kuali.hr.time.timesheet.TimesheetDocument;
037    import org.kuali.hr.time.timesheet.web.TimesheetWebTestBase;
038    import org.kuali.hr.time.util.TKContext;
039    import org.kuali.hr.time.util.TKUtils;
040    import org.kuali.hr.time.util.TimeDetailTestUtils;
041    
042    import com.gargoylesoftware.htmlunit.html.HtmlForm;
043    import com.gargoylesoftware.htmlunit.html.HtmlPage;
044    
045    public class DailyOvertimeWebIntegrationTest extends TimesheetWebTestBase {
046    
047        public static final String USER_PRINCIPAL_ID = "admin";
048            private Date JAN_AS_OF_DATE = new Date((new DateTime(2010, 1, 1, 0, 0, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis());
049    
050        @Test
051        /**
052         * Tests daily overtime for:
053         *
054         * 3/2/2011 10hrs; 8 RGN, 2 OVT
055         * 3/3/2011 10hrs; 8 RGN, 2 OVT
056         */
057        public void testSimpleDOTCalculationIntegration() throws Exception {
058            Date asOfDate = new Date((new DateTime(2011, 3, 1, 12, 0, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis());
059    
060            CalendarEntries pcd = TkServiceLocator.getCalendarService().getCurrentCalendarDates(USER_PRINCIPAL_ID, asOfDate);
061            Assert.assertNotNull("No PayCalendarDates", pcd);
062    
063            TimesheetDocument tdoc = TkServiceLocator.getTimesheetService().openTimesheetDocument(USER_PRINCIPAL_ID, pcd);
064            String tdocId = tdoc.getDocumentId();
065            HtmlPage page = loginAndGetTimeDetailsHtmlPage(getWebClient(), "admin", tdocId,true);
066            Assert.assertNotNull(page);
067            HtmlForm form = page.getFormByName("TimeDetailActionForm");
068            Assert.assertNotNull(form);
069    
070            // 1. Obtain User Data
071            Assignment assignment = TkServiceLocator.getAssignmentService().getAssignment(TKContext.getPrincipalId(), new AssignmentDescriptionKey("30_30_30"), JAN_AS_OF_DATE);
072            EarnCode earnCode = TkServiceLocator.getEarnCodeService().getEarnCode("RGN", JAN_AS_OF_DATE);
073            Assert.assertEquals("There should be no existing time blocks.", 0, tdoc.getTimeBlocks().size());
074    
075            // 2. Set Timeblock Start and End time
076            // 3/02/2011 - 8:00a to 6:00pm
077            // OVT - 2 Hrs Expected
078            DateTime start = new DateTime(2011, 3, 2, 8, 0, 0, 0, TKUtils.getSystemDateTimeZone());
079            DateTime end = new DateTime(2011, 3, 3, 18, 0, 0, 0, TKUtils.getSystemDateTimeZone());
080    
081            // Build an action form - we're using it as a POJO, it ties into the
082            // existing TK validation setup
083            TimeDetailActionFormBase tdaf = TimeDetailTestUtils.buildDetailActionForm(tdoc, assignment, earnCode, start, end, null, true, null, true);
084            List<String> errors = TimeDetailTestUtils.setTimeBlockFormDetails(form, tdaf);
085            // Check for errors
086            Assert.assertEquals("There should be no errors in this time detail submission", 0, errors.size());
087    
088            page = TimeDetailTestUtils.submitTimeDetails(getWebClient(), getTimesheetDocumentUrl(tdocId), tdaf);
089            Assert.assertNotNull(page);
090            //HtmlUnitUtil.createTempFile(page, "TimeBlockPresent");
091    
092            // Verify block present on rendered page.
093            String pageAsText = page.asText();
094            //HtmlUnitUtil.createTempFile(page, "Hours");
095            // JSON
096            //
097            //
098            // Grab the timeblock data from the text area. We can check specifics there
099            // to be more fine grained in our validation.
100            String dataText = page.getElementById("timeBlockString").getFirstChild().getNodeValue();
101            JSONArray jsonData = (JSONArray) JSONValue.parse(dataText);
102            final JSONObject jsonDataObject = (JSONObject) jsonData.get(0);
103            Assert.assertTrue("TimeBlock #1 Data Missing.", checkJSONValues(new JSONObject() {{ put("outer", jsonDataObject); }},
104                    new ArrayList<Map<String, Object>>() {{
105                        add(new HashMap<String, Object>() {{
106                            put("earnCode", "RGN");
107                            put("hours", "8.0");
108                        }});
109                        add(new HashMap<String, Object>() {{
110                            put("earnCode", "OVT");
111                            put("hours", "2.0");
112                        }});
113                    }},
114                    new HashMap<String, Object>() {{
115                        put("earnCode", "RGN");
116                        put("startNoTz", "2011-03-02T08:00:00");
117                        put("endNoTz", "2011-03-02T18:00:00");
118                        put("title", "SDR1 Work Area");
119                        put("assignment", "30_30_30");
120                    }}
121            ));
122            final JSONObject jsonDataObject2 = (JSONObject) jsonData.get(1);
123            Assert.assertTrue("TimeBlock #2 Data Missing.", checkJSONValues(new JSONObject() {{ put("outer", jsonDataObject2); }},
124                    new ArrayList<Map<String, Object>>() {{
125                        add(new HashMap<String, Object>() {{
126                            put("earnCode", "RGN");
127                            put("hours", "8.0");
128                        }});
129                        add(new HashMap<String, Object>() {{
130                            put("earnCode", "OVT");
131                            put("hours", "2.0");
132                        }});
133                    }},
134                    new HashMap<String, Object>() {{
135                        put("earnCode", "RGN");
136                        put("startNoTz", "2011-03-03T08:00:00");
137                        put("endNoTz", "2011-03-03T18:00:00");
138                        put("title", "SDR1 Work Area");
139                        put("assignment", "30_30_30");
140                    }}
141            ));
142    
143    
144            // Check the Display Rendered Text for Time Block, Quick Check
145            // Not as accurate as teh checkJSONValues tests above.
146            Assert.assertTrue("TimeBlock not Present.", pageAsText.contains("08:00 AM - 06:00 PM"));
147            Assert.assertTrue("TimeBlock not Present.", pageAsText.contains("RGN - 8.00 hours"));
148            Assert.assertTrue("TimeBlock not Present.", pageAsText.contains("OVT - 2.00 hours"));
149        }
150    
151    }