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.sql.Date;
019    import java.util.Map;
020    
021    import org.joda.time.DateTime;
022    import org.junit.Assert;
023    import org.junit.Test;
024    import org.kuali.hr.test.KPMETestCase;
025    import org.kuali.hr.time.detail.web.ActionFormUtils;
026    import org.kuali.hr.time.test.TkTestUtils;
027    import org.kuali.hr.time.timesheet.TimesheetDocument;
028    
029    public class ActionFormUtilsTest extends KPMETestCase {
030            
031            @Test
032            public void testBuildAssignmentStyleClassMap() {
033                    Date aDate = new Date((new DateTime(2011, 7, 7, 0, 0, 0, 0, TKUtils.getSystemDateTimeZone())).getMillis());
034                    TimesheetDocument doc = TkTestUtils.populateTimesheetDocument(aDate);
035                    Map<String, String> aMap = ActionFormUtils.buildAssignmentStyleClassMap(doc.getTimeBlocks());
036                    Assert.assertEquals("Wrong number of classes in style class map", 1, aMap.size());
037                    Assert.assertEquals("Wrong key for class assignment0", "assignment0", aMap.get("1_1234_1"));
038            }
039    
040    }