def _test_connect_to_host(self, url, host): """Check that a ProberFactory created with the given url will actually connect to the given host. """ prober = ProberFactory(url) def fakeConnect(host, port, factory): factory.connecting_to = host factory.succeeded('200') prober.connecting_to = None orig_connect = reactor.connectTCP reactor.connectTCP = fakeConnect def restore_connect(result, orig_connect): self.failUnlessEqual(prober.connecting_to, host) reactor.connectTCP = orig_connect return None deferred = prober.probe() return deferred.addCallback(restore_connect, orig_connect)