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..aa700440fa494aeb9b8a2f88d267c2705168ccb4 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.""" |
@@ -140,4 +148,4 @@ class TestCase(unittest.TestCase): |
def main(): |
- unittest.main(verbosity=0, argv=sys.argv[:1]) |
+ unittest.main(argv=sys.argv[:1]) |