UcteNetworkExtTest.java

/**
 * Copyright (c) 2017, 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.ucte.network.ext;

import com.powsybl.ucte.network.*;
import org.junit.jupiter.api.Test;

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

/**
 * @author Mathieu Bague {@literal <mathieu.bague at rte-france.com>}
 */
class UcteNetworkExtTest extends UcteNetworkImplTest {

    @Test
    void test() {
        UcteNetworkExt network = (UcteNetworkExt) UcteNetworkFactory.createNetwork(() -> new UcteNetworkExt(new UcteNetworkImpl(), 0.01));
        testNetwork(network);
    }

    private void testNetwork(UcteNetworkExt network) {
        super.testNetwork(network);

        UcteNodeCode code1 = new UcteNodeCode(UcteCountryCode.XX, "AAAAA", UcteVoltageLevelCode.VL_380, '1');
        UcteNodeCode code2 = new UcteNodeCode(UcteCountryCode.XX, "BBBBB", UcteVoltageLevelCode.VL_220, '1');

        assertEquals(2, network.getSubstations().size());
        assertEquals(network.getVoltageLevel(code1).getSubstation(), network.getVoltageLevel(code2).getSubstation());
    }
}