SweCneExporterTest.java

/*
 * Copyright (c) 2024, 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/.
 */


package com.powsybl.openrao.data.raoresult.io.cne.swe;

import com.powsybl.openrao.commons.OpenRaoException;
import com.powsybl.openrao.data.crac.api.CracCreationContext;
import com.powsybl.openrao.data.crac.io.cim.craccreator.CimCracCreationContext;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;

import java.util.Properties;
import java.util.Set;

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

/**
 * @author Thomas Bouquet {@literal <thomas.bouquet at rte-france.com>}
 */
class SweCneExporterTest {
    private final SweCneExporter exporter = new SweCneExporter();

    @Test
    void testFormat() {
        assertEquals("SWE-CNE", exporter.getFormat());
    }

    @Test
    void testExportWithWrongCracCreationContext() {
        CracCreationContext cracCreationContext = Mockito.mock(CracCreationContext.class);
        OpenRaoException exception = assertThrows(OpenRaoException.class, () -> exporter.exportData(null, cracCreationContext, null, null));
        assertEquals("SWE-CNE exporter expects a CimCracCreationContext.", exception.getMessage());
    }

    @Test
    void testProperties() {
        assertEquals(Set.of("rao-result.export.swe-cne.document-id", "rao-result.export.swe-cne.revision-number", "rao-result.export.swe-cne.domain-id", "rao-result.export.swe-cne.process-type", "rao-result.export.swe-cne.sender-id", "rao-result.export.swe-cne.sender-role", "rao-result.export.swe-cne.receiver-id", "rao-result.export.swe-cne.receiver-role", "rao-result.export.swe-cne.time-interval"), exporter.getRequiredProperties());
    }

    @Test
    void testCracCreationContextClass() {
        assertEquals(CimCracCreationContext.class, exporter.getCracCreationContextClass());
    }

    @Test
    void testWrongCracCreationContextClass() {
        OpenRaoException exception = assertThrows(OpenRaoException.class, () -> exporter.validateDataToExport(Mockito.mock(CracCreationContext.class), null));
        assertEquals("SWE-CNE exporter expects a CimCracCreationContext.", exception.getMessage());
    }

    @Test
    void testNullProperties() {
        OpenRaoException exception = assertThrows(OpenRaoException.class, () -> exporter.validateDataToExport(Mockito.mock(CimCracCreationContext.class), null));
        assertEquals("The export properties cannot be null for SWE-CNE export.", exception.getMessage());
    }

    @Test
    void testMissingRequiredProperty() {
        Properties properties = new Properties();
        properties.setProperty("rao-result.export.swe-cne.document-id", "");
        properties.setProperty("rao-result.export.swe-cne.revision-number", "");
        properties.setProperty("rao-result.export.swe-cne.domain-id", "");
        properties.setProperty("rao-result.export.swe-cne.sender-id", "");
        properties.setProperty("rao-result.export.swe-cne.sender-role", "");
        properties.setProperty("rao-result.export.swe-cne.receiver-id", "");
        properties.setProperty("rao-result.export.swe-cne.receiver-role", "");
        properties.setProperty("rao-result.export.swe-cne.time-interval", "");
        OpenRaoException exception = assertThrows(OpenRaoException.class, () -> exporter.validateDataToExport(Mockito.mock(CimCracCreationContext.class), properties));
        assertEquals("The mandatory rao-result.export.swe-cne.process-type property is missing for SWE-CNE export.", exception.getMessage());
    }
}