Changeset 35571fb for protocols/skype/skyped.py

Timestamp:

2013-01-13T12:06:01Z (11 years ago)

Author:

Wilmer van der Gaast <wilmer@…>

Branches:

master

Children:

75bda8b

Parents:

4cb21b7 (diff), 650d2b4 (diff)
Note: this is a merge changeset, the changes displayed below correspond to the merge itself.
Use the (diff) links above to see all the changes relative to each parent.

Message:

Merge from vmiklos.

File:

• protocols/skype/skyped.py (modified) (7 diffs)

protocols/skype/skyped.py

@@ -3 +3 @@
 #   skyped.py
 #  
-#   Copyright (c) 2007, 2008, 2009, 2010, 2011 by Miklos Vajna <vmiklos@frugalware.org>
+#   Copyright (c) 2007-2013 by Miklos Vajna <vmiklos@vmiklos.hu>
 #
 #   This program is free software; you can redistribute it and/or modify
@@ -276 +276 @@
                 sock.close()
 
+class MockedSkype:
+        """Mock class for Skype4Py.Skype(), in case the -m option is used."""
+        def __init__(self, mock):
+                sock = open(mock)
+                self.lines = sock.readlines()
+        
+        def SendCommand(self, c):
+                pass
+
+        def Command(self, msg, Block):
+                if msg == "PING":
+                        return ["PONG"]
+                line = self.lines[0].strip()
+                if not line.startswith(">> "):
+                        raise Exception("Corrupted mock input")
+                line = line[3:]
+                if line != msg:
+                        raise Exception("'%s' != '%s'" % (line, msg))
+                self.lines = self.lines[1:] # drop the expected incoming line
+                ret = []
+                while True:
+                        # and now send back all the following lines, up to the next expected incoming line
+                        if len(self.lines) == 0:
+                                break
+                        if self.lines[0].startswith(">> "):
+                                break
+                        if not self.lines[0].startswith("<< "):
+                                raise Exception("Corrupted mock input")
+                        ret.append(self.lines[0][3:].strip())
+                        self.lines = self.lines[1:]
+                return ret
+
 class SkypeApi:
-        def __init__(self):
-                self.skype = Skype4Py.Skype()
-                self.skype.OnNotify = self.recv
-                self.skype.Client.Start()
+        def __init__(self, mock):
+                if not mock:
+                        self.skype = Skype4Py.Skype()
+                        self.skype.OnNotify = self.recv
+                        self.skype.Client.Start()
+                else:
+                        self.skype = MockedSkype(mock)
 
         def recv(self, msg_text):
@@ -335 +370 @@
                         c = self.skype.Command(e, Block=True)
                         self.skype.SendCommand(c)
-                        self.recv(c.Reply)
+                        if hasattr(c, "Reply"):
+                                self.recv(c.Reply) # Skype4Py answer
+                        else:
+                                for i in c: # mock may return multiple iterable answers
+                                        self.recv(i)
                 except Skype4Py.SkypeError:
                         pass
@@ -360 +399 @@
                 # this will be read first by the input handler
                 self.buf = None
+                self.mock = None
 
 
@@ -422 +462 @@
         options = Options()
         try:
-                opts, args = getopt.getopt(sys.argv[1:], "c:dhH:l:np:v", ["config=", "debug", "help", "host=", "log=", "nofork", "port=", "version"])
+                opts, args = getopt.getopt(sys.argv[1:], "c:dhH:l:m:np:v", ["config=", "debug", "help", "host=", "log=", "mock=", "nofork", "port=", "version"])
         except getopt.GetoptError:
                 options.usage(1)
@@ -436 +476 @@
                 elif opt in ("-l", "--log"):
                         options.log = arg
+                elif opt in ("-m", "--mock"):
+                        options.mock = arg
                 elif opt in ("-n", "--nofork"):
                         options.daemon = False
@@ -486 +528 @@
                 server(options.host, options.port)
         try:
-                skype = SkypeApi()
+                skype = SkypeApi(options.mock)
         except Skype4Py.SkypeAPIError, s:
                 sys.exit("%s. Are you sure you have started Skype?" % s)