JIMRI/java/test/jmri/jmrit/operations/trains/TrainIconTest.java

package jmri.jmrit.operations.trains;

import java.awt.Color;
import java.awt.GraphicsEnvironment;

import org.junit.Assert;
import org.junit.Assume;
import org.junit.jupiter.api.*;

import jmri.jmrit.operations.OperationsTestCase;
import jmri.util.JUnitUtil;

/**
 *
 * @author Paul Bender Copyright (C) 2017
 */
public class TrainIconTest extends OperationsTestCase {

    private jmri.jmrit.display.EditorScaffold editor = null;
    private TrainIcon trainicon = null;

    @Test
    public void testCTor() {
        Assume.assumeFalse(GraphicsEnvironment.isHeadless());
        Assert.assertNotNull("exists",trainicon);
    }

    // test TrainIcon attributes
    @Test
    public void testTrainIconAttributes() {
        Assume.assumeFalse(GraphicsEnvironment.isHeadless());
        Train train1 = new Train("TESTTRAINID", "TESTNAME");
        trainicon.setTrain(train1);
        Assert.assertEquals("TrainIcon set train", "TESTNAME", trainicon.getTrain().getName());
    }

    @Test
    public void testTrainIconColorChangeAttributes() {
        Assume.assumeFalse(GraphicsEnvironment.isHeadless());
        // test color change
        for (String color : TrainIcon.getLocoColors()) {
            trainicon.setLocoColor(color);
            if (color.equals(TrainIcon.WHITE)) {
                Assert.assertEquals("White train icon", Color.WHITE, trainicon.getLocoColor());
            }
            if (color.equals(TrainIcon.GREEN)) {
                Assert.assertEquals("Green train icon", Color.GREEN, trainicon.getLocoColor());
            }
            if (color.equals(TrainIcon.GRAY)) {
                Assert.assertEquals("Gray train icon", Color.GRAY, trainicon.getLocoColor());
            }
            if (color.equals(TrainIcon.RED)) {
                Assert.assertEquals("Red train icon", Color.RED, trainicon.getLocoColor());
            }
            if (color.equals(TrainIcon.YELLOW)) {
                Assert.assertEquals("Yellow train icon", Color.YELLOW, trainicon.getLocoColor());
            }
            if (color.equals(TrainIcon.BLUE)) {
                Assert.assertEquals("Blue train icon", TrainIcon.COLOR_BLUE, trainicon.getLocoColor());
            }
        }
    }

    @Test
    public void testNumberColors(){
        // confirm that there are 6 icon colors
        Assert.assertEquals("Six colors", 6, TrainIcon.getLocoColors().length);
    }

    @Override
    @BeforeEach
    public void setUp() {
        super.setUp();
        if(!GraphicsEnvironment.isHeadless()){
           editor = new jmri.jmrit.display.EditorScaffold();
           trainicon = editor.addTrainIcon("TestName");
        }
    }

    @Override
    @AfterEach
    public void tearDown() {
        if(editor!=null){
           JUnitUtil.dispose(editor);
        }
        editor = null;
        trainicon = null;
        JUnitUtil.deregisterBlockManagerShutdownTask();
        super.tearDown();
    }

    // private static final Logger log = LoggerFactory.getLogger(TrainIconTest.class);

}