SourceProductSelectorTest.java

/*
 * Copyright (C) 2014 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 org.esa.snap.HeadlessTestRunner;
import org.esa.snap.core.datamodel.Product;
import org.esa.snap.ui.DefaultAppContext;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;

import javax.swing.JComboBox;
import java.io.File;

import static org.junit.Assert.*;

/**
 * @author Ralf Quast
 * @version $Revision$ $Date$
 */
@RunWith(HeadlessTestRunner.class)
public class SourceProductSelectorTest {

    private Product[] defaultProducts;
    private DefaultAppContext appContext;

    @Before
    public void setUp() throws Exception {
        appContext = new DefaultAppContext("Fart, fart!");
        defaultProducts = new Product[4];
        for (int i = 0; i < defaultProducts.length; i++) {
            defaultProducts[i] = new Product("P" + i, "T" + i, 10, 10);
            appContext.getProductManager().addProduct(defaultProducts[i]);
        }
        appContext.setSelectedProduct(defaultProducts[0]);
    }

    @Test
    public void testCreatedUIComponentsNotNull() {
        SourceProductSelector selector = new SourceProductSelector(appContext, "Source:");
        selector.initProducts();
        Assert.assertNotNull(selector.getProductNameLabel());
        Assert.assertNotNull(selector.getProductNameComboBox());
        Assert.assertNotNull(selector.getProductFileChooserButton());
    }

    @Test
    public void testCreatedUIComponentsAreSame() {
        SourceProductSelector selector = new SourceProductSelector(appContext, "Source:");
        selector.initProducts();
        assertSame(selector.getProductNameLabel(), selector.getProductNameLabel());
        assertSame(selector.getProductNameComboBox(), selector.getProductNameComboBox());
        assertSame(selector.getProductFileChooserButton(), selector.getProductFileChooserButton());
    }

    @Test
    public void testThatComboboxContains_4_EntriesIfEmptySelectionIsDisabled() {
        boolean enableEmptySelection = false;
        SourceProductSelector selector = new SourceProductSelector(appContext, "Source", enableEmptySelection);
        selector.initProducts();

        final JComboBox<Object> comboBox = selector.getProductNameComboBox();
        assertEquals(4, comboBox.getItemCount());
        assertEquals(comboBox.getItemAt(0), defaultProducts[0]);
        assertEquals(comboBox.getItemAt(1), defaultProducts[1]);
        assertEquals(comboBox.getItemAt(2), defaultProducts[2]);
        assertEquals(comboBox.getItemAt(3), defaultProducts[3]);
        assertEquals(4, selector.getProductCount());
    }

    @Test
    public void testThatComboboxContains_5_EntriesIfEmptySelectionIsEnabled() {
        boolean enableEmptySelection = true;
        SourceProductSelector selector = new SourceProductSelector(appContext, "Source", enableEmptySelection);
        selector.initProducts();

        final JComboBox<Object> comboBox = selector.getProductNameComboBox();
        assertEquals(5, comboBox.getItemCount());
        assertEquals(comboBox.getItemAt(0), null);
        assertEquals(comboBox.getItemAt(1), defaultProducts[0]);
        assertEquals(comboBox.getItemAt(2), defaultProducts[1]);
        assertEquals(comboBox.getItemAt(3), defaultProducts[2]);
        assertEquals(comboBox.getItemAt(4), defaultProducts[3]);
        assertEquals(4, selector.getProductCount());
    }

    @Test
    public void testSetSelectedProduct() throws Exception {
        SourceProductSelector selector = new SourceProductSelector(appContext, "Source");
        selector.initProducts();
        Product selectedProduct = selector.getSelectedProduct();
        assertSame(appContext.getSelectedProduct(), selectedProduct);

        selector.setSelectedProduct(defaultProducts[1]);
        selectedProduct = selector.getSelectedProduct();
        assertSame(defaultProducts[1], selectedProduct);

        Product oldProduct = new Product("new", "T1", 0, 0);
        oldProduct.setFileLocation(new File(""));
        selector.setSelectedProduct(oldProduct);
        selectedProduct = selector.getSelectedProduct();
        assertSame(oldProduct, selectedProduct);

        Product newProduct = new Product("new", "T2", 0, 0);
        selector.setSelectedProduct(newProduct);
        selectedProduct = selector.getSelectedProduct();
        assertSame(newProduct, selectedProduct);
        Assert.assertNull(oldProduct.getFileLocation()); // assert that old product is disposed
    }

    @Test
    public void testSelectedProductIsRemoved() {
        SourceProductSelector selector = new SourceProductSelector(appContext, "Source");
        selector.initProducts();
        appContext.getProductManager().removeProduct(defaultProducts[0]);
        Assert.assertEquals(defaultProducts.length - 1, selector.getProductCount());
    }

    @Test
    public void testNotSelectedProductIsRemoved() {
        SourceProductSelector selector = new SourceProductSelector(appContext, "Source");
        selector.initProducts();
        appContext.getProductManager().removeProduct(defaultProducts[2]);
        Assert.assertEquals(defaultProducts.length - 1, selector.getProductCount());
    }

    @Test
    public void testNewProductIsDisposed() throws Exception {
        SourceProductSelector selector = new SourceProductSelector(appContext, "Source");
        selector.initProducts();
        Product newProduct = new Product("new", "T1", 0, 0);
        newProduct.setFileLocation(new File(""));
        selector.setSelectedProduct(newProduct);
        assertSame(newProduct, selector.getSelectedProduct());
        selector.setSelectedProduct(defaultProducts[0]);
        assertSame(defaultProducts[0], selector.getSelectedProduct());

        Assert.assertNotNull(newProduct.getFileLocation());
        selector.releaseProducts();
        Assert.assertNull(newProduct.getFileLocation()); // assert that new product is disposed, because it is not selected
    }

    @Test
    public void testNewProductIsNotDisposed() throws Exception {
        SourceProductSelector selector = new SourceProductSelector(appContext, "Source");
        selector.initProducts();
        selector.setSelectedProduct(defaultProducts[0]);
        assertSame(defaultProducts[0], selector.getSelectedProduct());
        Product newProduct = new Product("new", "T1", 0, 0);
        newProduct.setFileLocation(new File(""));
        selector.setSelectedProduct(newProduct);
        assertSame(newProduct, selector.getSelectedProduct());

        Assert.assertNotNull(newProduct.getFileLocation());
        selector.releaseProducts();
        Assert.assertNotNull(newProduct.getFileLocation()); // assert that new product is not disposed while it is selected
    }

    @Test
    public void testSetSelectedIndex() throws Exception {
        SourceProductSelector selector = new SourceProductSelector(appContext, "Source");

        selector.initProducts();
        assertSame(defaultProducts[0], selector.getSelectedProduct());

        selector.setSelectedIndex(1);
        assertSame(defaultProducts[1], selector.getSelectedProduct());

        selector.setSelectedIndex(2);
        assertSame(defaultProducts[2], selector.getSelectedProduct());
    }
}