USPSParserTest.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.util.List;
import org.apache.commons.io.FileUtils;
import org.junit.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.io.Resource;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.AbstractJUnit4SpringContextTests;

import org.bitsofinfo.util.address.usps.ais.Copyright;
import org.bitsofinfo.util.address.usps.ais.USPSProductType;
import org.bitsofinfo.util.address.usps.ais.USPSRecord;
import org.bitsofinfo.util.reflection.ClassFinder;

@ContextConfiguration
public class USPSParserTest extends AbstractJUnit4SpringContextTests {

	@Autowired
	private ClassFinder classFinder;
	
	@Autowired
	private USPSParser uspsParser;

	
	private List<String> loadLinesFromFile(String filename) throws Exception {
		Resource file = applicationContext.getResource("classpath:" + filename);
		List<String> lines = FileUtils.readLines(file.getFile());
		return lines;
	}
	
	@Test
	public void parseCityStateSample() throws Exception {
		
		// locate the raw source data 
		List<String> lines = loadLinesFromFile("usps.ais.cityState.sample.txt");
		
		assert(lines.size() == 59);
		
		// get the copyright
		Copyright copyright = uspsParser.parseCopyright(lines.remove(0));
		
		// load up all the data
		List<USPSRecord> records = uspsParser.parseData(USPSProductType.CITY_STATE, copyright, lines);
		
		assert(records.size() == 58);

	}
	

	
	@Test
	public void parseZipPlus4Sample() throws Exception {
		
		// locate the raw source data 
		List<String> lines = loadLinesFromFile("usps.ais.zipPlus4.sample.txt");
		
		assert(lines.size() == 40);
		
		// get the copyright
		Copyright copyright = uspsParser.parseCopyright(lines.remove(0));
		
		// load up all the data
		List<USPSRecord> records = uspsParser.parseData(USPSProductType.ZIP_PLUS_4, copyright, lines);
		
		assert(records.size() == 39);

	}
}