#!/usr/bin/env python
# **********************************************************************
#
# Copyright (c) 2003-2017 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, traceback, threading, Ice, Glacier2

Ice.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):
        Glacier2.Application.__init__(self)
        self._restart = 0
        self._destroyed = False
        self._receiver = CallbackReceiverI()

    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:default -p 12010");
        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

try:
    initData = Ice.InitializationData()
    initData.properties = Ice.createProperties(sys.argv)
    initData.properties.setProperty("Ice.Default.Router", "Glacier2/router:default -p 12020")

    app = Application()
    status = app.main(sys.argv, initData=initData)
    test(app._restart == 5)
    test(app._destroyed)
    initData.properties.setProperty("Ice.Default.Router", "");
    with Ice.initialize(initData) as communicator:
        sys.stdout.write("testing stringToProxy for process object... ")
        sys.stdout.flush()
        processBase = communicator.stringToProxy("Glacier2/admin -f Process:default -p 12021");
        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")
except:
    traceback.print_exc()
    status = 1
sys.exit(status)