From ed2f0e3ac112ad8beb36dbcecbdd806201dc9ae8 Mon Sep 17 00:00:00 2001
From: MarcoFalke <falke.marco@gmail.com>
Date: Sun, 20 Mar 2016 18:56:13 +0100
Subject: [PATCH] [qa] maxblocksinflight: Actually enable test

Github-Pull: #7803
Rebased-From: fac724c78f281168ea174c36cada4f95112aea6d
---
 qa/rpc-tests/maxblocksinflight.py | 52 ++++++++++++++-----------------
 1 file changed, 24 insertions(+), 28 deletions(-)

diff --git a/qa/rpc-tests/maxblocksinflight.py b/qa/rpc-tests/maxblocksinflight.py
index 1a9ae480a..ec4bb5d6f 100755
--- a/qa/rpc-tests/maxblocksinflight.py
+++ b/qa/rpc-tests/maxblocksinflight.py
@@ -41,40 +41,36 @@ class TestManager(NodeConnCB):
         self.disconnectOkay = False
 
     def run(self):
-        try:
-            fail = False
-            self.connection.rpc.generate(1) # Leave IBD
+        self.connection.rpc.generate(1)  # Leave IBD
 
-            numBlocksToGenerate = [ 8, 16, 128, 1024 ]
-            for count in range(len(numBlocksToGenerate)):
-                current_invs = []
-                for i in range(numBlocksToGenerate[count]):
-                    current_invs.append(CInv(2, random.randrange(0, 1<<256)))
-                    if len(current_invs) >= 50000:
-                        self.connection.send_message(msg_inv(current_invs))
-                        current_invs = []
-                if len(current_invs) > 0:
+        numBlocksToGenerate = [8, 16, 128, 1024]
+        for count in range(len(numBlocksToGenerate)):
+            current_invs = []
+            for i in range(numBlocksToGenerate[count]):
+                current_invs.append(CInv(2, random.randrange(0, 1 << 256)))
+                if len(current_invs) >= 50000:
                     self.connection.send_message(msg_inv(current_invs))
-                
-                # Wait and see how many blocks were requested
-                time.sleep(2)
+                    current_invs = []
+            if len(current_invs) > 0:
+                self.connection.send_message(msg_inv(current_invs))
 
-                total_requests = 0
-                with mininode_lock:
-                    for key in self.blockReqCounts:
-                        total_requests += self.blockReqCounts[key]
-                        if self.blockReqCounts[key] > 1:
-                            raise AssertionError("Error, test failed: block %064x requested more than once" % key)
-                if total_requests > MAX_REQUESTS:
-                    raise AssertionError("Error, too many blocks (%d) requested" % total_requests)
-                print "Round %d: success (total requests: %d)" % (count, total_requests)
-        except AssertionError as e:
-            print "TEST FAILED: ", e.args
+            # Wait and see how many blocks were requested
+            time.sleep(2)
+
+            total_requests = 0
+            with mininode_lock:
+                for key in self.blockReqCounts:
+                    total_requests += self.blockReqCounts[key]
+                    if self.blockReqCounts[key] > 1:
+                        raise AssertionError("Error, test failed: block %064x requested more than once" % key)
+            if total_requests > MAX_REQUESTS:
+                raise AssertionError("Error, too many blocks (%d) requested" % total_requests)
+            print "Round %d: success (total requests: %d)" % (count, total_requests)
 
         self.disconnectOkay = True
         self.connection.disconnect_node()
 
-        
+
 class MaxBlocksInFlightTest(BitcoinTestFramework):
     def add_options(self, parser):
         parser.add_option("--testbinary", dest="testbinary",
@@ -86,7 +82,7 @@ class MaxBlocksInFlightTest(BitcoinTestFramework):
         initialize_chain_clean(self.options.tmpdir, 1)
 
     def setup_network(self):
-        self.nodes = start_nodes(1, self.options.tmpdir, 
+        self.nodes = start_nodes(1, self.options.tmpdir,
                                  extra_args=[['-debug', '-whitelist=127.0.0.1']],
                                  binary=[self.options.testbinary])