/**
 * Copyright 2010-2011 Joseph Panico
 *
 * Licensed under the Apache 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.apache.org/licenses/LICENSE-2.0
 *
 * 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.diffkit.diff.sns.tst

import java.io.File;



import org.diffkit.db.DKDBColumn 
import org.diffkit.db.DKDBConnectionInfo 
import org.diffkit.db.DKDatabase 
import org.diffkit.db.DKDBFlavor;
import org.diffkit.db.DKDBPrimaryKey;
import org.diffkit.db.DKDBTable
import org.diffkit.db.DKDBH2Loader
import org.diffkit.db.DKDBTableDataAccess;
import org.diffkit.diff.engine.DKColumnModel;
import org.diffkit.diff.engine.DKTableModel;
import org.diffkit.diff.sns.DKDBSource 
import org.diffkit.diff.sns.DKTableModelUtil;
import org.diffkit.util.DKResourceUtil
import org.diffkit.util.DKTimeUtil;

import groovy.util.GroovyTestCase;


/**
 * @author jpanico
 */
public class TestDBSource extends GroovyTestCase {
   
   public void testValidateKeyColumnNames() {
      DKDBConnectionInfo connectionInfo = ['test', DKDBFlavor.H2,"mem:test", null, null, 'test', 'test']
      println "connectionInfo->$connectionInfo"
      DKDatabase database = [connectionInfo]
      def connection = database.connection
      def dbTable = this.createCustomerMetaTable()
      assert database.createTable( dbTable)
      
      shouldFail(RuntimeException) {
         DKDBSource source = new DKDBSource(dbTable.tableName, null, database, null, (String[])['maiden_name'], null)
      }
      assert database.dropTable( dbTable)
   }
   
   public void testDescription() {
      DKDBConnectionInfo connectionInfo = ['test', DKDBFlavor.H2,"mem:test", null, null, 'test', 'test']
      println "connectionInfo->$connectionInfo"
      DKDatabase database = [connectionInfo]
      def connection = database.connection
      def dbTable = this.createCustomerMetaTable()
      assert database.createTable( dbTable)
      
      DKDBSource source = new DKDBSource(dbTable.tableName, null, database, null, (String[])['first_name'], null)
      println "source->$source"
      assert source.description  == 'DKDBSource[tableName=CUSTOMER, whereClause=null, keyColumnNames=[first_name], database=DKDBConnectionInfo[test(flavor=H2,database=mem:test,host=null,port=null)]]'
      assert database.dropTable( dbTable)
   }
   
   public void testToString() {
      DKDBConnectionInfo connectionInfo = ['test', DKDBFlavor.H2,"mem:test", null, null, 'test', 'test']
      println "connectionInfo->$connectionInfo"
      DKDatabase database = [connectionInfo]
      def connection = database.connection
      def dbTable = this.createCustomerMetaTable()
      assert database.createTable( dbTable)
      
      DKDBSource source = new DKDBSource(dbTable.tableName, null, database, null, (String[])['first_name'], null)
      println "source->$source"
      assert source.toString().startsWith( 'DKDBSource')
      assert source.toString().endsWith( '[CUSTOMER,jdbc:h2:mem:test]')
      assert database.dropTable( dbTable)
   }
   
   public void testKeyColumnNames() {
      DKDBConnectionInfo connectionInfo = ['test', DKDBFlavor.H2,"mem:test", null, null, 'test', 'test']
      println "connectionInfo->$connectionInfo"
      DKDatabase database = [connectionInfo]
      def connection = database.connection
      def dbTable = this.createCustomerMetaTable()
      assert database.createTable( dbTable)
      
      DKDBSource source = new DKDBSource(dbTable.tableName, null, database, null, (String[])['first_name'], null)
      def model = source.model
      assert model
      assert model.key == [0]
      assert database.dropTable( dbTable)
   }
   
   public void testRead(){
      DKDBConnectionInfo connectionInfo = ['test', DKDBFlavor.H2,"mem:test", null, null, 'test', 'test']
      println "connectionInfo->$connectionInfo"
      DKDatabase database = [connectionInfo]
      def connection = database.connection
      def dbTable = this.createCustomerMetaTable()
      assert database.createTable( dbTable)
      DKDBH2Loader loader = [database]
      def DKDBTableDataAccess tableDataAccess = [database]
      def fetchedTable = tableDataAccess.getTable(dbTable.tableName)
      def csvFile = this.getCsvFile()
      assert loader.load(dbTable, csvFile)
      def tableModel = DKTableModelUtil.createDefaultTableModel(database.flavor,dbTable, null)
      assert tableModel
      
      // goof up the model to force a fail
      def columns = tableModel.columns
      columns[0]= new DKColumnModel(0, 'does_not_exist', DKColumnModel.Type.STRING)
      tableModel = new DKTableModel(tableModel.name, columns, tableModel.key)
      shouldFail(IllegalArgumentException){
         DKDBSource source = new DKDBSource(dbTable.tableName, null, database, tableModel, null, null)
      }
      
      tableModel = DKTableModelUtil.createDefaultTableModel(database.flavor,dbTable, null)
      DKDBSource source = new DKDBSource(dbTable.tableName, null, database, tableModel, null, null)
      source.open(null)
      
      def row = source.nextRow
      assert row
      println "row->$row"
      println "row[6]->${row[6].class}"
      assert row[6] == DKTimeUtil.createDate(1956, 11, 12)
      assert row[7] == DKTimeUtil.createDate(2004, 4, 23, 14, 25, 10, 487)
      assert row == (Object[])['rob','smith', '100 spruce st', 'Phila', 'usa', 50, DKTimeUtil.createDate(1956, 11, 12), DKTimeUtil.createDate(2004, 4, 23, 14, 25, 10, 487)]
      
      row = source.nextRow
      assert row == (Object[])['steve','jobs', 'infinite, loop', 'Cupertino', 'usa', 54, DKTimeUtil.createDate(1955, 0, 1), DKTimeUtil.createDate(2004, 4, 23, 14, 25, 10, 487)]
      
      row = source.nextRow
      assert !row
      row = source.nextRow
      assert !row
      
      source.close(null)
      
      shouldFail() { row = source.nextRow }
      
      assert database.dropTable( dbTable)
   }
   
   private File getCsvFile(){
      def csvFile = DKResourceUtil.findResourceAsFile('org/diffkit/db/tst/customers.csv')
      println "csvFile->$csvFile"
      assert csvFile
      return csvFile
   }
   
   private DKDBTable createCustomerMetaTable(){
      DKDBColumn column1 = ['FIRST_NAME', 1, 'VARCHAR', 50, true]
      DKDBColumn column2 = ['LAST_NAME', 2, 'VARCHAR', 50, true]
      DKDBColumn column3 = ['ADDRESS', 3, 'VARCHAR', 50, true]
      DKDBColumn column4 = ['CITY', 4, 'VARCHAR', 50, true]
      DKDBColumn column5 = ['COUNTRY', 5, 'VARCHAR', 25, true]
      DKDBColumn column6 = ['AGE', 6, 'INTEGER', -1, true]
      DKDBColumn column7 = ['BIRTH', 7, 'DATE', -1, true]
      DKDBColumn column8 = ['NOW', 8, 'TIMESTAMP', -1, true]
      DKDBColumn[] columns = [column1, column2, column3, column4, column5, column6, column7, column8]
      String[] pkColNames = ['FIRST_NAME', 'LAST_NAME']
      DKDBPrimaryKey pk = ['pk_customer', pkColNames]
      DKDBTable table = [ null, null, 'CUSTOMER', columns, pk]
      return table
   }
}