1 files changed, 155 insertions, 0 deletions
diff --git a/cpp/test/IceStorm/stress/test.py b/cpp/test/IceStorm/stress/test.py
new file mode 100644
index 00000000000..16f30f28a30
--- /dev/null
+++ b/cpp/test/IceStorm/stress/test.py
@@ -0,0 +1,155 @@
+# -*- coding: utf-8 -*-
+# **********************************************************************
+#
+# Copyright (c) 2003-2016 ZeroC, Inc. All rights reserved.
+#
+# This copy of Ice is licensed to you under the terms described in the
+# ICE_LICENSE file included in this distribution.
+#
+# **********************************************************************
+
+#
+# Publisher/subscriber test cases, publisher publishes on TestIceStorm1 instance(s) and
+# the subscriber subscribes to the TestIceStorm2 instance(s)
+#
+pubSub=lambda si, pi, s={}, p={}: ClientServerTestCase(client=Publisher(args=s), server=Subscriber(args=p))
+
+class IceStormStressTestCase(IceStormTestCase):
+
+    def runClientSide(self, current):
+
+        icestorm1 = [icestorm for icestorm in self.icestorm if icestorm.getInstanceName() == "TestIceStorm1"]
+        icestorm2 = [icestorm for icestorm in self.icestorm if icestorm.getInstanceName() == "TestIceStorm2"]
+
+        def doTest(subOpts, pubOpts):
+            # Create the subscribers
+            subscribers = []
+            for (instanceName, opts) in subOpts if type(subOpts) == list else [subOpts]:
+                subscribers.append(Subscriber(instanceName or "TestIceStorm1", args=opts.split(" ")))
+
+            # Create the publisher
+            publisher = Publisher("TestIceStorm1", args=pubOpts.split(" "))
+
+            # Run the test case
+            ClientServerTestCase(client=publisher, servers=subscribers).run(current)
+
+        current.write("setting up the topics...")
+        self.runadmin(current, "create TestIceStorm1/fed1 TestIceStorm2/fed1")
+        current.writeln("ok")
+
+        current.write("Sending 5000 ordered events... ")
+        doTest(("TestIceStorm1", '--events 5000 --qos "reliability,ordered"'), '--events 5000')
+        current.writeln("ok")
+
+        self.runadmin(current, "link TestIceStorm1/fed1 TestIceStorm2/fed1")
+        current.write("Sending 5000 ordered events across a link... ")
+        doTest(("TestIceStorm2", '--events 5000 --qos "reliability,ordered"'), '--events 5000')
+        current.writeln("ok")
+
+        self.runadmin(current, "unlink TestIceStorm1/fed1 TestIceStorm2/fed1")
+        current.write("Sending 20000 unordered events... ")
+        doTest(("TestIceStorm1", '--events 20000'), '--events 20000 --oneway')
+        current.writeln("ok")
+
+        self.runadmin(current, "link TestIceStorm1/fed1 TestIceStorm2/fed1")
+        current.write("Sending 20000 unordered events across a link... ")
+        doTest(("TestIceStorm2", '--events 20000'), '--events 20000 --oneway')
+        current.writeln("ok")
+
+        self.runadmin(current, "unlink TestIceStorm1/fed1 TestIceStorm2/fed1")
+        current.write("Sending 20000 unordered batch events... ")
+        doTest(("TestIceStorm1", '--events 20000 --qos "reliability,batch"'), '--events 20000 --oneway')
+        current.writeln("ok")
+
+        self.runadmin(current, "link TestIceStorm1/fed1 TestIceStorm2/fed1")
+        current.write("Sending 20000 unordered batch events across a link... ")
+        doTest(("TestIceStorm2", '--events 20000 --qos "reliability,batch"'), '--events 20000 --oneway')
+        current.writeln("ok")
+
+        self.runadmin(current, "unlink TestIceStorm1/fed1 TestIceStorm2/fed1")
+        current.write("Sending 20000 unordered events with slow subscriber... ")
+        doTest([("TestIceStorm1", '--events 2 --slow'),
+                ("TestIceStorm1", '--events 20000 ')],
+                '--events 20000 --oneway')
+        current.writeln("ok")
+
+        self.runadmin(current, "link TestIceStorm1/fed1 TestIceStorm2/fed1")
+        current.write("Sending 20000 unordered events with slow subscriber & link... ")
+        doTest([("TestIceStorm1", '--events 2 --slow'),
+                ("TestIceStorm1", '--events 20000'),
+                ("TestIceStorm2", '--events 2 --slow'),
+                ("TestIceStorm2", '--events 20000')],
+                '--events 20000 --oneway')
+        current.writeln("ok")
+
+        current.write("shutting down icestorm services... ")
+        self.stopIceStorm(current)
+        current.writeln("ok")
+
+        current.write("starting icestorm services... ")
+        #
+        # The erratic tests emit lots of connection warnings so they are
+        # disabled here. The IceStorm servers are stopped and restarted so the
+        # settings will take effect.
+        #
+        self.startIceStorm(current)
+        current.writeln("ok")
+
+        self.runadmin(current, "unlink TestIceStorm1/fed1 TestIceStorm2/fed1")
+        current.write("Sending 20000 unordered events with erratic subscriber... ")
+        doTest([ ("TestIceStorm1", '--erratic 5 --qos "reliability,ordered" --events 20000'),
+                 ("TestIceStorm1", '--erratic 5 --events 20000'),
+                 ("TestIceStorm1", '--events 20000')],
+                 '--events 20000 --oneway')
+        current.writeln("ok")
+
+        self.runadmin(current, "link TestIceStorm1/fed1 TestIceStorm2/fed1")
+        current.write("Sending 20000 unordered events with erratic subscriber across a link... ")
+        doTest([ ("TestIceStorm1", '--events 20000'),
+                 ("TestIceStorm1", '--erratic 5 --qos "reliability,ordered" --events 20000 '),
+                 ("TestIceStorm1", '--erratic 5 --events 20000 '),
+                 ("TestIceStorm2", '--events 20000'),
+                 ("TestIceStorm2", '--erratic 5 --qos "reliability,ordered" --events 20000 '),
+                 ("TestIceStorm2", '--erratic 5 --events 20000 ')],
+                 '--events 20000 --oneway ')
+        current.writeln("ok")
+
+        #
+        # Shutdown icestorm.
+        #
+        current.write("shutting down icestorm services... ")
+        self.stopIceStorm(current)
+        current.writeln("ok")
+
+        current.write("Sending 5000 ordered events with max queue size drop events... ")
+        opts = " --IceStorm.Send.QueueSizeMax=2000 --IceStorm.Send.QueueSizeMaxPolicy=DropEvents"
+        for s in icestorm1:
+            s.start(current, args=opts.split(" "))
+        doTest(("TestIceStorm1", '--events 5000 --qos "reliability,ordered" --maxQueueDropEvents=2000'),
+               '--events 5000 --maxQueueTest')
+        for s in icestorm1:
+            s.shutdown(current)
+            s.stop(current, True)
+        current.writeln("ok")
+
+        current.write("Sending 5000 ordered events with max queue size remove subscriber... ")
+        opts = " --IceStorm.Send.QueueSizeMax=2000 --IceStorm.Send.QueueSizeMaxPolicy=RemoveSubscriber"
+        for s in icestorm1:
+            s.start(current, args=opts.split(" "))
+        doTest(("TestIceStorm1", '--events 5000 --qos "reliability,ordered" --maxQueueRemoveSub=2000'),
+               '--events 5000 --maxQueueTest')
+        for s in icestorm1:
+            s.shutdown(current)
+            s.stop(current, True)
+        current.writeln("ok")
+
+TestSuite(__file__, [
+
+    IceStormStressTestCase("persistent", icestorm=[IceStorm("TestIceStorm1", quiet=True),
+                                                   IceStorm("TestIceStorm2", quiet=True, portnum=20)]),
+
+    IceStormStressTestCase("replicated with replicated publisher", icestorm=
+                           [IceStorm("TestIceStorm1", i, 3, quiet=True) for i in range(0,3)] +
+                           [IceStorm("TestIceStorm2", i, 3, portnum=20, quiet=True) for i in range(0,3)]),
+
+], multihost=False)