LocalDataLoaderTest.java

/*****************************************************************************
* Copyright 2012  bitsofinfo.g [at] gmail [dot] com 
* 
* 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
* 
* Author: bitsofinfo.g [at] gmail [dot] com 
* @see bitsofinfo.wordpress.com
*****************************************************************************/
package org.bitsofinfo.util.address.usps.ais.loader;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

import org.junit.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.AbstractJUnit4SpringContextTests;

import org.bitsofinfo.util.address.usps.ais.loader.local.LocalDataLoader;
import org.bitsofinfo.util.address.usps.ais.source.Zip4SourceDataExtractor;
import org.bitsofinfo.util.address.usps.ais.store.USPSDataStore;
import org.bitsofinfo.util.io.FileHandle;

@ContextConfiguration
public class LocalDataLoaderTest extends AbstractJUnit4SpringContextTests {

	@Autowired
	private LocalDataLoader dataLoader;
	
	@Autowired
	private Zip4SourceDataExtractor reader;

	@Autowired
	private USPSDataStore uspsDataStore;
	
	@Test
	public void testDataLoader() throws Exception{ 
		// load all source data from the CD
		ArrayList<File> sourceDirs = new ArrayList<File>();
		sourceDirs.add(new File("/Volumes/ZIP4NATL__11/ctystate"));
		sourceDirs.add(new File("/Volumes/ZIP4NATL__11/zip4"));
		sourceDirs.add(new File("/Volumes/ZIP4NATL__11/z4trans"));
		List<FileHandle> handles = reader.extractSourceData(sourceDirs, new File("/tmp"));
		
		
		//ArrayList<File> extractedDirs = new ArrayList<File>();
		//extractedDirs.add(new File("/tmp/ctystate"));
		//extractedDirs.add(new File("/tmp/zip4"));
		//extractedDirs.add(new File("/tmp/z4trans"));
		//List<FileHandle> handles = reader.createHandles(extractedDirs);
		
		
		LoaderJob job = dataLoader.createJob("LocalDataLoaderTest", handles, uspsDataStore);
		dataLoader.load(job);
		
		boolean complete = false;
		while(!complete) {
			Thread.currentThread();
			Thread.sleep(30000);
			complete = job.isComplete();
			int v=0;
		}
		
		int v=0;
	}
	
}