package org.openlmis.referencedata.util;

import static org.hamcrest.Matchers.greaterThan;
import static org.hamcrest.Matchers.is;
import static org.hamcrest.Matchers.lessThan;
import static org.junit.Assert.assertThat;

import org.junit.Before;
import org.junit.Test;
import org.openlmis.referencedata.dto.ProcessingPeriodDto;
import org.openlmis.referencedata.exception.ValidationMessageException;

import java.time.LocalDate;

public class ProcessingPeriodDtoComparatorTest {

  private ProcessingPeriodDto firstPeriod;
  private ProcessingPeriodDto secondPeriod;
  private ProcessingPeriodDtoComparator comparator;

  @Before
  public void setUp() {
    firstPeriod = new ProcessingPeriodDto();
    secondPeriod = new ProcessingPeriodDto();
  }

  @Test
  public void shouldReturnNegativeIntWhenFirstStartDateBeforeSecond() {
    comparator = new ProcessingPeriodDtoComparator("startDate");

    firstPeriod.setStartDate(LocalDate.of(2016, 1, 1));
    secondPeriod.setStartDate(LocalDate.of(2016, 2, 1));

    assertThat(comparator.compare(firstPeriod, secondPeriod), lessThan(0));
  }

  @Test
  public void shouldReturnPositiveIntWhenFirstStartDateAfterSecond() {
    comparator = new ProcessingPeriodDtoComparator("startDate");

    firstPeriod.setStartDate(LocalDate.of(2016, 2, 1));
    secondPeriod.setStartDate(LocalDate.of(2016, 1, 1));

    assertThat(comparator.compare(firstPeriod, secondPeriod), greaterThan(0));
  }

  @Test
  public void shouldReturnZeroWhenFirstStartDateEqualsSecond() {
    comparator = new ProcessingPeriodDtoComparator("startDate");

    firstPeriod.setStartDate(LocalDate.of(2016, 1, 1));
    secondPeriod.setStartDate(LocalDate.of(2016, 1, 1));

    assertThat(comparator.compare(firstPeriod, secondPeriod), is(0));
  }

  @Test(expected = ValidationMessageException.class)
  public void shouldThrowExceptionWhenInvalidSortingColumn() {
    comparator = new ProcessingPeriodDtoComparator("");

    firstPeriod.setStartDate(LocalDate.of(2016, 1, 1));
    secondPeriod.setStartDate(LocalDate.of(2016, 1, 1));

    comparator.compare(firstPeriod, secondPeriod);
  }
}
@@ -1,8 +1,863 @@
+/*%0A * This program is part of the OpenLMIS logistics management information system platform software.%0A * Copyright %C2%A9 2017 VillageReach%0A *%0A * This program is free software: you can redistribute it and/or modify it under the terms%0A * of the GNU Affero General Public License as published by the Free Software Foundation, either%0A * version 3 of the License, or (at your option) any later version.%0A * %C2%A0%0A * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;%0A * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.%C2%A0%0A * See the GNU Affero General Public License for more details. You should have received a copy of%0A * the GNU Affero General Public License along with this program. If not, see%0A * http://www.gnu.org/licenses. %C2%A0For additional information contact info@OpenLMIS.org.%C2%A0%0A */%0A%0A
 package 

add header