com.github.mkolisnyk.aerial.expressions.value.DateRangeValueExpression Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of aerial Show documentation
Generates Cucumber scenarios based on text descriptions
There is a newer version: 0.0.6.1
/**
 * .
 */
package com.github.mkolisnyk.aerial.expressions.value;

import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.Date;
import java.util.List;

import org.joda.time.DateTime;
import org.joda.time.Days;
import org.junit.Assert;

import com.github.mkolisnyk.aerial.document.InputRecord;
import com.github.mkolisnyk.aerial.expressions.ValueExpression;
import com.github.mkolisnyk.aerial.expressions.ValueExpressionType;
import com.github.mkolisnyk.aerial.util.Clock;
import com.github.mkolisnyk.aerial.util.SystemClock;

/**
 * @author Myk Kolisnyk
 *
 */
public class DateRangeValueExpression extends ValueExpression {
    private static final String OPEN_RANGE_BRACKET_PATTERN = "(\\[|\\()";
    private static final String CLOSE_RANGE_BRACKET_PATTERN = "(\\]|\\))";
    private static final String DATE_PATTERN = "([0-9\\-/]+)";
    private static final String SPACE_DELIMITER_PATTERN = "(\\s*)";
    private boolean includeLower = false;
    private boolean includeUpper = false;
    private String formatString;
    private Date lower;
    private Date upper;

    private Clock clock;

    /**
     * @param inputValue
     * @throws Exception .
     */
    public DateRangeValueExpression(InputRecord inputValue) throws Exception {
        this(inputValue, new SystemClock());
    }

    public DateRangeValueExpression(InputRecord inputValue, Clock clockValue) throws Exception {
        super(inputValue);
        this.clock = clockValue;
        this.lower = clock.now();
        this.upper = clock.now();
        this.formatString = "dd-MM-yyyy";
    }

    /**
     * @param clockValue the clock to set
     */
    public final void setClock(Clock clockValue) {
        this.clock = clockValue;
    }

    /* (non-Javadoc)
     * @see com.github.mkolisnyk.aerial.expressions.ValueExpression#generate()
     */
    @Override
    public List generate() throws Exception {
        List result = new ArrayList();
        this.validate();

        InputRecord inputValue = (InputRecord) this.getInput().clone();
        Calendar calendar = Calendar.getInstance();
        SimpleDateFormat format = new SimpleDateFormat(this.formatString);
        DateTime start = new DateTime(lower);
        DateTime end = new DateTime(upper);

        // Valid in range value
        int range = Days.daysBetween(start, end).getDays();
        calendar.setTime(lower);
        calendar.add(Calendar.DAY_OF_YEAR, range / 2);
        inputValue.setValue(format.format(calendar.getTime()));
        inputValue.setValidInput(true);
        result.add(inputValue);
        // Another valid value
        if (range > 1) {
            inputValue = (InputRecord) this.getInput().clone();
            calendar.setTime(lower);
            calendar.add(Calendar.DAY_OF_YEAR, range / 2 + 1);
            inputValue.setValue(format.format(calendar.getTime()));
            inputValue.setValidInput(true);
            result.add(inputValue);
        }
        // Lower border value
        inputValue = (InputRecord) this.getInput().clone();
        inputValue.setValue(format.format(lower));
        inputValue.setValidInput(this.includeLower);
        result.add(inputValue);
        // Upper border value
        inputValue = (InputRecord) this.getInput().clone();
        inputValue.setValue(format.format(upper));
        inputValue.setValidInput(this.includeUpper);
        result.add(inputValue);
        // Day before lower
        inputValue = (InputRecord) this.getInput().clone();
        calendar.setTime(lower);
        calendar.add(Calendar.DAY_OF_YEAR, -1);
        inputValue.setValue(format.format(calendar.getTime()));
        inputValue.setValidInput(false);
        result.add(inputValue);
        // Day after upper
        inputValue = (InputRecord) this.getInput().clone();
        calendar.setTime(upper);
        calendar.add(Calendar.DAY_OF_YEAR, 1);
        inputValue.setValue(format.format(calendar.getTime()));
        inputValue.setValidInput(false);
        result.add(inputValue);

        return result;
    }

    /* (non-Javadoc)
     * @see com.github.mkolisnyk.aerial.expressions.ValueExpression#getValueType()
     */
    @Override
    public ValueExpressionType getValueType() {
        return ValueExpressionType.DATE;
    }

    /* (non-Javadoc)
     * @see com.github.mkolisnyk.aerial.expressions.ValueExpression#getMatchPattern()
     */
    @Override
    public String getMatchPattern() {
        String dateMatchPattern = "([mMdDnYyhHsS\\-/:]+)";
        return String.format("%s%s%s%s;%s%s%s%s, Format: %s%s",
                OPEN_RANGE_BRACKET_PATTERN,
                SPACE_DELIMITER_PATTERN,
                DATE_PATTERN,
                SPACE_DELIMITER_PATTERN,
                SPACE_DELIMITER_PATTERN,
                DATE_PATTERN,
                SPACE_DELIMITER_PATTERN,
                CLOSE_RANGE_BRACKET_PATTERN,
                SPACE_DELIMITER_PATTERN,
                dateMatchPattern);
    }

    /* (non-Javadoc)
     * @see com.github.mkolisnyk.aerial.expressions.ValueExpression#validate()
     */
    @Override
    public void validate() throws Exception {
        super.validate();
        this.parse();
        Assert.assertTrue("The upper date should be after the initial date",
                this.upper.after(this.lower));
    }

    public void parse() throws Exception {
        String input = this.getInput().getValue();
        String pattern = this.getMatchPattern();
        this.includeLower = input.replaceFirst(pattern, "$1").equals("[");
        this.includeUpper = input.replaceFirst(pattern, "$8").equals("]");
        this.formatString = input.replaceFirst(pattern, "$10");
        SimpleDateFormat format = new SimpleDateFormat(this.formatString);
        String lowerString = input.replaceFirst(pattern, "$3");
        String upperString = input.replaceFirst(pattern, "$6");
        this.lower = format.parse(lowerString);
        this.upper = format.parse(upperString);
    }

    /* (non-Javadoc)
     * @see java.lang.Object#toString()
     */
    @Override
    public String toString() {
        String start = "(";
        String end = ")";
        if (includeLower) {
            start = "[";
        }
        if (includeUpper) {
            end = "]";
        }
        SimpleDateFormat format = new SimpleDateFormat("dd-MM-yyyy");
        return String.format("%s%s;%s%s,Format:%s",
                start,
                format.format(lower),
                format.format(upper),
                end,
                format.toPattern());
    }
}