def test_disconnect(self): """ When disconnect is called, the on demand thrift client is disconnected """ client = TestingCQLClient(self.endpoint, 'blah') self.assertFired(client.describe_version()) client.disconnect() self.twisted_transport.loseConnection.assert_called_once_with()