[tests] Add test for InitNode

Author: cbentejac

tests/plugins/meshroom/pluginA/PluginAInputInitNode.py

@@ -0,0 +1,27 @@
+__version__ = "1.0"
+
+from meshroom.core import desc
+
+
+class PluginAInputInitNode(desc.InputNode, desc.InitNode):
+    inputs = [
+        desc.File(
+            name="input",
+            label="Input",
+            description="",
+            value="",
+        ),
+    ]
+
+    outputs = [
+        desc.File(
+            name="output",
+            label="Output",
+            description="",
+            value="",
+        ),
+    ]
+
+    def initialize(self, node, inputs, recursiveInputs):
+        if len(inputs) >= 1:
+            self.setAttributes(node, {"input": inputs[0]})

tests/test_nodes.py

@@ -101,3 +101,38 @@ def test_expVariables(self):
             assert n._expVars["uid"] == n._uid
             assert n.internalFolder
             assert n.internalFolder == n._expVars["nodeCacheFolder"]
+
+
+class TestInitNode:
+    plugin = None
+
+    @classmethod
+    def setup_class(cls):
+        folder = os.path.join(os.path.dirname(__file__), "plugins", "meshroom")
+        package = "pluginA"
+        cls.plugin = Plugin(package, folder)
+        nodes = loadClassesNodes(folder, package)
+        for node in nodes:
+            cls.plugin.addNodePlugin(node)
+        pluginManager.addPlugin(cls.plugin)
+
+    @classmethod
+    def teardown_class(cls):
+        for node in cls.plugin.nodes.values():
+            pluginManager.unregisterNode(node)
+        pluginManager.removePlugin(cls.plugin)
+        cls.plugin = None
+
+    def test_initNode(self):
+        g = Graph("")
+
+        node = g.addNewNode("PluginAInputInitNode")
+
+        # Check that the init node is correctly detected
+        initNodes = g.findInitNodes()
+        assert len(initNodes) == 1 and node in initNodes
+
+        # Check that the init node's initialize method has been set
+        inputs = ["/path/to/file", "/path/to/file/2"]
+        node.nodeDesc.initialize(node, inputs, None)
+        assert node.input.value == inputs[0]