Add test for exception catching.

mozilla-services · Apr 9, 2015 · bd31fb5 · bd31fb5
1 parent 945a033
commit bd31fb5
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/autopush/tests/test_websocket.py b/autopush/tests/test_websocket.py
@@ -80,6 +80,20 @@ def handle_message(result):
         self._wait_for_message(d)
         return d
 
+    def test_exc_catcher(self):
+        self.proto.log_err = Mock()
+        req = Mock()
+
+        def raise_error(*args, **kwargs):
+            raise Exception("Oops")
+
+        req.headers.get.side_effect = raise_error
+
+        with self.assertRaises(Exception):
+            self.proto.onConnect(req)
+
+        req.headers.get.assert_called_with("user-agent")
+
     def test_producer_interface(self):
         self._connect()
         self.proto.pauseProducing()