From c2457f7cce90516a9d83cd0aeb1f5811daab3a91 Mon Sep 17 00:00:00 2001
From: Tyler Goodlet <tgoodlet@gmail.com>
Date: Sun, 2 Sep 2018 22:07:32 -0400
Subject: [PATCH] Move basic spawning tests to separate module

---
 tests/{test_tractor.py => test_local.py} |  0
 tests/test_spawning.py                   | 96 ++++++++++++++++++++++++
 2 files changed, 96 insertions(+)
 rename tests/{test_tractor.py => test_local.py} (100%)
 create mode 100644 tests/test_spawning.py

diff --git a/tests/test_tractor.py b/tests/test_local.py
similarity index 100%
rename from tests/test_tractor.py
rename to tests/test_local.py
diff --git a/tests/test_spawning.py b/tests/test_spawning.py
new file mode 100644
index 0000000..07dd737
--- /dev/null
+++ b/tests/test_spawning.py
@@ -0,0 +1,96 @@
+"""
+Spawning basics
+"""
+import trio
+import tractor
+
+from conftest import tractor_test
+
+statespace = {'doggy': 10, 'kitty': 4}
+
+
+async def spawn(is_arbiter):
+    namespaces = [__name__]
+
+    await trio.sleep(0.1)
+    actor = tractor.current_actor()
+    assert actor.is_arbiter == is_arbiter
+    assert actor.statespace == statespace
+
+    if actor.is_arbiter:
+        async with tractor.open_nursery() as nursery:
+            # forks here
+            portal = await nursery.run_in_actor(
+                'sub-actor',
+                spawn,
+                is_arbiter=False,
+                statespace=statespace,
+                rpc_module_paths=namespaces,
+            )
+
+            assert len(nursery._children) == 1
+            assert portal.channel.uid in tractor.current_actor()._peers
+            # be sure we can still get the result
+            result = await portal.result()
+            assert result == 10
+            return result
+    else:
+        return 10
+
+
+def test_local_arbiter_subactor_global_state(arb_addr):
+    result = tractor.run(
+        spawn,
+        True,
+        name='arbiter',
+        statespace=statespace,
+        arbiter_addr=arb_addr,
+    )
+    assert result == 10
+
+
+def movie_theatre_question():
+    """A question asked in a dark theatre, in a tangent
+    (errr, I mean different) process.
+    """
+    return 'have you ever seen a portal?'
+
+
+@tractor_test
+async def test_movie_theatre_convo():
+    """The main ``tractor`` routine.
+    """
+    async with tractor.open_nursery() as n:
+
+        portal = await n.start_actor(
+            'frank',
+            # enable the actor to run funcs from this current module
+            rpc_module_paths=[__name__],
+        )
+
+        print(await portal.run(__name__, 'movie_theatre_question'))
+        # call the subactor a 2nd time
+        print(await portal.run(__name__, 'movie_theatre_question'))
+
+        # the async with will block here indefinitely waiting
+        # for our actor "frank" to complete, we cancel 'frank'
+        # to avoid blocking indefinitely
+        await portal.cancel_actor()
+
+
+def cellar_door():
+    return "Dang that's beautiful"
+
+
+@tractor_test
+async def test_most_beautiful_word():
+    """The main ``tractor`` routine.
+    """
+    async with tractor.open_nursery() as n:
+
+        portal = await n.run_in_actor('some_linguist', cellar_door)
+
+    # The ``async with`` will unblock here since the 'some_linguist'
+    # actor has completed its main task ``cellar_door``.
+
+    print(await portal.result())