/**

  * Copyright 2005-2011 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.rice.test.persistence

 import org.kuali.rice.krad.service.KRADServiceLocator

 import org.kuali.rice.krad.service.BusinessObjectService

 import javax.sql.DataSource

 import org.kuali.rice.core.api.resourceloader.GlobalResourceLoader

 import org.kuali.rice.krad.bo.PersistableBusinessObject

 import java.sql.Timestamp

 import org.springframework.jdbc.core.simple.SimpleJdbcTemplate

 import org.joda.time.DateTime

 import org.junit.Assert

 /**

  * Helps with BO persistence tests

  */

 class PersistenceTestHelper {

     def BusinessObjectService boService;

     def DataSource datasource;

     PersistenceTestHelper(String dsName) {

         boService = (BusinessObjectService) KRADServiceLocator.getBusinessObjectService()

         datasource = (DataSource) GlobalResourceLoader.getService(dsName)

         if (!datasource) {

             throw new RuntimeException("DataSource bean not found: " + dsName)

         }

     }

     def bool(value, column) {

         [ (column): value ? "Y" : "N" ]

     }

     def default_field(bo) {

         bool(bo.dflt, 'DFLT_IND')

     }

     def active_field(bo) {

         bool(bo.active, 'ACTV_IND')

     }

     def edit_field(bo) {

         bool(bo.edit, 'EDIT_FLAG')

     }

     def basic_fields(PersistableBusinessObject bo) {

         [ OBJ_ID: bo.objectId,

           VER_NBR: new BigDecimal(bo.versionNumber) ]

     }

     def standard_fields(PersistableBusinessObject bo) {

         active_field(bo) + basic_fields(bo)

     }

     def genDbTimestamp() {

         // this should not be rocket science but we have to deal

         // but it appears mysql (driver?) is truncating time component of datetimes

         // so we can only portably test timestamps without times...

         new Timestamp(new Date().time)

     }

     def toDbTimestamp(DateTime datetime) {

         return toDbTimestamp(datetime.millis)

     }

     def toDbTimestamp(long millis) {

         def timestamp = new java.sql.Timestamp(millis)

         timestamp.nanos = 0

         timestamp

     }

     def assertRow(Map fields, table, pk="id", ignore=["LAST_UPDT_DT"]) {

         def pk_val = fields[pk]

         if (!pk_val) {

             throw new RuntimeException("No primary key value found for field: " + pk)

         }

         Map row = new SimpleJdbcTemplate(datasource).queryForMap("select * from " + table + " where " + pk + "=?", pk_val)

         row.keySet().removeAll(ignore)

         /*for (Map.Entry e: fields.entrySet()) {

             println(e.getKey().getClass());

             println(e.getValue().getClass());

             println(e.getKey());

             println(e.getValue());

         }

         for (Map.Entry e: row.entrySet()) {

             println(e.getKey().getClass());

             println(e.getValue().getClass());

             println(e.getKey());

             println(e.getValue());

         }*/

         Assert.assertEquals(new HashMap(fields), new HashMap(row))

     }

 }