core/org.eclipse.smila.connectivity.framework.crawler.jdbc.test/code/src/org/eclipse/smila/connectivity/framework/crawler/jdbc/test/TestRecordRetrievalDerbyWithGrouping.java

/*******************************************************************************
 * Copyright (c) 2008, 2011 Attensity Europe GmbH and brox IT Solutions GmbH. All rights reserved. This program and the
 * accompanying materials are made available under the terms of the Eclipse Public License v1.0 which accompanies this
 * distribution, and is available at http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors: Michael Breidenband (brox IT Solutions GmbH) - initial creator
 **********************************************************************************************************************/
package org.eclipse.smila.connectivity.framework.crawler.jdbc.test;

import java.util.ArrayList;
import java.util.List;

import org.eclipse.smila.connectivity.framework.DataReference;
import org.eclipse.smila.connectivity.framework.schema.ConfigurationLoader;
import org.eclipse.smila.connectivity.framework.schema.config.DataSourceConnectionConfig;
import org.eclipse.smila.datamodel.Record;

/**
 * Testcase for {@link org.eclipse.smila.connectivity.framework.crawler.jdbc.JdbcCrawler}.
 * 
 * 
 */
public class TestRecordRetrievalDerbyWithGrouping extends AbstractDataEnabledJdbcCrawlerTestCase {

  /** Number of Records expected. */
  private static final int EXPECTED_RECORDS = AbstractDataEnabledJdbcCrawlerTestCase.RECORDS_TO_INSERT;

  /**
   * Tests {@link org.eclipse.smila.connectivity.framework.crawler.jdbc.JdbcCrawler#getRecord(int)}.
   * 
   * @throws Exception
   *           If anything goes critically wrong ...
   */
  public void testRecordRetrieval() throws Exception {

    assertNotNull("Crawler instance should not be null after initializing TestCase", _crawler);
    final DataSourceConnectionConfig config =
      ConfigurationLoader.unmarshall(TestConfigLoad.class.getResourceAsStream("JdbcDerbyWithGrouping.xml"));
    assertNotNull("DataSourceConnection Config should not be null after unmarshalling", config);

    _crawler.initialize(config);
    final long start = System.currentTimeMillis();
    Thread.sleep(WAIT_FOR_CRAWLER_INIT);

    DataReference[] indexingData = null;
    final List<Record> recordList = new ArrayList<Record>();

    while ((indexingData = _crawler.getNext()) != null) {
      if (indexingData.length == 0) {
        break;
      }
      for (final DataReference element : indexingData) {
        recordList.add(element.getRecord());
      }
    }
    assertEquals("The current DB configuration should result in the creation of " + EXPECTED_RECORDS + " records",
      EXPECTED_RECORDS, recordList.size());
    final long stop = System.currentTimeMillis();
    _log.info("Elapsed Time:" + (stop - start));
    _log.info("Records created: " + recordList.size());

  }
}