summaryrefslogtreecommitdiff
path: root/python/test/Glacier2/application/Client.py
blob: aa8ab6177b3546990d8c7f0f790451c32326227c (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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
#!/usr/bin/env python
# **********************************************************************
#
# Copyright (c) 2003-2018 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 sys
import threading
import Ice
import Glacier2

from TestHelper import TestHelper
TestHelper.loadSlice("Callback.ice")
import Test


def test(b):
    if not b:
        raise RuntimeError('test assertion failed')


class CallbackReceiverI(Test.CallbackReceiver):

    def __init__(self):
        self._received = False
        self._cond = threading.Condition()

    def callback(self, current):
        with self._cond:
            self._received = True
            self._cond.notify()

    def waitForCallback(self):
        with self._cond:
            while not self._received:
                self._cond.wait()
            self._received = False;


class Application(Glacier2.Application):

    def __init__(self, helper):
        Glacier2.Application.__init__(self)
        self._restart = 0
        self._destroyed = False
        self._receiver = CallbackReceiverI()
        self._helper = helper

    def createSession(self):
        return Glacier2.SessionPrx.uncheckedCast(self.router().createSession("userid", "abc123"))

    def runWithSession(self, args):

        test(self.router());
        test(self.categoryForClient());
        test(self.objectAdapter());

        if self._restart == 0:
            sys.stdout.write("testing Glacier2::Application restart... ")
            sys.stdout.flush()

        base = self.communicator().stringToProxy("callback:{0}".format(
            self._helper.getTestEndpoint(properties=self.communicator().getProperties())));
        callback = Test.CallbackPrx.uncheckedCast(base)
        self._restart += 1
        if self._restart < 5:
            receiver = Test.CallbackReceiverPrx.uncheckedCast(self.addWithUUID(self._receiver))
            callback.initiateCallback(receiver)
            self._receiver.waitForCallback()
            self.restart()

        print("ok")

        sys.stdout.write("testing server shutdown... ")
        callback.shutdown()
        print("ok")

        return 0

    def sessionDestroyed(self):
        self._destroyed = True


class Client(TestHelper):

    def run(self, args):
        initData = Ice.InitializationData()
        initData.properties = self.createTestProperties(args)
        initData.properties.setProperty("Ice.Default.Router", "Glacier2/router:{0}".format(
            self.getTestEndpoint(properties=initData.properties, num=50)))

        app = Application(self)
        status = app.main(sys.argv, initData=initData)
        test(status == 0)
        test(app._restart == 5)
        test(app._destroyed)

        initData.properties.setProperty("Ice.Default.Router", "")
        with self.initialize(initData=initData) as communicator:
            sys.stdout.write("testing stringToProxy for process object... ")
            sys.stdout.flush()
            processBase = communicator.stringToProxy("Glacier2/admin -f Process:{0}".format(
                self.getTestEndpoint(properties=initData.properties, num=51)))
            print("ok")

            sys.stdout.write("testing checked cast for admin object... ")
            sys.stdout.flush()
            process = Ice.ProcessPrx.checkedCast(processBase)
            test(process)
            print("ok")

            sys.stdout.write("testing Glacier2 shutdown... ")
            sys.stdout.flush()
            process.shutdown()
            try:
                process.ice_ping()
                test(False)
            except Ice.LocalException:
                print("ok")