Basic tests, small fixes

tests/test_DynamoFile.py

@@ -0,0 +1,49 @@
+import os
+import unittest
+import dyn2py
+import pathlib
+
+
+INPUT_DIR = "tests/input_files"
+OUTPUT_DIR = "tests/output_files"
+
+def cleanup():
+    output_dir = pathlib.Path(OUTPUT_DIR)
+    if output_dir.exists():
+        for f in output_dir.iterdir():
+            f.unlink()
+    else:
+        output_dir.mkdir()
+
+class TestDynamoFile(unittest.TestCase):
+
+    def test_read(self):
+        dyn = dyn2py.DynamoFile(f"{INPUT_DIR}/python_nodes.dyn")
+        dyn.read()
+
+        self.assertEqual(dyn.uuid, "3c3b4c05-9716-4e93-9360-ca0637cb5486")
+        self.assertEqual(dyn.name, "python_nodes")
+        self.assertTrue(dyn in dyn.open_files)
+
+    def test_get_python_node(self):
+        dyn = dyn2py.DynamoFile(f"{INPUT_DIR}/python_nodes.dyn")
+        py_nodes = dyn.get_python_nodes()
+        py_node = dyn.get_python_node_by_id("d7704617c75e4bf1a5c387b7c3f001ea")
+
+        self.assertEqual(len(py_nodes), 6)
+        self.assertTrue(py_node)
+        self.assertTrue(py_node in py_nodes)
+        self.assertEqual(py_node.checksum, "1f3d9e6153804fe1ed37571a9cda8e26")
+
+        with self.assertRaises(dyn2py.PythonNodeNotFoundException):
+            dyn.get_python_node_by_id("wrongid")
+
+    def test_extract_python(self):
+        cleanup()
+
+        opt = dyn2py.Options(python_folder="tests/output_files")
+        dyn = dyn2py.DynamoFile(f"{INPUT_DIR}/python_nodes.dyn")
+        dyn.extract_python(options=opt)
+
+        output_dir = pathlib.Path("tests/output_files")
+        self.assertEqual(len(list(output_dir.iterdir())), 6)