UnitSelection.java

public class UnitSelection {

	private Unit selectedUnit;
	private List<Unit> units;
	private boolean renderSelection;

	public UnitSelection(List<Unit> units) {
		this.units = units;
		validateNotEmpty();
		setupPreselectedAndRenderSelection();
	}

	private void setupPreselectedAndRenderSelection() {
		if (units.size() == 1) {
			setSelectedUnit(units.get(0));
			this.units = null;
		} else {
			renderSelection = true;
		}
	}
	protected void validateNotEmpty() {
		if (CollectionUtils.isEmpty(this.units)) {
			throw new IllegalArgumentException("At least one unit is required");
		}
	}
	public Unit getSelectedUnit() {
		return selectedUnit;
	}
	public void setSelectedUnit(Unit selectedUnit) {
		this.selectedUnit = selectedUnit;
	}
	public boolean isRenderSelection() {
		return renderSelection;
	}

}