/**
 * AnalyzerBeans
 * Copyright (C) 2014 Neopost - Customer Information Management
 *
 * This copyrighted material is made available to anyone wishing to use, modify,
 * copy, or redistribute it subject to the terms and conditions of the GNU
 * Lesser General Public License, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
 * for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this distribution; if not, write to:
 * Free Software Foundation, Inc.
 * 51 Franklin Street, Fifth Floor
 * Boston, MA  02110-1301  USA
 */
package org.eobjects.analyzer.beans.transform;

import junit.framework.TestCase;

import org.easymock.EasyMock;
import org.eobjects.analyzer.beans.api.OutputColumns;
import org.eobjects.analyzer.beans.api.OutputRowCollector;
import org.eobjects.analyzer.beans.transform.TokenizerTransformer.TokenTarget;
import org.eobjects.analyzer.data.InputColumn;
import org.eobjects.analyzer.data.MetaModelInputColumn;
import org.eobjects.analyzer.data.MockInputRow;
import org.apache.metamodel.schema.MutableColumn;

public class TokenizerTransformerTest extends TestCase {

	public void testGetOutputColumns() throws Exception {
		InputColumn<?> col = new MetaModelInputColumn(new MutableColumn("name"));

		@SuppressWarnings("unchecked")
		InputColumn<String> castColumn = (InputColumn<String>) col;
		TokenizerTransformer transformer = new TokenizerTransformer(castColumn, 2);

		OutputColumns oc = transformer.getOutputColumns();

		assertEquals(2, oc.getColumnCount());
		assertEquals("name (token 1)", oc.getColumnName(0));
		assertEquals("name (token 2)", oc.getColumnName(1));

		transformer = new TokenizerTransformer(castColumn, 1);
		assertEquals("OutputColumns[name (token 1)]", transformer.getOutputColumns().toString());

		transformer = new TokenizerTransformer(castColumn, 0);
		try {
			transformer.getOutputColumns();
			fail("Exception expected");
		} catch (IllegalArgumentException e) {
			assertEquals("Column names length must be 1 or greater", e.getMessage());
		}
	}

	public void testTransformToColumns() throws Exception {
		InputColumn<?> col = new MetaModelInputColumn(new MutableColumn("name"));

		@SuppressWarnings("unchecked")
		TokenizerTransformer transformer = new TokenizerTransformer((InputColumn<String>) col, 2);

		assertEquals(2, transformer.getOutputColumns().getColumnCount());

		MockInputRow row = new MockInputRow();
		row.put(col, "Kasper Sorensen");
		String[] values = transformer.transform(row);
		assertEquals(2, values.length);
		assertEquals("Kasper", values[0]);
		assertEquals("Sorensen", values[1]);

		row = new MockInputRow();
		row.put(col, "Kasper ");
		values = transformer.transform(row);
		assertEquals(2, values.length);
		assertEquals("Kasper", values[0]);
		assertNull(values[1]);
	}

	public void testTransformNull() throws Exception {
		InputColumn<?> col = new MetaModelInputColumn(new MutableColumn("name"));

		@SuppressWarnings("unchecked")
		TokenizerTransformer transformer = new TokenizerTransformer((InputColumn<String>) col, 2);

		assertEquals(2, transformer.getOutputColumns().getColumnCount());

		MockInputRow row = new MockInputRow();
		row.put(col, null);
		String[] values = transformer.transform(row);
		assertEquals(2, values.length);
		assertEquals(null, values[0]);
		assertEquals(null, values[1]);
	}

	public void testTransformToRows() throws Exception {
		InputColumn<?> col = new MetaModelInputColumn(new MutableColumn("name"));

		@SuppressWarnings("unchecked")
		TokenizerTransformer transformer = new TokenizerTransformer((InputColumn<String>) col, 2);
		transformer.tokenTarget = TokenTarget.ROWS;
		OutputRowCollector collectorMock = EasyMock.createMock(OutputRowCollector.class);
		transformer.outputRowCollector = collectorMock;

		assertEquals(1, transformer.getOutputColumns().getColumnCount());
		assertEquals("name (token)", transformer.getOutputColumns().getColumnName(0));
		
		collectorMock.putValues("Hello");
		collectorMock.putValues("world");

		EasyMock.replay(collectorMock);

		String[] result = transformer.transform(new MockInputRow().put(col, "Hello world"));
		assertNull(result);

		EasyMock.verify(collectorMock);
	}
}