AnnotationExamples.java

/*******************************************************************************
 * Copyright 2018 Univocity Software Pty Ltd
 *
 * 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 com.univocity.parsers.examples.annotation;

import com.univocity.parsers.examples.*;
import com.univocity.parsers.fixed.*;
import org.testng.annotations.*;

import java.io.*;

public class AnnotationExamples extends Example {

	private FixedWidthParserSettings getSettings(FixedWidthFields fields) {
		FixedWidthParserSettings settings = new FixedWidthParserSettings(fields);
		settings.getFormat().setLineSeparator("\n");
		settings.getFormat().setPadding('.');
		settings.setHeaderExtractionEnabled(true);
		return settings;
	}

	private FixedWidthParserSettings getBasicProfileSettings() {
		FixedWidthFields fields = new FixedWidthFields();
		fields.addField("profile_id", 12);
		fields.addField("username", 15);
		fields.addField("followers", 10);
		return getSettings(fields);
	}

	private FixedWidthParserSettings getBasicProfile2Settings() {
		FixedWidthFields fields = new FixedWidthFields();
		fields.addField("id", 12);
		fields.addField("user", 15);
		fields.addField("created_at", 12);
		fields.addField("fees", 9);
		fields.addField("type", 6);
		fields.addField("admin", 7);
		fields.addField("stars", 5);

		return getSettings(fields);
	}

	private FixedWidthParserSettings getOffenderProfileSettings() {
		FixedWidthFields fields = new FixedWidthFields();
		fields.addField("id", 12);
		fields.addField("user", 15);
		fields.addField("words", 14);

		return getSettings(fields);
	}

	private FixedWidthParserSettings getAddressBookSettings() {
		FixedWidthFields fields = new FixedWidthFields();
		fields.addField("mail_street", 17);
		fields.addField("mail_city", 11);
		fields.addField("mail_state", 13);
		fields.addField("main_street", 17);
		fields.addField("main_city", 11);
		fields.addField("main_state", 10);

		return getSettings(fields);
	}

	private FixedWidthParserSettings getDateSettings() {
		FixedWidthFields fields = new FixedWidthFields();
		fields.addField("id", 5);
		fields.addField("created_at", 12);
		fields.addField("updated_at", 12);
		fields.addField("deleted_at", 10);

		return getSettings(fields);
	}

	@Test
	public void parseProfileByFieldName() {
		Reader input = getReader("/examples/annotation/basic_profile.txt");

		for (ProfileByFieldName profile : new FixedWidthRoutines(getBasicProfileSettings()).iterate(ProfileByFieldName.class, input)) {
			println(profile);
		}
		printAndValidate();
	}

	@Test
	public void parseProfileByFieldPosition() {
		Reader input = getReader("/examples/annotation/basic_profile.txt");

		for (ProfileByFieldPosition profile : new FixedWidthRoutines(getBasicProfileSettings()).iterate(ProfileByFieldPosition.class, input)) {
			println(profile);
		}

		printAndValidate();
	}

	@Test
	public void parseProfileByMultipleFieldNames() {
		Reader input;

		input = getReader("/examples/annotation/basic_profile_2.txt");
		for (ProfileByMultipleFieldNames profile : new FixedWidthRoutines(getBasicProfile2Settings()).iterate(ProfileByMultipleFieldNames.class, input)) {
			println(profile);
		}

		input = getReader("/examples/annotation/basic_profile.txt");
		for (ProfileByMultipleFieldNames profile : new FixedWidthRoutines(getBasicProfileSettings()).iterate(ProfileByMultipleFieldNames.class, input)) {
			println(profile);
		}

		printAndValidate();
	}

	@Test
	public void parseProfileWithDate() {
		Reader input;

		input = getReader("/examples/annotation/basic_profile_2.txt");
		for (ProfileWithDate profile : new FixedWidthRoutines(getBasicProfile2Settings()).iterate(ProfileWithDate.class, input)) {
			println(profile);
		}

		printAndValidate();
	}

	@Test
	public void parseProfile() {
		Reader input;

		input = getReader("/examples/annotation/basic_profile_2.txt");
		for (Profile profile : new FixedWidthRoutines(getBasicProfile2Settings()).iterate(Profile.class, input)) {
			println(profile);
		}
		printAndValidate();
	}

	@Test
	public void parseOffenders(){
		Reader input;

		input = getReader("/examples/annotation/offender_profiles.txt");
		for (Offender offender : new FixedWidthRoutines(getOffenderProfileSettings()).iterate(Offender.class, input)) {
			println(offender);
		}
		printAndValidate();
	}

	@Test
	public void parseBetterOffenders(){
		Reader input;

		input = getReader("/examples/annotation/offender_profiles.txt");
		for (BetterOffender offender : new FixedWidthRoutines(getOffenderProfileSettings()).iterate(BetterOffender.class, input)) {
			println(offender);
		}
		printAndValidate();
	}

	@Test
	public void parseAddressBook(){
		Reader input;

		input = getReader("/examples/annotation/addresses.txt");
		for (AddressBook addressBook : new FixedWidthRoutines(getAddressBookSettings()).iterate(AddressBook.class, input)) {
			println(addressBook);
		}
		printAndValidate();
	}

	@Test
	public void parseDates(){
		Reader input;

		input = getReader("/examples/annotation/dates.txt");
		for (DatesRepetitive dates : new FixedWidthRoutines(getDateSettings()).iterate(DatesRepetitive.class, input)) {
			println(dates);
		}
		printAndValidate();
	}

	@Test
	public void parseDatesWithMetaAnnotation(){
		Reader input;

		input = getReader("/examples/annotation/dates.txt");
		for (DatesWithMetaAnnotation dates : new FixedWidthRoutines(getDateSettings()).iterate(DatesWithMetaAnnotation.class, input)) {
			println(dates);
		}

		printAndValidate();
	}
}