From 278adf193d2acb07c9721106d83fd3b8f238380c Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Mon, 17 Aug 2015 23:17:13 +0800
Subject: [PATCH] test/pc_rpc: use builtin_terminate

---
 artiq/test/pc_rpc.py | 18 ++++--------------
 1 file changed, 4 insertions(+), 14 deletions(-)

diff --git a/artiq/test/pc_rpc.py b/artiq/test/pc_rpc.py
index 1b60d245f..5bd0a64cd 100644
--- a/artiq/test/pc_rpc.py
+++ b/artiq/test/pc_rpc.py
@@ -45,7 +45,7 @@ class RPCCase(unittest.TestCase):
             self.assertEqual(test_object, test_object_back)
             with self.assertRaises(pc_rpc.RemoteError):
                 remote.non_existing_method()
-            remote.quit()
+            remote.terminate()
         finally:
             remote.close_rpc()
 
@@ -68,7 +68,7 @@ class RPCCase(unittest.TestCase):
             self.assertEqual(test_object, test_object_back)
             with self.assertRaises(pc_rpc.RemoteError):
                 yield from remote.non_existing_method()
-            yield from remote.quit()
+            yield from remote.terminate()
         finally:
             remote.close_rpc()
 
@@ -97,16 +97,6 @@ class FireAndForgetCase(unittest.TestCase):
 
 
 class Echo:
-    def __init__(self):
-        self.terminate_notify = asyncio.Semaphore(0)
-
-    @asyncio.coroutine
-    def wait_quit(self):
-        yield from self.terminate_notify.acquire()
-
-    def quit(self):
-        self.terminate_notify.release()
-
     def echo(self, x):
         return x
 
@@ -116,10 +106,10 @@ def run_server():
     asyncio.set_event_loop(loop)
     try:
         echo = Echo()
-        server = pc_rpc.Server({"test": echo})
+        server = pc_rpc.Server({"test": echo}, builtin_terminate=True)
         loop.run_until_complete(server.start(test_address, test_port))
         try:
-            loop.run_until_complete(echo.wait_quit())
+            loop.run_until_complete(server.wait_terminate())
         finally:
             loop.run_until_complete(server.stop())
     finally: