Ticket #2876: 2876.diff

twisted/test/test_pb.py

@@ -1132 +1132 @@
             def clientConnectionLost(self, connector, reason):
                 reconnecting = not self.reconnectedAlready
                 self.reconnectedAlready = True
-                connector.connect()
+                if reconnecting:
+                    connector.connect()
                 return pb.PBClientFactory.clientConnectionLost(
                     self, connector, reason, reconnecting)
 
         factory, rootObjDeferred = self.getFactoryAndRootObject(ReconnectOnce)
 
         def gotRootObject(rootObj):
-            self.failUnless(
-                isinstance(rootObj, pb.RemoteReference),
-                "%r is not a RemoteReference" % (rootObj,))
+            self.assertIsInstance(rootObj, pb.RemoteReference)
 
             d = defer.Deferred()
             rootObj.notifyOnDisconnect(d.callback)
@@ -1151 +1150 @@
                 d = factory.getRootObject()
 
                 def gotAnotherRootObject(anotherRootObj):
-                    self.failUnless(
-                        isinstance(rootObj, pb.RemoteReference),
-                        "%r is not a RemoteReference" % (rootObj,))
+                    self.assertIsInstance(anotherRootObj, pb.RemoteReference)
 
                     d = defer.Deferred()
                     anotherRootObj.notifyOnDisconnect(d.callback)