mirror of
https://github.com/infeeeee/dyn2py
synced 2025-12-16 22:16:18 +01:00
Move read() methods to __init__(), new tests
This commit is contained in:
@@ -9,48 +9,42 @@ class TestDynamoFile(unittest.TestCase):
|
||||
|
||||
# Missing methods:
|
||||
# get_related_python_files
|
||||
# get_open_file_by_uuid
|
||||
# update_python_node
|
||||
# write
|
||||
|
||||
def test_read_and_variables(self):
|
||||
def test_init(self):
|
||||
dyn2py.DynamoFile.open_files.clear()
|
||||
|
||||
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)
|
||||
|
||||
# Dynamo 1 file:
|
||||
with self.assertRaises(dyn2py.DynamoFileException):
|
||||
dyn1 = dyn2py.DynamoFile(f"{INPUT_DIR}/dynamo1file.dyn")
|
||||
dyn1.read()
|
||||
|
||||
# Not existing file:
|
||||
with self.assertRaises(FileNotFoundError):
|
||||
dyn2 = dyn2py.DynamoFile(f"{INPUT_DIR}/not_existing.dyn")
|
||||
dyn2.read()
|
||||
|
||||
|
||||
# No python nodes:
|
||||
with self.assertRaises(dyn2py.PythonNodeNotFoundException):
|
||||
dyn2 = dyn2py.DynamoFile(f"{INPUT_DIR}/no_python.dyn")
|
||||
|
||||
def test_get_python_nodes(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.assertEqual(len(dyn.python_nodes), 6)
|
||||
self.assertTrue(py_node)
|
||||
self.assertTrue(py_node in py_nodes)
|
||||
self.assertTrue(py_node in dyn.python_nodes)
|
||||
self.assertEqual(py_node.checksum, "1f3d9e6153804fe1ed37571a9cda8e26")
|
||||
|
||||
with self.assertRaises(dyn2py.PythonNodeNotFoundException):
|
||||
dyn.get_python_node_by_id("wrongid")
|
||||
|
||||
dyn2 = dyn2py.DynamoFile(f"{INPUT_DIR}/no_python.dyn")
|
||||
|
||||
# Raise error on file without python nodes:
|
||||
with self.assertRaises(dyn2py.DynamoFileException):
|
||||
dyn2.get_python_nodes()
|
||||
|
||||
|
||||
def test_extract_python(self):
|
||||
cleanup_output_dir()
|
||||
|
||||
@@ -60,3 +54,15 @@ class TestDynamoFile(unittest.TestCase):
|
||||
|
||||
output_dir = pathlib.Path(OUTPUT_DIR)
|
||||
self.assertEqual(len(list(output_dir.iterdir())), 6)
|
||||
|
||||
def test_get_open_file_by_uuid(self):
|
||||
dyn2py.DynamoFile.open_files.clear()
|
||||
|
||||
dyn1 = dyn2py.DynamoFile(f"{INPUT_DIR}/python_nodes.dyn")
|
||||
dyn2 = dyn2py.DynamoFile(f"{INPUT_DIR}/single_node.dyn")
|
||||
|
||||
|
||||
self.assertEqual(dyn1,
|
||||
dyn2py.DynamoFile.get_open_file_by_uuid("3c3b4c05-9716-4e93-9360-ca0637cb5486"))
|
||||
self.assertEqual(dyn2,
|
||||
dyn2py.DynamoFile.get_open_file_by_uuid("76de5c79-17c5-4c74-9f90-ad99a213d339"))
|
||||
|
||||
Reference in New Issue
Block a user