Merge pull request #5562

13864702 functional_tests: fix rare get_output_distribution failure (moneromooo-monero)
2019-06-01 20:30:54 +02:00 · 2019-06-01 20:30:54 +02:00 · 2f5efc7f59
parent 0565fe21ec 13864702f1
commit 2f5efc7f59
1 changed files with 10 additions and 1 deletions
--- a/tests/functional_tests/get_output_distribution.py
+++ b/tests/functional_tests/get_output_distribution.py
@ -213,5 +213,14 @@ class GetOutputDistributionTest():
            assert d.distribution[h] == 0


+class Guard:
+    def __enter__(self):
+        for i in range(4):
+            Wallet(idx = i).auto_refresh(False)
+    def __exit__(self, exc_type, exc_value, traceback):
+        for i in range(4):
+            Wallet(idx = i).auto_refresh(True)
+
 if __name__ == '__main__':
+    with Guard() as guard:
        GetOutputDistributionTest().run_test()