PostContingencyResultComparatorTest.java

/**
 * Copyright (c) 2018, RTE (http://www.rte-france.com)
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 * SPDX-License-Identifier: MPL-2.0
 */
package com.powsybl.security.comparator;

import static org.junit.jupiter.api.Assertions.assertEquals;

import java.util.Arrays;
import java.util.Collections;
import java.util.List;

import com.powsybl.security.PostContingencyComputationStatus;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;

import com.powsybl.contingency.Contingency;
import com.powsybl.security.results.PostContingencyResult;

/**
 *
 * @author Massimo Ferraro {@literal <massimo.ferraro@techrain.eu>}
 */
class PostContingencyResultComparatorTest {

    @Test
    void compare() {
        Contingency contingency1 = Mockito.mock(Contingency.class);
        Mockito.when(contingency1.getId()).thenReturn("contingency1");
        PostContingencyResult result1 = new PostContingencyResult(contingency1, PostContingencyComputationStatus.CONVERGED, Collections.emptyList());
        Contingency contingency2 = Mockito.mock(Contingency.class);
        Mockito.when(contingency2.getId()).thenReturn("contingency2");
        PostContingencyResult result2 = new PostContingencyResult(contingency2, PostContingencyComputationStatus.CONVERGED, Collections.emptyList());
        Contingency contingency3 = Mockito.mock(Contingency.class);
        Mockito.when(contingency3.getId()).thenReturn("contingency3");
        PostContingencyResult result3 = new PostContingencyResult(contingency3, PostContingencyComputationStatus.CONVERGED, Collections.emptyList());

        List<PostContingencyResult> results = Arrays.asList(result3, result1, result2);
        Collections.sort(results, new PostContingencyResultComparator());

        assertEquals(result1, results.get(0));
        assertEquals(result2, results.get(1));
        assertEquals(result3, results.get(2));
    }

}