From 723c3862e22bbfb1e2f9070223e77de02bf01b4f Mon Sep 17 00:00:00 2001
From: Tobias Oberstein <tobias.oberstein@tavendo.de>
Date: Mon, 14 Apr 2014 03:00:43 +0200
Subject: [PATCH] add test

---
 autobahn/Makefile                             |   3 +-
 autobahn/autobahn/wamp/test/test_component.py | 475 ++++++++++++++++++
 examples/twisted/wamp/basic/test.py           |  64 ++-
 3 files changed, 518 insertions(+), 24 deletions(-)
 create mode 100644 autobahn/autobahn/wamp/test/test_component.py

diff --git a/autobahn/Makefile b/autobahn/Makefile
index ff308c6e..13eefb13 100644
--- a/autobahn/Makefile
+++ b/autobahn/Makefile
@@ -25,10 +25,11 @@ publish: clean
 	python setup.py sdist upload
 
 test:
-	trial autobahn
+#	trial autobahn
 #	trial autobahn.websocket.test
 #	trial autobahn.wamp.test
 #	trial autobahn.wamp.test.test_message
+	trial autobahn.wamp.test.test_component
 #	trial autobahn.wamp.test.test_protocol
 #	trial autobahn.wamp.test.test_protocol_peer
 #	trial autobahn.wamp.test.test_serializer
diff --git a/autobahn/autobahn/wamp/test/test_component.py b/autobahn/autobahn/wamp/test/test_component.py
new file mode 100644
index 00000000..bf76d10b
--- /dev/null
+++ b/autobahn/autobahn/wamp/test/test_component.py
@@ -0,0 +1,475 @@
+###############################################################################
+##
+##  Copyright (C) 2014 Tavendo GmbH
+##
+##  Licensed under the Apache License, Version 2.0 (the "License");
+##  you may not use this file except in compliance with the License.
+##  You may obtain a copy of the License at
+##
+##      http://www.apache.org/licenses/LICENSE-2.0
+##
+##  Unless required by applicable law or agreed to in writing, software
+##  distributed under the License is distributed on an "AS IS" BASIS,
+##  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+##  See the License for the specific language governing permissions and
+##  limitations under the License.
+##
+###############################################################################
+
+from __future__ import absolute_import
+
+from twisted.trial import unittest
+#import unittest
+
+from autobahn import wamp
+from autobahn.wamp import message
+from autobahn.wamp import role
+from autobahn.wamp import serializer
+
+
+###############################################################################
+##
+##  Copyright (C) 2011-2014 Tavendo GmbH
+##
+##  Licensed under the Apache License, Version 2.0 (the "License");
+##  you may not use this file except in compliance with the License.
+##  You may obtain a copy of the License at
+##
+##      http://www.apache.org/licenses/LICENSE-2.0
+##
+##  Unless required by applicable law or agreed to in writing, software
+##  distributed under the License is distributed on an "AS IS" BASIS,
+##  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+##  See the License for the specific language governing permissions and
+##  limitations under the License.
+##
+###############################################################################
+
+import sys
+import io
+import six
+import datetime
+
+from twisted.python import log
+from twisted.internet.defer import inlineCallbacks, Deferred, DeferredList
+from twisted.internet.endpoints import serverFromString
+from twisted.internet.endpoints import clientFromString
+
+from autobahn.wamp import router
+from autobahn.twisted.util import sleep
+from autobahn.twisted import wamp, websocket
+
+
+class CaseComponent(wamp.ApplicationSession):
+   """
+   Application code goes here. This is an example component that calls
+   a remote procedure on a WAMP peer, subscribes to a topic to receive
+   events, and then stops the world after some events.
+   """
+
+   def __init__(self, config, done):
+      wamp.ApplicationSession.__init__(self, config)
+      self._done = done
+      self.stop = False
+      self._logline = 1
+      self.finished = False
+
+   def log(self, *args):
+      if len(args) > 1:
+         sargs = ", ".join(str(s) for s in args)
+      elif len(args) == 1:
+         sargs = args[0]
+      else:
+         sargs = "-"
+
+      msg = u'= : {:>3} : {:<20} : {}'.format(self._logline, self.__class__.__name__, sargs)
+      self._logline += 1
+      print(msg)
+      if self.config.log and not self.config.log.closed:
+         self.config.log.write(msg + "\n")
+         self.config.log.flush()
+      else:
+         print("log already closed")
+
+   def finish(self):
+      if not self.finished:
+         self._done.callback(None)
+         self.finished = True
+         self.disconnect()
+      else:
+         print("already finished")
+
+
+
+class Case1_Backend(CaseComponent):
+
+   @inlineCallbacks
+   def onJoin(self, details):
+
+      self.log("joined")
+
+      def add2(x, y):
+         self.log("add2 invoked: {}, {}".format(x, y))
+         return x + y
+
+      yield self.register(add2, 'com.mathservice.add2')
+      self.log("add2 registered")
+
+      self.finish()
+
+
+
+class Case1_Frontend(CaseComponent):
+
+   @inlineCallbacks
+   def onJoin(self, details):
+
+      self.log("joined")
+
+      try:
+         res = yield self.call('com.mathservice.add2', 2, 3)
+      except Exception as e:
+         self.log("call error: {}".format(e))
+      else:
+         self.log("call result: {}".format(res))
+
+      self.finish()
+
+
+
+class Case2_Backend(CaseComponent):
+
+   @inlineCallbacks
+   def onJoin(self, details):
+
+      self.log("joined")
+
+      def ping():
+         self.log("ping() is invoked")
+         return
+
+      def add2(a, b):
+         self.log("add2() is invoked", a, b)
+         return a + b
+
+      def stars(nick = "somebody", stars = 0):
+         self.log("stars() is invoked", nick, stars)
+         return u"{} starred {}x".format(nick, stars)
+
+      def orders(product, limit = 5):
+         self.log("orders() is invoked", product, limit)
+         return [u"Product {}".format(i) for i in range(50)][:limit]
+
+      def arglen(*args, **kwargs):
+         self.log("arglen() is invoked", args, kwargs)
+         return [len(args), len(kwargs)]
+
+      yield self.register(ping, u'com.arguments.ping')
+      yield self.register(add2, u'com.arguments.add2')
+      yield self.register(stars, u'com.arguments.stars')
+      yield self.register(orders, u'com.arguments.orders')
+      yield self.register(arglen, u'com.arguments.arglen')
+
+      self.log("procedures registered")
+
+      yield sleep(3)
+
+      self.log("finishing")
+
+      self.finish()
+
+
+from autobahn.twisted.util import sleep
+
+class Case2_Frontend(CaseComponent):
+
+   @inlineCallbacks
+   def onJoin(self, details):
+
+      self.log("joined")
+
+      yield sleep(1)
+
+      yield self.call(u'com.arguments.ping')
+      self.log("Pinged!")
+
+      res = yield self.call(u'com.arguments.add2', 2, 3)
+      self.log("Add2: {}".format(res))
+
+      starred = yield self.call(u'com.arguments.stars')
+      self.log("Starred 1: {}".format(starred))
+
+      starred = yield self.call(u'com.arguments.stars', nick = u'Homer')
+      self.log("Starred 2: {}".format(starred))
+
+      starred = yield self.call(u'com.arguments.stars', stars = 5)
+      self.log("Starred 3: {}".format(starred))
+
+      starred = yield self.call(u'com.arguments.stars', nick = u'Homer', stars = 5)
+      self.log("Starred 4: {}".format(starred))
+
+      orders = yield self.call(u'com.arguments.orders', u'coffee')
+      self.log("Orders 1: {}".format(orders))
+
+      orders = yield self.call(u'com.arguments.orders', u'coffee', limit = 10)
+      self.log("Orders 2: {}".format(orders))
+
+      arglengths = yield self.call(u'com.arguments.arglen')
+      self.log("Arglen 1: {}".format(arglengths))
+
+      arglengths = yield self.call(u'com.arguments.arglen', 1, 2, 3)
+      self.log("Arglen 1: {}".format(arglengths))
+
+      arglengths = yield self.call(u'com.arguments.arglen', a = 1, b = 2, c = 3)
+      self.log("Arglen 2: {}".format(arglengths))
+
+      arglengths = yield self.call(u'com.arguments.arglen', 1, 2, 3, a = 1, b = 2, c = 3)
+      self.log("Arglen 3: {}".format(arglengths))
+
+      self.log("finishing")
+
+      self.finish()
+
+
+
+class TestRpc(unittest.TestCase):
+
+   def setUp(self):
+      self.debug = False
+      self.realm = "realm1"
+      self.transport = "websocket"
+      self.url = "ws://127.0.0.1:8080"
+      self.client = "tcp:127.0.0.1:8080"
+      self.server = "tcp:8080"
+
+
+   def test_minimal(self):
+
+      embedded_components, client_components = [], [Case2_Backend, Case2_Frontend]
+
+      ## create a WAMP router factory
+      ##
+      from autobahn.wamp.router import RouterFactory
+      router_factory = RouterFactory()
+
+
+      ## create a WAMP router session factory
+      ##
+      from autobahn.twisted.wamp import RouterSessionFactory
+      session_factory = RouterSessionFactory(router_factory)
+
+
+
+      ## .. and create and add an WAMP application session to
+      ## run next to the router
+      ##
+      from autobahn.wamp import types
+
+      config = types.ComponentConfig(realm = self.realm,
+         extra = {
+            'caselog': 'case1.log'
+         }
+      )
+      try:
+         log = io.open('caselog.log', 'w')
+         print "WWW"*10, log
+      except Exception as e:
+         print "X"*10, e
+#      log = io.open(config.extra['caselog'], 'w')
+      config.log = log
+      config.components = []
+
+      config.all_done = []
+
+      for C in embedded_components:
+         one_done = Deferred()
+         config.all_done.append(one_done)
+         c = C(config, one_done)
+         config.components.append(c)
+         session_factory.add(c)
+
+      print "2"*10
+
+
+      if self.transport == "websocket":
+
+         ## create a WAMP-over-WebSocket transport server factory
+         ##
+         from autobahn.twisted.websocket import WampWebSocketServerFactory
+         transport_factory = WampWebSocketServerFactory(session_factory, debug_wamp = self.debug)
+         transport_factory.setProtocolOptions(failByDrop = False, openHandshakeTimeout = 0, closeHandshakeTimeout = 0)
+
+      elif self.transport in ['rawsocket-json', 'rawsocket-msgpack']:
+
+         ## create a WAMP-over-RawSocket transport server factory
+         ##
+         if self.transport == 'rawsocket-msgpack':
+            from autobahn.wamp.serializer import MsgPackSerializer
+            serializer = MsgPackSerializer()
+         elif self.transport == 'rawsocket-json':
+            from autobahn.wamp.serializer import JsonSerializer
+            serializer = JsonSerializer()
+         else:
+            raise Exception("should not arrive here")
+
+         from autobahn.twisted.rawsocket import WampRawSocketServerFactory
+         transport_factory = WampRawSocketServerFactory(session_factory, serializer, debug = self.debug)
+
+      else:
+         raise Exception("should not arrive here")
+
+
+      from autobahn.twisted.websocket import WampWebSocketClientFactory, WampWebSocketClientProtocol
+
+      ## start the server from an endpoint
+      ##
+      from twisted.internet import reactor
+      server = serverFromString(reactor, self.server)
+      d = server.listen(transport_factory)
+
+      def onlisten(port):
+         print "onlisten", port
+         config.port = port
+
+      d.addCallback(onlisten)
+
+
+      print "3"*10
+
+      clients = []
+      clients_d = []
+      for C in client_components:
+         ## create a WAMP application session factory
+         ##
+         from autobahn.twisted.wamp import ApplicationSessionFactory
+         session_factory = ApplicationSessionFactory(config)
+
+         one_done = Deferred()
+         config.all_done.append(one_done)
+
+         def make_make(Klass, done):
+            def make(config):
+               c = Klass(config, done)
+               config.components.append(c)
+               return c
+            return make
+
+         ## .. and set the session class on the factory
+         ##
+         session_factory.session = make_make(C, one_done)
+
+         print "4"*10
+
+         if self.transport == "websocket":
+
+            from autobahn.wamp.serializer import JsonSerializer
+
+            serializers = [JsonSerializer()]
+
+            ## create a WAMP-over-WebSocket transport client factory
+            ##
+            transport_factory = WampWebSocketClientFactory(session_factory, serializers = serializers, url = self.url, debug_wamp = self.debug)
+
+            print "5"*10
+
+            if True:
+               def maker(Klass):
+                  class TestClientProtocol(WampWebSocketClientProtocol):
+                     def onOpen(self):
+                        print "onOpen"
+                        self.txcnt = 0
+                        self.rxcnt = 0
+                        WampWebSocketClientProtocol.onOpen(self)
+
+                     def sendMessage(self, bytes, isBinary):
+                        self.txcnt += 1
+                        print("> : {:>3} : {:<20} : {}".format(self.txcnt, Klass.__name__, bytes))
+                        WampWebSocketClientProtocol.sendMessage(self, bytes, isBinary)
+
+                     def onMessage(self, bytes, isBinary):
+                        self.rxcnt += 1
+                        print("< : {:>3} : {:<20} : {}".format(self.rxcnt, Klass.__name__, bytes))
+                        WampWebSocketClientProtocol.onMessage(self, bytes, isBinary)
+                  return TestClientProtocol
+
+               transport_factory.protocol = maker(C)
+            else:
+               transport_factory.protocol = WampWebSocketClientProtocol
+
+            transport_factory.setProtocolOptions(failByDrop = False, openHandshakeTimeout = 0, closeHandshakeTimeout = 0)
+
+         elif self.transport in ['rawsocket-json', 'rawsocket-msgpack']:
+
+            ## create a WAMP-over-RawSocket transport client factory
+            ##
+            if self.transport == 'rawsocket-msgpack':
+               from autobahn.wamp.serializer import MsgPackSerializer
+               serializer = MsgPackSerializer()
+            elif self.transport == 'rawsocket-json':
+               from autobahn.wamp.serializer import JsonSerializer
+               serializer = JsonSerializer()
+            else:
+               raise Exception("should not arrive here")
+
+            from autobahn.twisted.rawsocket import WampRawSocketClientFactory
+            transport_factory = WampRawSocketClientFactory(session_factory, serializer, debug = self.debug)
+
+
+         ## start the client from an endpoint
+         ##
+         cl = clientFromString(reactor, self.client)
+         clients_d.append(cl.connect(transport_factory))
+
+         print "6"*10
+
+         clients.append(cl)
+
+      print config.all_done
+
+      config.connected_clients = None
+
+      def client_connected(res):
+         config.connected_clients = [proto for success, proto in res if success]
+         print "XXXXXXX", config.connected_clients
+
+      DeferredList(clients_d).addCallback(client_connected)
+
+
+      d = DeferredList(config.all_done, consumeErrors = True)
+      #d = config.components[1]._done
+
+      print "7"
+
+      def done(res):
+         print "DOne"
+         log.flush()
+         log.close()
+         if config.port:
+            config.port.stopListening()
+         if config.connected_clients:
+            for proto in config.connected_clients:
+               proto.transport.abortConnection()
+         #for c in config.components:
+         #   c.leave()
+         #from twisted.internet import reactor
+         #reactor.callLater(1, reactor.stop)
+
+      def error(err):
+         print err
+
+      d.addCallbacks(done, error)
+
+      print "8", d
+
+#      d2 = Deferred()
+
+      return d
+
+
+
+
+
+
+
+if __name__ == '__main__':
+   unittest.main()
diff --git a/examples/twisted/wamp/basic/test.py b/examples/twisted/wamp/basic/test.py
index 69bbdc09..91642d05 100644
--- a/examples/twisted/wamp/basic/test.py
+++ b/examples/twisted/wamp/basic/test.py
@@ -44,8 +44,15 @@ class CaseComponent(wamp.ApplicationSession):
       self._logline = 1
       self.finished = False
 
-   def log(self, msg):
-      msg = u'= : {:>3} : {:<20} : {}'.format(self._logline, self.__class__.__name__, msg)
+   def log(self, *args):
+      if len(args) > 1:
+         sargs = ", ".join(str(s) for s in args)
+      elif len(args) == 1:
+         sargs = args[0]
+      else:
+         sargs = "-"
+
+      msg = u'= : {:>3} : {:<20} : {}'.format(self._logline, self.__class__.__name__, sargs)
       self._logline += 1
       print(msg)
       if self.config.log and not self.config.log.closed:
@@ -108,18 +115,23 @@ class Case2_Backend(CaseComponent):
       self.log("joined")
 
       def ping():
+         self.log("ping() is invoked")
          return
 
       def add2(a, b):
+         self.log("add2() is invoked", a, b)
          return a + b
 
       def stars(nick = "somebody", stars = 0):
+         self.log("stars() is invoked", nick, stars)
          return u"{} starred {}x".format(nick, stars)
 
       def orders(product, limit = 5):
+         self.log("orders() is invoked", product, limit)
          return [u"Product {}".format(i) for i in range(50)][:limit]
 
       def arglen(*args, **kwargs):
+         self.log("arglen() is invoked", args, kwargs)
          return [len(args), len(kwargs)]
 
       yield self.register(ping, u'com.arguments.ping')
@@ -269,8 +281,8 @@ if __name__ == '__main__':
    else:
       embedded_components = []
       #client_components = [Case2_Backend]
-      client_components = [Case1_Backend, Case1_Frontend]
-      #client_components = [Case2_Backend, Case2_Frontend]
+      #client_components = [Case1_Backend, Case1_Frontend]
+      client_components = [Case2_Backend, Case2_Frontend]
 
    log = io.open(config.extra['caselog'], 'w')
    config.log = log
@@ -349,28 +361,34 @@ if __name__ == '__main__':
 
          serializers = [JsonSerializer()]
 
-         def maker(Klass):
-            class TestClientProtocol(WampWebSocketClientProtocol):
-               def onOpen(self):
-                  self.txcnt = 0
-                  self.rxcnt = 0
-                  WampWebSocketClientProtocol.onOpen(self)
-
-               def sendMessage(self, bytes, isBinary):
-                  self.txcnt += 1
-                  print("> : {:>3} : {:<20} : {}".format(self.txcnt, Klass.__name__, bytes))
-                  WampWebSocketClientProtocol.sendMessage(self, bytes, isBinary)
-
-               def onMessage(self, bytes, isBinary):
-                  self.rxcnt += 1
-                  print("< : {:>3} : {:<20} : {}".format(self.rxcnt, Klass.__name__, bytes))
-                  WampWebSocketClientProtocol.onMessage(self, bytes, isBinary)
-            return TestClientProtocol
-
          ## create a WAMP-over-WebSocket transport client factory
          ##
          transport_factory = WampWebSocketClientFactory(session_factory, serializers = serializers, url = args.url, debug_wamp = args.debug)
-         transport_factory.protocol = maker(C)
+
+
+         if False:
+            def maker(Klass):
+               class TestClientProtocol(WampWebSocketClientProtocol):
+                  def onOpen(self):
+                     self.txcnt = 0
+                     self.rxcnt = 0
+                     WampWebSocketClientProtocol.onOpen(self)
+
+                  def sendMessage(self, bytes, isBinary):
+                     self.txcnt += 1
+                     print("> : {:>3} : {:<20} : {}".format(self.txcnt, Klass.__name__, bytes))
+                     WampWebSocketClientProtocol.sendMessage(self, bytes, isBinary)
+
+                  def onMessage(self, bytes, isBinary):
+                     self.rxcnt += 1
+                     print("< : {:>3} : {:<20} : {}".format(self.rxcnt, Klass.__name__, bytes))
+                     WampWebSocketClientProtocol.onMessage(self, bytes, isBinary)
+               return TestClientProtocol
+
+            transport_factory.protocol = maker(C)
+         else:
+            transport_factory.protocol = WampWebSocketClientProtocol
+
          transport_factory.setProtocolOptions(failByDrop = False)
 
       elif args.transport in ['rawsocket-json', 'rawsocket-msgpack']: