X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/3e3389504c6841e160ac086328693001261e016d..7b2d04380952ac79453bd0771679e40c81281f5c:/sdk/python/tests/test_websockets.py

diff --git a/sdk/python/tests/test_websockets.py b/sdk/python/tests/test_websockets.py
index 6b57fe3c8e..1dae978c84 100644
--- 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()