2800: Introduce TestCaseWithServers to Python SDK.

[arvados.git] / sdk / python / tests / test_websockets.py

diff --git a/sdk/python/tests/test_websockets.py b/sdk/python/tests/test_websockets.py
index 6b57fe3c8ecdf8fb64c0618c24c49ff201785b5f..1dae978c843b052a19469c1e8dfc0ee876a0ede9 100644 (file)
--- a/sdk/python/tests/test_websockets.py
+++ b/sdk/python/tests/test_websockets.py
@@ -4,9 +4,8 @@ import arvados
 import arvados.events
 import time
 
-class WebsocketTest(unittest.TestCase):
-    def setUp(self):
-        run_test_server.run(websockets=True)
+class WebsocketTest(run_test_server.TestCaseWithServers):
+    MAIN_SERVER = {'websockets': True}
 
     def on_event(self, ev):
         if self.state == 1:
@@ -27,6 +26,3 @@ class WebsocketTest(unittest.TestCase):
         time.sleep(1)
         self.h = api.humans().create(body={}).execute()
         time.sleep(1)
-
-    def tearDown(self):
-        run_test_server.stop()