TargetProductSelectorModelTest.java

/*
 * Copyright (C) 2010 Brockmann Consult GmbH (info@brockmann-consult.de)
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the Free
 * Software Foundation; either version 3 of the License, or (at your option)
 * any later version.
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, see http://www.gnu.org/licenses/
 */

package org.esa.snap.core.gpf.ui;

import com.bc.ceres.core.ProgressMonitor;
import org.esa.snap.core.dataio.AbstractProductWriter;
import org.esa.snap.core.dataio.EncodeQualification;
import org.esa.snap.core.dataio.ProductIOPlugInManager;
import org.esa.snap.core.dataio.ProductWriter;
import org.esa.snap.core.dataio.ProductWriterPlugIn;
import org.esa.snap.core.datamodel.Band;
import org.esa.snap.core.datamodel.Product;
import org.esa.snap.core.datamodel.ProductData;
import org.esa.snap.core.util.io.SnapFileFilter;
import org.junit.Before;
import org.junit.Test;

import java.io.File;
import java.io.IOException;
import java.util.Locale;

import static org.junit.Assert.*;

/**
 * Tests for class {@link TargetProductSelectorModel}.
 *
 * @author Ralf Quast
 * @version $Revision$ $Date$
 */
public class TargetProductSelectorModelTest {

    private TargetProductSelectorModel model;
    private DummyTestProductWriterPlugIn writerPlugIn;

    @Before
    public void setUp() throws Exception {
        writerPlugIn = new DummyTestProductWriterPlugIn();
        ProductIOPlugInManager.getInstance().addWriterPlugIn(writerPlugIn);
        model = new TargetProductSelectorModel();
    }

    @Test
    public void testSetGetProductName() {
        model.setProductName("Obelix");
        assertEquals("Obelix", model.getProductName());
    }

    @Test
    public void testSetGetInvalidProductName() {
        assertNull(model.getProductName());
        try {
            model.setProductName("Obel/x");
            fail();
        } catch (IllegalArgumentException e) {
            assertTrue(e.getMessage().startsWith("The product name 'Obel/x' is not valid"));
        }
        assertNull(model.getProductName());
    }

    @Test
    public void testSetGetFormatName() {
        model.setFormatName("Majestix");
        assertEquals("Majestix", model.getFormatName());
    }

    @Test
    public void testGetFileName() {
        model.setProductName("Obelix");
        assertEquals("Obelix", model.getProductName());
        assertEquals("Obelix.dim", model.getProductFileName());

        // other format
        model.setFormatName(writerPlugIn.getFormatNames()[0]);
        assertEquals("Obelix.x", model.getProductFileName());

        model.setProductName("Idefix.dim");
        assertEquals("Idefix.dim.x", model.getProductFileName());

        model.setProductName("Idefix.x");
        assertEquals("Idefix.x", model.getProductFileName());
    }

    @Test
    public void testSetGetDirectory() {
        final File directory = new File("Gallien");
        model.setProductDir(directory);
        assertEquals(directory, model.getProductDir());

        model.setProductName("Obelix");
        assertEquals(new File("Gallien", "Obelix.dim"), model.getProductFile());

        // other format
        model.setFormatName(writerPlugIn.getFormatNames()[0]);
        assertEquals(new File("Gallien", "Obelix.x"), model.getProductFile());
    }

    @Test
    public void testSelections() {
        assertTrue(model.isSaveToFileSelected());
        assertTrue(model.isOpenInAppSelected());

        model.setOpenInAppSelected(false);
        assertFalse(model.isOpenInAppSelected());
        assertTrue(model.isSaveToFileSelected());

        model.setSaveToFileSelected(false);
        assertFalse(model.isSaveToFileSelected());
        assertTrue(model.isOpenInAppSelected());
        model.setOpenInAppSelected(false);
        assertFalse(model.isOpenInAppSelected());
        assertTrue(model.isSaveToFileSelected());
    }

    private class DummyTestProductWriter extends AbstractProductWriter {

        public DummyTestProductWriter(ProductWriterPlugIn writerPlugIn) {
            super(writerPlugIn);
        }

        @Override
        protected void writeProductNodesImpl() throws IOException {

        }

        public void writeBandRasterData(Band sourceBand, int sourceOffsetX, int sourceOffsetY, int sourceWidth,
                                        int sourceHeight, ProductData sourceBuffer, ProgressMonitor pm) throws
                IOException {

        }

        public void flush() throws IOException {

        }

        public void close() throws IOException {

        }

        public void deleteOutput() throws IOException {

        }
    }

    private class DummyTestProductWriterPlugIn implements ProductWriterPlugIn {

        @Override
        public EncodeQualification getEncodeQualification(Product product) {
            return EncodeQualification.FULL;
        }

        public Class[] getOutputTypes() {
            return new Class[]{String.class};
        }

        public ProductWriter createWriterInstance() {
            return new DummyTestProductWriter(this);
        }

        public String[] getFormatNames() {
            return new String[]{"Dummy"};
        }

        public String[] getDefaultFileExtensions() {
            return new String[]{".x"};
        }

        public String getDescription(Locale locale) {
            return "";
        }

        public SnapFileFilter getProductFileFilter() {
            return new SnapFileFilter();
        }
    }
}