/*******************************************************************************
 * (c) Copyright 2014 Hewlett-Packard Development Company, L.P.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Apache License v2.0 which accompany this distribution.
 *
 * The Apache License is available at
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 *******************************************************************************/
package org.openscore.lang.entities;

import org.apache.commons.lang3.Validate;
import org.apache.commons.lang3.builder.EqualsBuilder;
import org.apache.commons.lang3.builder.HashCodeBuilder;

import java.io.Serializable;

public class LoopStatement implements Serializable{

    public enum Type {
        FOR, WHILE
    }

    private final String varName;
    private final String collectionExpression;
    private final Type type;

    public LoopStatement(String varName, String collectionExpression, Type type) {
        Validate.notBlank(varName, "var name cannot be empty");
        Validate.notBlank(collectionExpression, "collection expression cannot be empty");
        Validate.notNull(type, "type cannot be null");
        //todo add tests

        this.varName = varName;
        this.collectionExpression = collectionExpression;
        this.type = type;
    }

    public String getVarName() {
        return varName;
    }

    public String getCollectionExpression() {
        return collectionExpression;
    }

    public Type getType() {
        return type;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o)
            return true;
        if (o == null || getClass() != o.getClass())
            return false;

        LoopStatement that = (LoopStatement) o;

        return new EqualsBuilder()
                .append(varName, that.varName)
                .append(collectionExpression, that.collectionExpression)
                .append(type, that.type)
                .isEquals();
    }

    @Override
    public int hashCode() {
        return new HashCodeBuilder()
                .append(varName)
                .append(collectionExpression)
                .append(type)
                .toHashCode();
    }
}

@@ -1033,16 +1033,25 @@
 rName, %22
+for loop 
 var name
@@ -1119,16 +1119,25 @@
 ssion, %22
+for loop 
 collecti

fix error message