Python test suite updates

1 files changed, 17 insertions, 18 deletions

diff --git a/python/test/Ice/servantLocator/ServerAMD.py b/python/test/Ice/servantLocator/ServerAMD.py
index d7e24ae2816..a8879fafd0a 100755
--- a/python/test/Ice/servantLocator/ServerAMD.py
+++ b/python/test/Ice/servantLocator/ServerAMD.py
@@ -8,26 +8,25 @@
 #
 # **********************************************************************
 
-import os, sys, traceback, time
-
+from TestHelper import TestHelper
+TestHelper.loadSlice('Test.ice')
 import Ice
-Ice.loadSlice('Test.ice')
-import Test, TestAMDI, TestActivationAMDI
+import TestAMDI
+import TestActivationAMDI
 
-class TestServer(Ice.Application):
-    def run(self, args):
-        self.communicator().getProperties().setProperty("TestAdapter.Endpoints", "default -p 12010")
-        self.communicator().getProperties().setProperty("Ice.Warn.Dispatch", "0")
 
-        adapter = self.communicator().createObjectAdapter("TestAdapter")
-        adapter.addServantLocator(TestAMDI.ServantLocatorI("category"), "category")
-        adapter.addServantLocator(TestAMDI.ServantLocatorI(""), "")
-        adapter.add(TestAMDI.TestI(), Ice.stringToIdentity("asm"))
-        adapter.add(TestActivationAMDI.TestActivationAMDI(), Ice.stringToIdentity("test/activation"))
+class ServerAMD(TestHelper):
+
+    def run(self, args):
+        with self.initialize(args=args) as communicator:
+            communicator.getProperties().setProperty("TestAdapter.Endpoints", self.getTestEndpoint())
+            communicator.getProperties().setProperty("Ice.Warn.Dispatch", "0")
 
-        adapter.activate()
-        adapter.waitForDeactivate()
-        return 0
+            adapter = communicator.createObjectAdapter("TestAdapter")
+            adapter.addServantLocator(TestAMDI.ServantLocatorI("category"), "category")
+            adapter.addServantLocator(TestAMDI.ServantLocatorI(""), "")
+            adapter.add(TestAMDI.TestI(), Ice.stringToIdentity("asm"))
+            adapter.add(TestActivationAMDI.TestActivationAMDI(), Ice.stringToIdentity("test/activation"))
 
-app = TestServer()
-sys.exit(app.main(sys.argv))
+            adapter.activate()
+            adapter.waitForDeactivate()