Index: testing/legion/legion_test_case.py |
diff --git a/testing/legion/legion_test_case.py b/testing/legion/legion_test_case.py |
index 43f2ad1973bb3bca0893e9fb8026c0e6ae980ae6..90072488c7ae96fdb6c9f0a3bff5fd63e7e6ad40 100644 |
--- a/testing/legion/legion_test_case.py |
+++ b/testing/legion/legion_test_case.py |
@@ -14,6 +14,7 @@ import unittest |
from lib import common_lib |
common_lib.SetupEnvironment() |
+from legion.lib import event_server |
from legion.lib import task_controller |
from legion.lib import task_registration_server |
@@ -98,10 +99,17 @@ class TestCase(unittest.TestCase): |
@classmethod |
def _SetUpFramework(cls): |
"""Perform the framework-specific setup operations.""" |
+ # Setup the registration server |
cls._registration_server = ( |
task_registration_server.TaskRegistrationServer()) |
+ common_lib.OnShutdown += cls._registration_server.Shutdown |
cls._registration_server.Start() |
+ # Setup the event server |
+ cls.event_server = event_server.ThreadedServer() |
+ common_lib.OnShutdown += cls.event_server.shutdown |
+ cls.event_server.start() |
+ |
@classmethod |
def _TearDownFramework(cls): |
"""Perform the framework-specific teardown operations.""" |