/*
 * Copyright 2017 The Mifos Initiative.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.mifos.portfolio.api.v1.domain;

import io.mifos.core.test.domain.ValidationTest;
import io.mifos.core.test.domain.ValidationTestCase;
import org.apache.commons.lang.RandomStringUtils;
import org.junit.runners.Parameterized;

import java.util.ArrayList;
import java.util.Collection;

/**
 * @author Myrle Krantz
 */
public class TaskInstanceTest extends ValidationTest<TaskInstance> {

  public TaskInstanceTest(ValidationTestCase<TaskInstance> testCase) {
    super(testCase);
  }

  @Override
  protected TaskInstance createValidTestSubject() {
    final TaskInstance ret = new TaskInstance();
    ret.setComment("Hey how y'all doin'.");
    ret.setTaskIdentifier("validIdentifier");
    return ret;
  }

  @Parameterized.Parameters
  public static Collection testCases() {
    final Collection<ValidationTestCase> ret = new ArrayList<>();

    ret.add(new ValidationTestCase<TaskInstance>("valid")
        .adjustment(x -> {})
        .valid(true));
    ret.add(new ValidationTestCase<TaskInstance>("nullIdentifier")
        .adjustment(x -> x.setTaskIdentifier(null))
        .valid(false));
    ret.add(new ValidationTestCase<TaskInstance>("null Comment")
        .adjustment(x -> x.setComment(null))
        .valid(true));
    ret.add(new ValidationTestCase<TaskInstance>("too long Comment")
        .adjustment(x -> x.setComment(RandomStringUtils.random(5000)))
        .valid(false));

    return ret;
  }
}