package jmri.jmrix.sprog.update;

import java.awt.GraphicsEnvironment;

import jmri.jmrix.sprog.SprogSystemConnectionMemo;
import jmri.jmrix.sprog.SprogTrafficControlScaffold;
import jmri.util.JUnitUtil;

import org.junit.jupiter.api.*;

/**
 *
 * @author Paul Bender Copyright (C) 2017
 */
public class SprogIIUpdateFrameTest extends jmri.util.JmriJFrameTestBase {

    private SprogTrafficControlScaffold stcs = null;
    private SprogSystemConnectionMemo m = null;

    @BeforeEach
    @Override
    public void setUp() {
        JUnitUtil.setUp();
        m = new jmri.jmrix.sprog.SprogSystemConnectionMemo(jmri.jmrix.sprog.SprogConstants.SprogMode.OPS);
        stcs = new SprogTrafficControlScaffold(m);
        m.setSprogTrafficController(stcs);
        m.configureCommandStation();
        if (!GraphicsEnvironment.isHeadless()) {
            frame = new SprogIIUpdateFrame(m);
        }
    }

    @AfterEach
    @Override
    public void tearDown() {
        // we need to close window before the Traffic Controller.
        if(frame!=null) {
           JUnitUtil.dispose(frame); // frame dispose stops Update timer.
        }
        frame = null;
        m.getSlotThread().interrupt();
        JUnitUtil.waitFor(() -> {return m.getSlotThread().getState() == Thread.State.TERMINATED;}, "Slot thread failed to stop");
        m.dispose();
        stcs.dispose();
        m = null;
        stcs = null;
        super.tearDown();
    }

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

}