summaryrefslogtreecommitdiff
path: root/cpp/test/IceStorm/federation/run.py
blob: 0cf7f0f5ae92a18e5a5fa88b9d9871e39dbd5c7e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
#!/usr/bin/env python
# **********************************************************************
#
# Copyright (c) 2003-2007 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.
#
# **********************************************************************

import os, sys
import time

for toplevel in [".", "..", "../..", "../../..", "../../../.."]:
    toplevel = os.path.normpath(toplevel)
    if os.path.exists(os.path.join(toplevel, "config", "TestUtil.py")):
        break
else:
    raise "can't find toplevel directory!"

sys.path.append(os.path.join(toplevel, "config"))
import TestUtil
TestUtil.processCmdLine()

name = os.path.join("IceStorm", "federation")
testdir = os.path.dirname(os.path.abspath(__file__))

import IceStormUtil

def doTest(icestorm, batch):

    publisher = os.path.join(testdir, "publisher")
    subscriber = os.path.join(testdir, "subscriber")

    if batch:
        name = "batch subscriber"
        batchOptions = " -b"
    else:
        name = "subscriber"
        batchOptions = ""

    subscriberPipe = TestUtil.startServer(subscriber, batchOptions + icestorm.reference() + " 2>&1")
    TestUtil.getServerPid(subscriberPipe)
    TestUtil.getAdapterReady(subscriberPipe)

    #
    # Start the publisher. This should publish events which eventually
    # causes subscriber to terminate.
    #
    publisherPipe = TestUtil.startClient(publisher, icestorm.reference() + " 2>&1")

    TestUtil.printOutputFromPipe(publisherPipe)

    subscriberStatus = TestUtil.specificServerStatus(subscriberPipe, 30)
    publisherStatus = TestUtil.closePipe(publisherPipe)

    return subscriberStatus or publisherStatus

def runtest(type, **args):
    icestorm = IceStormUtil.init(toplevel, testdir, type, **args)

    icestorm.start()

    print "setting up topics...",
    sys.stdout.flush()
    icestorm.admin("create fed1 fed2 fed3; link fed1 fed2 10; link fed2 fed3 5")
    print "ok"

    #
    # Test oneway subscribers.
    #
    print "testing oneway subscribers...",
    sys.stdout.flush()
    onewayStatus = doTest(icestorm, 0)
    print "ok"

    #
    # Test batch oneway subscribers.
    #
    print "testing batch subscribers...",
    sys.stdout.flush()
    batchStatus = doTest(icestorm, 1)
    print "ok"

    #
    # Destroy the topics.
    #
    print "destroying topics...",
    icestorm.admin("destroy fed1 fed2 fed3")
    print "ok"

    #
    # Shutdown icestorm.
    #
    icestorm.stop()

    if TestUtil.serverStatus() or onewayStatus or batchStatus:
	TestUtil.killServers()
	sys.exit(1)

runtest("persistent")
runtest("transient")
runtest("replicated", replicatedPublisher = False)
runtest("replicated", replicatedPublisher = True)

sys.exit(0)