Fixed ICE-5486: added support for MessageSizeMax OA property

1 files changed, 4 insertions, 12 deletions

diff --git a/py/test/Ice/exceptions/AllTests.py b/py/test/Ice/exceptions/AllTests.py
index e49b2dfcc9c..1baaa87751c 100644
--- a/py/test/Ice/exceptions/AllTests.py
+++ b/py/test/Ice/exceptions/AllTests.py
@@ -473,13 +473,14 @@ def allTests(communicator):
         print("ok")
 
 
+    if thrower.ice_getConnection():
         sys.stdout.write("testing memory limit marshal exception...");
         sys.stdout.flush();
 
         try:
             thrower.throwMemoryLimitException(array.array('B'));
             test(False)
-        except Ice.UnknownLocalException:
+        except Ice.MemoryLimitException:
             pass
         except:
             print(sys.exc_info())
@@ -488,21 +489,12 @@ def allTests(communicator):
         try:
             thrower.throwMemoryLimitException(bytearray(20 * 1024)) # 20KB
             test(False)
-        except Ice.MemoryLimitException:
-            pass
-        except:
-            test(False)
-
-        try:
-            thrower.end_throwMemoryLimitException(thrower.begin_throwMemoryLimitException(bytearray(20 * 1024))) # 20KB
-            test(False)
-        except Ice.MemoryLimitException:
+        except Ice.ConnectionLostException:
             pass
         except:
             test(False)
 
-    print("ok");
-
+        print("ok");
 
     sys.stdout.write("catching object not exist exception... ")
     sys.stdout.flush()