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 }