Test RealtimeProvider for parallel group suppoprt

gitmarek · Oct 30, 2022 · b4a97f9 · b4a97f9
1 parent 25ca6df
commit b4a97f9
Show file tree

Hide file tree

Showing 2 changed files with 69 additions and 2 deletions.
diff --git a/supriya/providers.py b/supriya/providers.py
@@ -766,13 +766,13 @@ def add_group(
         target_node=None,
         add_action=AddAction.ADD_TO_HEAD,
         name: Optional[str] = None,
-        parallel: bool = False
+        parallel: bool = False,
     ) -> GroupProxy:
         if not self.moment:
             raise ValueError("No current moment")
         # TODO: implement (dummy) parallel flag for nonrealtime methods
         identifier = self._resolve_target_node(target_node).add_group(
-            add_action=add_action,
+            add_action=add_action
         )
         proxy = GroupProxy(identifier=identifier, provider=self, parallel=parallel)
         return proxy

diff --git a/tests/providers/test_RealtimeProvider.py b/tests/providers/test_RealtimeProvider.py
@@ -30,6 +30,13 @@ def server(persistent_server):
     yield persistent_server
 
 
+@pytest.fixture
+def server_supernova(persistent_server_supernova):
+    persistent_server_supernova.reset()
+    persistent_server_supernova.add_synthdef(default)
+    yield persistent_server_supernova
+
+
 def test_RealtimeProvider_init_error():
     with pytest.raises(ValueError):
         RealtimeProvider(23)
@@ -138,6 +145,66 @@ def test_RealtimeProvider_add_group_1(server):
     )
 
 
+def test_RealtimeProvider_add_group_1_parallel(server):
+    provider = Provider.from_context(server)
+    seconds = time.time()
+    with server.osc_protocol.capture() as transcript:
+        with provider.at(seconds) as provider_moment:
+            group_proxy = provider.add_group(parallel=True)
+    assert group_proxy == GroupProxy(identifier=1000, provider=provider, parallel=True)
+    assert provider_moment == ProviderMoment(
+        provider=provider,
+        seconds=seconds,
+        bus_settings=[],
+        node_additions=[(group_proxy, AddAction.ADD_TO_HEAD, server.default_group)],
+        node_removals=[],
+        node_reorderings=[],
+        node_settings=[],
+    )
+    assert [entry.message.to_list() for entry in transcript] == [
+        [seconds + provider.latency, [["/p_new", 1000, 0, 1]]]
+    ]
+    time.sleep(0.1)
+    assert str(server.query()) == normalize(
+        """
+        NODE TREE 0 group
+            1 group
+                1000 group
+    """
+    )
+
+
+def test_RealtimeProvider_add_group_1_parallel_supernova(server_supernova):
+    provider = Provider.from_context(server_supernova)
+    seconds = time.time()
+    with server_supernova.osc_protocol.capture() as transcript:
+        with provider.at(seconds) as provider_moment:
+            group_proxy = provider.add_group(parallel=True)
+    assert group_proxy == GroupProxy(identifier=1000, provider=provider, parallel=True)
+    assert provider_moment == ProviderMoment(
+        provider=provider,
+        seconds=seconds,
+        bus_settings=[],
+        node_additions=[
+            (group_proxy, AddAction.ADD_TO_HEAD, server_supernova.default_group)
+        ],
+        node_removals=[],
+        node_reorderings=[],
+        node_settings=[],
+    )
+    assert [entry.message.to_list() for entry in transcript] == [
+        [seconds + provider.latency, [["/p_new", 1000, 0, 1]]]
+    ]
+    time.sleep(0.1)
+    assert str(server_supernova.query()) == normalize(
+        """
+        NODE TREE 0 group
+            1 group
+                1000 group
+    """
+    )
+
+
 def test_RealtimeProvider_add_group_2(server):
     provider = Provider.from_context(server)
     seconds = time.time()