| 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."""
|
|
|