Added test to create process group inside another process group. (#101)

Chaffelson · Jan 31, 2019 · 9a48503 · 9a48503
1 parent 7c59c02
commit 9a48503
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -16,6 +16,7 @@
 test_host = 'localhost'
 test_basename = "nipyapi_test"
 test_pg_name = test_basename + "_ProcessGroup"
+test_another_pg_name = test_basename + "_AnotherProcessGroup"
 test_registry_client_name = test_basename + "_reg_client"
 test_processor_name = test_basename + "_proc"
 test_bucket_name = test_basename + "_bucket"

diff --git a/tests/test_canvas.py b/tests/test_canvas.py
@@ -64,6 +64,16 @@ def test_create_process_group(regress_nifi):
     assert r.position.x == r.position.y == 400
     assert r.component.parent_group_id == canvas.get_root_pg_id()
     assert isinstance(r, nifi.ProcessGroupEntity)
+
+    # Test process group creation on other than root process group.
+    s = canvas.create_process_group(parent_pg=canvas.get_process_group(conftest.test_pg_name), location=(200.0, 200.0),
+                                    new_pg_name=conftest.test_another_pg_name)
+    assert s.component.name == conftest.test_another_pg_name
+    assert s.position.x == s.position.y == 200
+    assert s.component.parent_group_id == canvas.get_process_group(conftest.test_pg_name, "name").id
+    assert isinstance(s, nifi.ProcessGroupEntity)
+
+
     with pytest.raises(ApiException):
         parent_pg = canvas.get_process_group('NiFi Flow')
         parent_pg.id = 'invalid'