com.github.invictum.reportportal.FinishEventBuilder Maven / Gradle / Ivy

Go to download
package com.github.invictum.reportportal;

import com.epam.ta.reportportal.ws.model.FinishTestItemRQ;

import java.time.ZonedDateTime;
import java.time.temporal.ChronoUnit;
import java.util.Date;
import java.util.Objects;

/**
 * Support class that allows to construct {@link FinishTestItemRQ} event in fluent way
 */
public class FinishEventBuilder {

    private FinishTestItemRQ finishEvent = new FinishTestItemRQ();

    public FinishEventBuilder withEndTime(ZonedDateTime start, long duration) {
        ZonedDateTime end = start.plus(duration, ChronoUnit.MILLIS);
        finishEvent.setEndTime(Date.from(end.toInstant()));
        return this;
    }

    public FinishEventBuilder withStatus(Status status) {
        finishEvent.setStatus(status.toString());
        return this;
    }

    public FinishTestItemRQ build() {
        Objects.requireNonNull(finishEvent.getStatus(), "Status must not be null");
        Objects.requireNonNull(finishEvent.getEndTime(), "End date must not be null");
        return finishEvent;
    }
}