/**
    * Copyright 2004-2014 The Kuali Foundation
    *
    * Licensed under the Educational Community License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    * http://www.opensource.org/licenses/ecl2.php
    *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package org.kuali.kpme.tklm.time.block.history;
  
  import org.joda.time.DateTime;
  import org.joda.time.DateTimeZone;
  import org.joda.time.Interval;
  import org.junit.Assert;
  import org.junit.Test;
  import org.kuali.kpme.core.IntegrationTest;
  import org.kuali.kpme.core.api.calendar.Calendar;
  import org.kuali.kpme.core.api.calendar.entry.CalendarEntry;
  import org.kuali.kpme.core.service.HrServiceLocator;
  import org.kuali.kpme.core.util.TKUtils;
  import org.kuali.kpme.tklm.TKLMIntegrationTestCase;
  import org.kuali.kpme.tklm.api.time.timeblock.TimeBlock;
  import org.kuali.kpme.tklm.time.timeblock.TimeBlockBo;
  import org.kuali.kpme.tklm.time.timehourdetail.TimeHourDetailBo;
  import org.kuali.kpme.tklm.time.util.TkTimeBlockAggregate;
  import org.kuali.rice.core.api.mo.ModelObjectUtils;
  
  import java.math.BigDecimal;
  import java.util.ArrayList;
  import java.util.Collections;
  import java.util.List;
  
  @IntegrationTest
  public class TimeBlockTest extends TKLMIntegrationTestCase {
      private static final ModelObjectUtils.Transformer<TimeBlockBo, TimeBlock> toTimeBlock =
              new ModelObjectUtils.Transformer<TimeBlockBo, TimeBlock>() {
                  public TimeBlock transform(TimeBlockBo input) {
                      return TimeBlockBo.to(input);
                  };
              };
  	@Test
  	public void testTimeBlockComparison() throws Exception {
  		DateTime beginDateTime = new DateTime();
  		DateTime endDateTime = new DateTime();
  		
  		TimeBlockBo timeBlock = new TimeBlockBo();
  		timeBlock.setJobNumber(2L);
  		timeBlock.setWorkArea(1234L);
  		timeBlock.setTask(1L);
  		timeBlock.setEarnCode("REG");
  		timeBlock.setBeginDateTime(beginDateTime);
  		timeBlock.setEndDateTime(endDateTime);
  		TimeHourDetailBo timeHourDetail = new TimeHourDetailBo();
  		timeHourDetail.setEarnCode("REG");
  		timeHourDetail.setHours(new BigDecimal(2.0));
  		timeBlock.getTimeHourDetails().add(timeHourDetail);
  		
  		TimeBlockBo timeBlock2 = new TimeBlockBo();
  		timeBlock2.setJobNumber(2L);
  		timeBlock2.setWorkArea(1234L);
  		timeBlock2.setTask(1L);
  		timeBlock2.setEarnCode("REG");
  		timeBlock2.setBeginDateTime(beginDateTime);
  		timeBlock2.setEndDateTime(endDateTime);
  		TimeHourDetailBo timeHourDetail2 = new TimeHourDetailBo();
  		timeHourDetail2.setEarnCode("REG");
  		timeHourDetail2.setHours(new BigDecimal(2.0));
  		timeBlock2.getTimeHourDetails().add(timeHourDetail);
  		
  		Assert.assertTrue("Timeblock has been equal", timeBlock.equals(timeBlock2));
  	}
  	
  	@Test
  	public void testTimeBlockBuilding() throws Exception {
          DateTimeZone zone = HrServiceLocator.getTimezoneService().getTargetUserTimezoneWithFallback();
  		CalendarEntry.Builder payCalendarEntry = CalendarEntry.Builder.create();
  		DateTime beginPeriodDateTime = new DateTime(2010, 1, 1, 12, 0, 0, 0, zone);
  		DateTime endPeriodDateTime = new DateTime(2010, 1, 15, 12, 0, 0, 0, zone);
  		payCalendarEntry.setBeginPeriodFullDateTime(beginPeriodDateTime);
  		payCalendarEntry.setEndPeriodFullDateTime(endPeriodDateTime);
  		
  		List<Interval> dayInterval = TKUtils.getDaySpanForCalendarEntry(payCalendarEntry.build());
  		DateTime beginDateTime = new DateTime(2010, 1, 1, 13, 0, 0, 0, zone);
  		DateTime endDateTime = new DateTime(2010, 1, 2, 14, 0, 0, 0, zone);
  		
  		Interval firstDay = null;
  		List<TimeBlockBo> lstTimeBlocks = new ArrayList<TimeBlockBo>();
  		for(Interval dayInt : dayInterval){
  			//on second day of span so safe to assume doesnt go furthur than this
  			if(firstDay != null){
  				TimeBlockBo tb = new TimeBlockBo();
  				tb.setBeginDateTime(dayInt.getStart());
 				tb.setEndDateTime(endDateTime);
 				lstTimeBlocks.add(tb);
 				break;
 			}
 			if(dayInt.contains(beginDateTime) ){
 				firstDay = dayInt;
 				if(dayInt.contains(endDateTime)){
 					//create one timeblock if contained in one day interval
 					TimeBlockBo tb = new TimeBlockBo();
 					tb.setBeginDateTime(beginDateTime);
 					tb.setEndDateTime(endDateTime);
 					lstTimeBlocks.add(tb);
 					break;
 				} else {
 					//create a timeblock that wraps the 24 hr day
 					TimeBlockBo tb = new TimeBlockBo();
 					tb.setBeginDateTime(beginDateTime);
 					tb.setEndDateTime(firstDay.getEnd());
 					lstTimeBlocks.add(tb);
 				}
 			}
 		}
 		Assert.assertTrue("Two timeblocks created", lstTimeBlocks.size() == 2);
 		
 		lstTimeBlocks.clear();
 		
 		beginDateTime = new DateTime(2010, 1, 1, 13, 0, 0, 0, zone);
 		endDateTime = new DateTime(2010, 1, 1, 15, 0, 0, 0, zone);
 		
 		firstDay = null;
 		for(Interval dayInt : dayInterval){
 			//on second day of span so safe to assume doesnt go furthur than this
 			if(firstDay != null){
 				TimeBlockBo tb = new TimeBlockBo();
 				tb.setBeginDateTime(dayInt.getStart());
 				tb.setEndDateTime(endDateTime);
 				lstTimeBlocks.add(tb);
 				break;
 			}
 			if(dayInt.contains(beginDateTime) ){
 				firstDay = dayInt;
 				if(dayInt.contains(endDateTime)){
 					//create one timeblock if contained in one day interval
 					TimeBlockBo tb = new TimeBlockBo();
 					tb.setBeginDateTime(beginDateTime);
 					tb.setEndDateTime(endDateTime);
 					lstTimeBlocks.add(tb);
 					break;
 				} else {
 					//create a timeblock that wraps the 24 hr day
 					TimeBlockBo tb = new TimeBlockBo();
 					tb.setBeginDateTime(beginDateTime);
 					tb.setEndDateTime(firstDay.getEnd());
 					lstTimeBlocks.add(tb);
 				}
 			}
 		}
 		Assert.assertTrue("One time block creation", lstTimeBlocks.size() == 1);
 	}
 	
 	@Test
 	public void testTimeBlockAggregate() throws Exception {
         DateTimeZone zone = HrServiceLocator.getTimezoneService().getTargetUserTimezoneWithFallback();
 		DateTime beginTime = new DateTime(2010, 1, 1, 12, 0, 0, 0, zone);
 		DateTime endTime = new DateTime(2010, 1, 16, 12, 0, 0, 0, zone);
 
         Calendar.Builder payCalendar = Calendar.Builder.create();
 		
 		CalendarEntry.Builder payCalendarEntry = CalendarEntry.Builder.create();
 		payCalendarEntry.setBeginPeriodFullDateTime(beginTime);
 		payCalendarEntry.setEndPeriodFullDateTime(endTime);
 		
 		List<TimeBlock> lstTimeBlocks = setupTimeBlocks(beginTime, endTime, payCalendarEntry.build());
 		TkTimeBlockAggregate tkTimeBlockAggregate = new TkTimeBlockAggregate(lstTimeBlocks, payCalendarEntry.build(), payCalendar.build());
 		Assert.assertTrue("Aggregate built correctly ", tkTimeBlockAggregate!= null && tkTimeBlockAggregate.getWeekTimeBlocks(0).size() == 7);
 		Assert.assertTrue("Total number of days is correct",tkTimeBlockAggregate.getDayTimeBlockList().size()==15);
 	}
 	
 	@Test
 	public void testTimeBlockSorting() throws Exception {
         DateTimeZone zone = HrServiceLocator.getTimezoneService().getTargetUserTimezoneWithFallback();
 		List<TimeBlockBo> tbList = new ArrayList<TimeBlockBo>();
 		TimeBlockBo tb1 = new TimeBlockBo();
 		// time block with 2010 time
 		tb1.setBeginDateTime(new DateTime(2010, 1, 1, 13, 0, 0, 0, zone));
 		tb1.setEndDateTime(new DateTime(2010, 1, 2, 14, 0, 0, 0, zone));
 		tbList.add(tb1);
 		//time block with 2009 time
 		TimeBlockBo tb2 = new TimeBlockBo();
 		tb2.setBeginDateTime(new DateTime(2009, 1, 1, 13, 0, 0, 0, zone));
 		tb2.setEndDateTime(new DateTime(2009, 1, 2, 14, 0, 0, 0, zone));
 		tbList.add(tb2);
 		
 		Assert.assertTrue(tbList.get(0) == tb1);
 		Assert.assertTrue(tbList.get(1) == tb2);
 		// after sort
 		Collections.sort(tbList);
 		Assert.assertTrue(tbList.get(0) == tb2);
 		Assert.assertTrue(tbList.get(1) == tb1);
 	}
 	private List<TimeBlock> setupTimeBlocks(DateTime startTime, DateTime endTime, CalendarEntry payCalendarEntry){
 		List<Interval> dayInterval = TKUtils.getDaySpanForCalendarEntry(payCalendarEntry);
         DateTimeZone zone = HrServiceLocator.getTimezoneService().getTargetUserTimezoneWithFallback();
 		DateTime beginDateTime = new DateTime(2010, 1, 1, 13, 0, 0, 0, zone);
 		DateTime endDateTime = new DateTime(2010, 1, 2, 14, 0, 0, 0, zone);
 		
 		Interval firstDay = null;
 		List<TimeBlockBo> lstTimeBlocks = new ArrayList<TimeBlockBo>();
 		for(Interval dayInt : dayInterval){
 			//on second day of span so safe to assume doesnt go furthur than this
 			if(firstDay != null){
 				TimeBlockBo tb = new TimeBlockBo();
 				tb.setBeginDateTime(dayInt.getStart());
 				tb.setEndDateTime(endDateTime);
 				lstTimeBlocks.add(tb);
 				break;
 			}
 			if(dayInt.contains(beginDateTime) ){
 				firstDay = dayInt;
 				if(dayInt.contains(endDateTime)){
 					//create one timeblock if contained in one day interval
 					TimeBlockBo tb = new TimeBlockBo();
 					tb.setBeginDateTime(beginDateTime);
 					tb.setEndDateTime(endDateTime);
 					lstTimeBlocks.add(tb);
 					break;
 				} else {
 					//create a timeblock that wraps the 24 hr day
 					TimeBlockBo tb = new TimeBlockBo();
 					tb.setBeginDateTime(beginDateTime);
 					tb.setEndDateTime(firstDay.getEnd());
 					lstTimeBlocks.add(tb);
 				}
 			}
 		}
 		Assert.assertTrue("Two timeblocks created", lstTimeBlocks.size() == 2);
 		
 		lstTimeBlocks.clear();
 		
 		beginDateTime = new DateTime(2010, 1, 1, 13, 0, 0, 0, zone);
 		endDateTime = new DateTime(2010, 1, 1, 15, 0, 0, 0, zone);
 		
 		firstDay = null;
 		for(Interval dayInt : dayInterval){
 			//on second day of span so safe to assume doesnt go furthur than this
 			if(firstDay != null){
 				TimeBlockBo tb = new TimeBlockBo();
 				tb.setBeginDateTime(dayInt.getStart());
 				tb.setEndDateTime(endDateTime);
 				lstTimeBlocks.add(tb);
 				break;
 			}
 			if(dayInt.contains(beginDateTime) ){
 				firstDay = dayInt;
 				if(dayInt.contains(endDateTime)){
 					//create one timeblock if contained in one day interval
 					TimeBlockBo tb = new TimeBlockBo();
 					tb.setBeginDateTime(beginDateTime);
 					tb.setEndDateTime(endDateTime);
 					lstTimeBlocks.add(tb);
 					break;
 				} else {
 					//create a timeblock that wraps the 24 hr day
 					TimeBlockBo tb = new TimeBlockBo();
 					tb.setBeginDateTime(beginDateTime);
 					tb.setEndDateTime(firstDay.getEnd());
 					lstTimeBlocks.add(tb);
 				}
 			}
 		}
 		return ModelObjectUtils.transform(lstTimeBlocks, toTimeBlock);
 	}
 }