/**
* 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.data;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;
/**
* A mock implementation of the InputRow interface. Allows for adhoc generation
* of a row using the put(...) method.
*/
public class MockInputRow extends AbstractInputRow {
private static final long serialVersionUID = 1L;
private static final AtomicInteger _idGenerator = new AtomicInteger(Integer.MIN_VALUE);
private final Map<InputColumn<?>, Object> _values;
private final int _id;
public MockInputRow() {
this(_idGenerator.getAndIncrement());
}
public MockInputRow(int id, Map<InputColumn<?>, Object> values) {
_values = values;
_id = id;
}
public MockInputRow(Map<InputColumn<?>, Object> values) {
this(_idGenerator.getAndIncrement(), values);
}
public MockInputRow(int id) {
this(id, new LinkedHashMap<InputColumn<?>, Object>());
}
public MockInputRow(InputColumn<?>[] columns, Object[] values) {
this(_idGenerator.getAndIncrement(), columns, values);
}
public MockInputRow(int id, InputColumn<?>[] columns, Object[] values) {
this(id);
for (int i = 0; i < values.length; i++) {
put(columns[i], values[i]);
}
}
@Override
public int getId() {
return _id;
}
@Override
public List<InputColumn<?>> getInputColumns() {
return new ArrayList<InputColumn<?>>(_values.keySet());
}
@Override
public boolean containsInputColumn(InputColumn<?> inputColumn) {
return _values.containsKey(inputColumn);
}
@SuppressWarnings("unchecked")
@Override
public <E> E getValueInternal(InputColumn<E> column) {
return (E) _values.get(column);
}
/**
* Puts/adds a new value to the row.
*
* @param column
* @param value
* @return
*/
public MockInputRow put(InputColumn<?> column, Object value) {
_values.put(column, value);
return this;
}
/**
* Puts/adds new values to the row.
*
* @param values
* @return
*/
public MockInputRow putAll(Map<InputColumn<?>, Object> values) {
_values.putAll(values);
return this;
}
@Override
public int hashCode() {
final int prime = 31;
int result = 1;
result = prime * result + _id;
result = prime * result + ((_values == null) ? 0 : _values.hashCode());
return result;
}
@Override
public boolean equals(Object obj) {
if (this == obj)
return true;
if (obj == null)
return false;
if (getClass() != obj.getClass())
return false;
MockInputRow other = (MockInputRow) obj;
if (_id != other._id)
return false;
if (_values == null) {
if (other._values != null)
return false;
} else if (!_values.equals(other._values))
return false;
return true;
}
@Override
public String toString() {
return "MockInputRow[id=" + _id + "]";
}
}