Added more features to the mock server.

Author: rshipp

tests/mockserver.py

@@ -4,6 +4,7 @@ class MockServer(object):
     def __init__(self, host=None, port=None, broken=True, ok=True,
             broken_username=False):
         self.valid = "test"
+        self.valid_desc = '"Test UPS 1"'
         self.broken = broken
         self.ok = ok
         self.broken_username = broken_username
@@ -33,7 +34,7 @@ def run_command(self):
             self.first = False
             return 'BEGIN LIST UPS\n'
         elif self.command == "LIST UPS\n":
-            return 'UPS '+self.valid+'"Test UPS 1"\nUPS Test_UPS2 "Test UPS 2"\nEND LIST UPS\n'
+            return 'UPS '+self.valid+' '+self.valid_desc+'\nUPS Test_UPS2 "Test UPS 2"\nEND LIST UPS\n'
         elif self.command == "LIST VAR %s\n" % self.valid and self.first:
             self.first = False
             return 'BEGIN LIST VAR '+self.valid+'\n'
@@ -62,7 +63,18 @@ def run_command(self):
             return 'CLIENT '+self.valid+' '+self.valid+'\nEND LIST CLIENTS\n'
         elif self.command.startswith("LIST CLIENTS"):
             return 'ERR INVALID-ARGUMENT\n'
-        # TODO: SET commands
+        elif self.command == "LIST ENUM %s %s\n" % (self.valid, self.valid) and self.first:
+            self.first = False
+            return 'BEGIN LIST ENUM %s %s\n' (self.valid, self.valid)
+        elif self.command == "LIST ENUM %s %s\n" % (self.valid, self.valid):
+            return 'ENUM %s %s %s\nEND LIST ENUM %s %s\n' % (self.valid,
+                    self.valid, self.valid_desc, self.valid, self.valid)
+        elif self.command == "LIST RANGE %s %s\n" % (self.valid, self.valid) and self.first:
+            self.first = False
+            return 'BEGIN LIST RANGE %s %s\n' (self.valid, self.valid)
+        elif self.command == "LIST RANGE %s %s\n" % (self.valid, self.valid):
+            return 'RANGE %s %s %s %s\nEND LIST RANGE %s %s\n' % (self.valid,
+                    self.valid, self.valid_desc, self.valid_desc, self.valid, self.valid)
         elif self.command == "SET VAR %s %s %s\n" % (self.valid, self.valid, self.valid):
             return 'OK\n'
         elif self.command.startswith("SET"):
@@ -75,11 +87,11 @@ def run_command(self):
         elif self.command == "USERNAME %s\n" % self.valid:
             return 'OK\n'
         elif self.command.startswith("USERNAME"):
-            return 'OK\n'  # FIXME
+            return 'ERR\n'  # FIXME: What does it say on invalid password?
         elif self.command == "PASSWORD %s\n" % self.valid:
             return 'OK\n'
         elif self.command.startswith("PASSWORD"):
-            return 'OK\n'  # FIXME
+            return 'ERR\n'  # FIXME: ^
         elif self.command == "STARTTLS\n":
             return 'ERR FEATURE-NOT-CONFIGURED\n'
         elif self.command == "MASTER %s\n" % self.valid:
@@ -88,5 +100,35 @@ def run_command(self):
             return 'OK FSD-SET\n'
         elif self.command == "FSD %s\n" % self.valid:
             return 'ERR\n'
+        elif self.command == "GET NUMLOGINS %s\n" % self.valid:
+            return 'NUMLOGINS %s 1\n' % self.valid
+        elif self.command.startswith("GET NUMLOGINS"):
+            return 'ERR UNKNOWN-UPS\n'
+        elif self.command == "GET UPSDESC %s\n" % self.valid:
+            return 'UPSDESC %s %s\n' % (self.valid, self.valid_desc)
+        elif self.command.startswith("GET UPSDESC"):
+            return 'ERR UNKNOWN-UPS\n'
+        elif self.command == "GET VAR %s %s\n" % (self.valid, self.valid):
+            return 'VAR %s %s "100"\n' % (self.valid, self.valid)
+        elif self.command.startswith("GET VAR %s" % self.valid):
+            return 'ERR VAR-NOT-SUPPORTED\n'
+        elif self.command.startswith("GET VAR "):
+            return 'ERR UNKNOWN-UPS\n'
+        elif self.command.startswith("GET VAR"):
+            return 'ERR INVALID-ARGUMENT\n'
+        elif self.command == "GET TYPE %s %s\n" % (self.valid, self.valid):
+            return 'TYPE %s %s RW STRING:3\n' % (self.valid, self.valid)
+        elif self.command.startswith("GET TYPE %s" % self.valid):
+            return 'ERR VAR-NOT-SUPPORTED\n'
+        elif self.command.startswith("GET TYPE"):
+            return 'ERR INVALID-ARGUMENT\n'
+        elif self.command == "GET DESC %s %s\n" % (self.valid, self.valid):
+            return 'DESC %s %s %s\n' % (self.valid, self.valid, self.valid_desc)
+        elif self.command.startswith("GET DESC"):
+            return 'ERR-INVALID-ARGUMENT\n'
+        elif self.command == "GET CMDDESC %s %s" % (self.valid, self.valid):
+            return 'CMDDESC %s %s %s\n' % (self.valid, self.valid, self.valid_desc)
+        elif self.command.startswith("GET CMDDESC"):
+            return 'ERR INVALID-ARGUMENT'
         else:
             return 'ERR UNKNOWN-COMMAND\n'