Extra tests.

Author: Tom Galloway

market/listeners.py

@@ -4,6 +4,7 @@
 import json
 import time
 import random
+from log import Logger
 from interfaces import MessageListener, BroadcastListener, NotificationListener
 from zope.interface import implements
 from protos.objects import PlaintextMessage, Following
@@ -19,6 +20,7 @@ class MessageListenerImpl(object):
     def __init__(self, web_socket_factory, database):
         self.ws = web_socket_factory
         self.db = database
+        self.log = Logger(system=self)
 
     def notify(self, plaintext, signature):
         try:
@@ -43,8 +45,8 @@ def notify(self, plaintext, signature):
             if plaintext.handle:
                 message_json["message"]["handle"] = plaintext.handle
             self.ws.push(str(bleach.clean(json.dumps(message_json, indent=4), tags=ALLOWED_TAGS)))
-        except Exception:
-            pass
+        except Exception as e:
+            self.log.error('Market.Listener.notify Exception: %s' % e)
 
 
 class BroadcastListenerImpl(object):

market/tests/test_listeners.py

@@ -74,3 +74,11 @@ def test_MarketListeners_notify_with_handle_success(self):
                                                          'test_avatar_hash',
                                                          signature, False)
         self.ws.push.assert_called_with(self._create_valid_message_json('test_handle'))
+
+    def test_MarketListeners_save_message_exception(self):
+        p = self._create_valid_plaintext_message('test_handle')
+        signature = 'test_signature'
+        l = MessageListenerImpl(self.ws, self.db)
+        self.db.messages.save_message.side_effect = Exception("test_exception")
+        l.notify(p, signature)
+        self.assertEqual('[ERROR] Market.Listener.notify Exception: test_exception', self.catcher[0]['message'][0])