package jmri.jmrix.mqtt.logixng.configurexml; import static org.junit.jupiter.api.Assertions.assertEquals; import static org.junit.jupiter.api.Assertions.assertNotNull; import static org.junit.jupiter.api.Assertions.assertTrue; import jmri.jmrit.logixng.actions.IfThenElse; import java.beans.PropertyVetoException; import java.io.*; import java.nio.charset.StandardCharsets; import java.util.*; import jmri.*; import jmri.jmrit.logixng.*; import jmri.jmrit.logixng.actions.DigitalMany; import jmri.jmrit.logixng.expressions.And; import jmri.jmrit.logixng.util.LogixNG_Thread; import jmri.jmrit.logixng.util.parser.ParserException; import jmri.jmrix.mqtt.MqttSystemConnectionMemo; import jmri.jmrix.mqtt.logixng.Publish; import jmri.jmrix.mqtt.logixng.Subscribe; import jmri.util.*; import jmri.util.junit.annotations.DisabledIfHeadless; import org.apache.commons.lang3.mutable.MutableInt; import org.junit.jupiter.api.*; /** * Creates a LogixNG with all actions and expressions to test store and load. *

* It uses the Base.printTree(PrintWriter writer, String indent) method to * compare the LogixNGs before and after store and load. * * @author Daniel Bergqvist Copyright 2022 */ public class StoreAndLoadTest { private MqttSystemConnectionMemo memo1; // private MqttSystemConnectionMemo memo2; @Test @DisabledIfHeadless public void testLogixNGs() throws PropertyVetoException, SocketAlreadyConnectedException, ParserException, IOException, JmriException { LogixNG_Manager logixNG_Manager = InstanceManager.getDefault(LogixNG_Manager.class); ConditionalNG_Manager conditionalNGManager = InstanceManager.getDefault(ConditionalNG_Manager.class); DigitalActionManager digitalActionManager = InstanceManager.getDefault(DigitalActionManager.class); DigitalExpressionManager digitalExpressionManager = InstanceManager.getDefault(DigitalExpressionManager.class); LogixNG logixNG = logixNG_Manager.createLogixNG("A logixNG"); ConditionalNG conditionalNG = conditionalNGManager.createConditionalNG(logixNG, "A conditionalNG"); logixNG.setEnabled(false); conditionalNG.setEnabled(true); FemaleSocket femaleSocket = conditionalNG.getFemaleSocket(); IfThenElse ifThenElse = new IfThenElse(digitalActionManager.getAutoSystemName(), null); MaleSocket maleSocket = digitalActionManager.registerAction(ifThenElse); femaleSocket.connect(maleSocket); And and = new And(digitalExpressionManager.getAutoSystemName(), null); and.setComment("A comment"); maleSocket = digitalExpressionManager.registerExpression(and); ifThenElse.getChild(0).connect(maleSocket); DigitalMany many = new DigitalMany(digitalActionManager.getAutoSystemName(), null); many.setComment("A comment"); maleSocket = digitalActionManager.registerAction(many); ifThenElse.getChild(1).connect(maleSocket); Publish publish = new Publish(digitalActionManager.getAutoSystemName(), null, memo1); publish.setComment("A comment"); maleSocket = digitalActionManager.registerAction(publish); many.getChild(0).connect(maleSocket); publish = new Publish(digitalActionManager.getAutoSystemName(), null, memo1); publish.setComment("A comment"); publish.getSelectTopic().setAddressing(NamedBeanAddressing.Direct); publish.getSelectTopic().setValue("TheTopic"); publish.getSelectMessage().setAddressing(NamedBeanAddressing.Direct); publish.getSelectMessage().setValue("TheMessage"); maleSocket = digitalActionManager.registerAction(publish); many.getChild(1).connect(maleSocket); Subscribe subscribe = new Subscribe(digitalActionManager.getAutoSystemName(), null, memo1); subscribe.setComment("A comment"); maleSocket = digitalActionManager.registerAction(subscribe); many.getChild(2).connect(maleSocket); subscribe = new Subscribe(digitalActionManager.getAutoSystemName(), null, memo1); subscribe.setComment("A comment"); subscribe.setSubscribeToTopic("TheTopic"); subscribe.setLastTopicLocalVariable("TopicVariable"); subscribe.setLastMessageLocalVariable("MessageVariable"); maleSocket = digitalActionManager.registerAction(subscribe); many.getChild(3).connect(maleSocket); /* if (1==1) { final String treeIndent = " "; StringWriter stringWriter = new StringWriter(); PrintWriter printWriter = new PrintWriter(stringWriter); logixNG_Manager.printTree(Locale.ENGLISH, printWriter, treeIndent); System.out.println("--------------------------------------------"); System.out.println("The current tree:"); System.out.println("XXX"+stringWriter.toString()+"XXX"); System.out.println("--------------------------------------------"); System.out.println("--------------------------------------------"); System.out.println("--------------------------------------------"); System.out.println("--------------------------------------------"); System.out.println("--------------------------------------------"); System.out.println("--------------------------------------------"); System.out.println("--------------------------------------------"); System.out.println("--------------------------------------------"); System.out.println("--------------------------------------------"); System.out.println("--------------------------------------------"); log.error("--------------------------------------------"); log.error("The current tree:"); log.error("XXX"+stringWriter.toString()+"XXX"); log.error("--------------------------------------------"); log.error("--------------------------------------------"); log.error("--------------------------------------------"); log.error("--------------------------------------------"); log.error("--------------------------------------------"); log.error("--------------------------------------------"); log.error("--------------------------------------------"); log.error("--------------------------------------------"); log.error("--------------------------------------------"); log.error("--------------------------------------------"); log.error("--------------------------------------------"); // return; } */ // Store panels jmri.ConfigureManager cm = InstanceManager.getNullableDefault(jmri.ConfigureManager.class); assertNotNull( cm, "Unable to get default configure manager"); FileUtil.createDirectory(FileUtil.getUserFilesPath() + "temp"); File firstFile = new File(FileUtil.getUserFilesPath() + "temp/" + "LogixNG_temp.xml"); File secondFile = new File(FileUtil.getUserFilesPath() + "temp/" + "LogixNG.xml"); log.info("Temporary first file: {}", firstFile.getAbsoluteFile()); log.info("Temporary second file: {}", secondFile.getAbsoluteFile()); final String treeIndent = " "; StringWriter stringWriter = new StringWriter(); PrintWriter printWriter = new PrintWriter(stringWriter); logixNG_Manager.printTree(Locale.ENGLISH, printWriter, treeIndent, new MutableInt(0)); final String originalTree = stringWriter.toString(); assertTrue( cm.storeUser(firstFile), () -> "Failed to store panel. Temporary first file: " + firstFile.getAbsoluteFile() + " Temporary second file: " + secondFile.getAbsoluteFile()); // Add the header comment to the xml file addHeader(firstFile, secondFile); //********************************** // Delete all the LogixNGs, ConditionalNGs, and so on before reading the file. //********************************** java.util.Set logixNG_Set = new java.util.HashSet<>(logixNG_Manager.getNamedBeanSet()); for (LogixNG aLogixNG : logixNG_Set) { logixNG_Manager.deleteLogixNG(aLogixNG); } java.util.Set conditionalNGSet = new java.util.HashSet<>(conditionalNGManager.getNamedBeanSet()); for (ConditionalNG aConditionalNG : conditionalNGSet) { conditionalNGManager.deleteConditionalNG(aConditionalNG); } java.util.Set digitalActionSet = new java.util.HashSet<>(digitalActionManager.getNamedBeanSet()); for (MaleDigitalActionSocket aDigitalActionSocket : digitalActionSet) { digitalActionManager.deleteDigitalAction(aDigitalActionSocket); } java.util.Set digitalExpressionSet = new java.util.HashSet<>(digitalExpressionManager.getNamedBeanSet()); for (MaleDigitalExpressionSocket aDigitalExpression : digitalExpressionSet) { digitalExpressionManager.deleteDigitalExpression(aDigitalExpression); } assertEquals(0, logixNG_Manager.getNamedBeanSet().size()); assertEquals(0, conditionalNGManager.getNamedBeanSet().size()); assertEquals(0, digitalActionManager.getNamedBeanSet().size()); assertEquals(0, digitalExpressionManager.getNamedBeanSet().size()); LogixNG_Thread.stopAllLogixNGThreads(); LogixNG_Thread.assertLogixNGThreadNotRunning(); //********************************** // Try to load file //********************************** java.util.Set conditionalNG_Set = new java.util.HashSet<>(conditionalNGManager.getNamedBeanSet()); for (ConditionalNG aConditionalNG : conditionalNG_Set) { conditionalNGManager.deleteConditionalNG(aConditionalNG); } java.util.SortedSet set3 = digitalActionManager.getNamedBeanSet(); List l = new ArrayList<>(set3); for (MaleSocket x3 : l) { digitalActionManager.deleteBean((MaleDigitalActionSocket)x3, "DoDelete"); } java.util.SortedSet set4 = digitalExpressionManager.getNamedBeanSet(); l = new ArrayList<>(set4); for (MaleSocket x4 : l) { digitalExpressionManager.deleteBean((MaleDigitalExpressionSocket)x4, "DoDelete"); } assertTrue( cm.load(secondFile), "store failed, Failed to load panel"); logixNG_Manager.setupAllLogixNGs(); StringWriter stringWriter2 = new StringWriter(); printWriter = new PrintWriter(stringWriter2); logixNG_Manager.printTree(Locale.ENGLISH, printWriter, treeIndent, new MutableInt(0)); assertEquals( originalTree, stringWriter2.toString(), () -> System.lineSeparator() + "Tree has changed" + System.lineSeparator() + "--------------------------------------------" + System.lineSeparator() + "Old tree:" + System.lineSeparator() + "XXX"+originalTree+"XXX" + System.lineSeparator() + "--------------------------------------------" + System.lineSeparator() + System.lineSeparator() + System.lineSeparator() + "--------------------------------------------" + System.lineSeparator() + "New tree:" + System.lineSeparator() + "XXX"+stringWriter2.toString()+"XXX" + System.lineSeparator() + "--------------------------------------------"); // log.error(conditionalNGManager.getBySystemName(originalTree).getChild(0).getConnectedSocket().getSystemName()); } private void addHeader(File inFile, File outFile) throws FileNotFoundException, IOException { try (BufferedReader reader = new BufferedReader(new InputStreamReader(new FileInputStream(inFile), StandardCharsets.UTF_8)); PrintWriter writer = new PrintWriter(new BufferedWriter(new OutputStreamWriter(new FileOutputStream(outFile), StandardCharsets.UTF_8)))) { String line = reader.readLine(); writer.println(line); writer.println(""); while ((line = reader.readLine()) != null) { writer.println(line); } } } @BeforeEach public void setUp() { JUnitUtil.setUp(); JUnitUtil.resetInstanceManager(); JUnitUtil.resetProfileManager(); JUnitUtil.initConfigureManager(); JUnitUtil.initInternalTurnoutManager(); JUnitUtil.initInternalLightManager(); JUnitUtil.initInternalSensorManager(); JUnitUtil.initInternalSignalHeadManager(); JUnitUtil.initDefaultSignalMastManager(); JUnitUtil.initOBlockManager(); JUnitUtil.initWarrantManager(); // The class under test uses MQTT connections that it pulls from the InstanceManager. memo1 = new MqttSystemConnectionMemo(); // jmri.InstanceManager.store(memo1, jmri.jmrix.mqtt.MqttSystemConnectionMemo.class); // The class under test uses MQTT connections that it pulls from the InstanceManager. // memo2 = new MqttSystemConnectionMemo(); // jmri.InstanceManager.store(memo2, MqttSystemConnectionMemo.class); } @AfterEach public void tearDown() { // JUnitAppender.clearBacklog(); // REMOVE THIS!!! // JUnitUtil.removeMatchingThreads("LnPowerManager LnTrackStatusUpdateThread"); // JUnitUtil.removeMatchingThreads("LnSensorUpdateThread"); // JUnitUtil.removeMatchingThreads("LocoNetThrottledTransmitter"); jmri.jmrit.logixng.util.LogixNG_Thread.stopAllLogixNGThreads(); // JUnitUtil.deregisterBlockManagerShutdownTask(); JUnitUtil.tearDown(); } private static final org.slf4j.Logger log = org.slf4j.LoggerFactory.getLogger(StoreAndLoadTest.class); }