check data identity in tests

kelvich · kelvich · commit f0ed19f119cc · 2017-11-13T02:17:36.000+03:00
diff --git a/tests2/lib/bank_client.py b/tests2/lib/bank_client.py
@@ -109,6 +109,24 @@ def initdb(self):
         cur.close()
         conn.close()
 
+    def is_data_identic(self):
+        hashes = set()
+
+        for dsn in self.dsns:
+            con = psycopg2.connect(dsn)
+            cur = con.cursor()
+            cur.execute("""
+                select
+                    md5('(' || string_agg(uid::text || ', ' || amount::text , '),(') || ')')
+                from
+                    (select * from bank_test order by uid) t;""")
+            hashes.add(cur.fetchone()[0])
+            cur.close()
+            con.close()
+
+        print(hashes)
+        return (len(hashes) == 1)
+
     @asyncio.coroutine
     def status(self):
         while self.running:
@@ -168,7 +186,7 @@ def exec_tx(self, tx_block, aggname_prefix, conn_i):
                 yield from asyncio.sleep(0.01)
             except BaseException as e:
                 agg.finish_tx(str(e).strip())
-                print('Catch exception ', e)
+                print('Catch exception ', str(e).strip())
                 # Give evloop some free time.
                 # In case of continuous excetions we can loop here without returning
                 # back to event loop and block it
diff --git a/tests2/test_recovery.py b/tests2/test_recovery.py
@@ -68,7 +68,7 @@ def performFailure(self, failure):
 class RecoveryTest(unittest.TestCase, TestHelper):
 
     @classmethod
-    def setUpClass(self):
+    def setUpClass(cls):
         subprocess.check_call(['docker-compose','up',
             '--force-recreate',
             '--build',
@@ -77,17 +77,21 @@ def setUpClass(self):
         # XXX: add normal wait here
         time.sleep(TEST_SETUP_TIME)
 
-        self.client = MtmClient([
+        cls.client = MtmClient([
             "dbname=regression user=postgres host=127.0.0.1 port=15432",
             "dbname=regression user=postgres host=127.0.0.1 port=15433",
             "dbname=regression user=postgres host=127.0.0.1 port=15434"
         ], n_accounts=1000)
-        self.client.bgrun()
+        cls.client.bgrun()
 
     @classmethod
-    def tearDownClass(self):
+    def tearDownClass(cls):
         print('tearDown')
-        self.client.stop()
+        cls.client.stop()
+
+        if not cls.client.is_data_identic():
+            raise AssertionError('Different data on nodes')
+
         # XXX: check nodes data identity here
 #        subprocess.check_call(['docker-compose','down'])
 
