[Supervisor-checkins] r909 - in supervisor/trunk/src/supervisor: . tests

[Mike Naberezny]

Author: Mike Naberezny <mike at maintainable.com>
Date: Sat Oct 24 23:17:54 2009
New Revision: 909

Log:
Use consistent error messages and add tests for upcheck().

Modified:
   supervisor/trunk/src/supervisor/supervisorctl.py
   supervisor/trunk/src/supervisor/tests/test_supervisorctl.py

Modified: supervisor/trunk/src/supervisor/supervisorctl.py
==============================================================================
--- supervisor/trunk/src/supervisor/supervisorctl.py	(original)
+++ supervisor/trunk/src/supervisor/supervisorctl.py	Sat Oct 24 23:17:54 2009
@@ -228,8 +228,7 @@
                 self.output('%s refused connection' % self.options.serverurl)
                 return False
             elif why[0] == errno.ENOENT:
-                msg = 'ERROR: %s no such file (already shut down?)'
-                self.output(msg % self.options.serverurl)
+                self.output('%s no such file' % self.options.serverurl)
                 return False
             raise
         return True

Modified: supervisor/trunk/src/supervisor/tests/test_supervisorctl.py
==============================================================================
--- supervisor/trunk/src/supervisor/tests/test_supervisorctl.py	(original)
+++ supervisor/trunk/src/supervisor/tests/test_supervisorctl.py	Sat Oct 24 23:17:54 2009
@@ -53,6 +53,40 @@
         '[rpcinterface:supervisor] section is enabled in the '
         'configuration file (see sample.conf).\n')
 
+    def test__upcheck_catches_socket_error_ECONNREFUSED(self):
+        options = DummyClientOptions()
+        import socket
+        import errno
+        def raise_fault(*arg, **kw):     
+            raise socket.error(errno.ECONNREFUSED, 'nobody home')
+        options._server.supervisor.getVersion = raise_fault
+
+        controller = self._makeOne(options)
+        controller.stdout = StringIO()
+
+        result = controller.upcheck()
+        self.assertEqual(result, False)
+
+        output = controller.stdout.getvalue() 
+        self.assertTrue('refused connection' in output)
+
+    def test__upcheck_catches_socket_error_ENOENT(self):
+        options = DummyClientOptions()
+        import socket
+        import errno
+        def raise_fault(*arg, **kw):     
+            raise socket.error(errno.ENOENT, 'nobody home')
+        options._server.supervisor.getVersion = raise_fault
+
+        controller = self._makeOne(options)
+        controller.stdout = StringIO()
+
+        result = controller.upcheck()
+        self.assertEqual(result, False)
+
+        output = controller.stdout.getvalue() 
+        self.assertTrue('no such file' in output)
+
     def test_onecmd(self):
         options = DummyClientOptions()
         controller = self._makeOne(options)