fix insert test: it is only valid to check rows count only when all load is stopped

Author: kelvich

tests2/test_referee.py

@@ -45,6 +45,10 @@ def tearDownClass(cls):
         if cls.client.no_prepared_tx() != 0:
             raise AssertionError('There are some uncommitted tx')
 
+        counts = cls.client.insert_counts()
+        if counts[0] != counts[1]:
+            raise AssertionError('Different number of tuples on nodes: %d / %d' % (counts[0], counts[1]) )
+
         # XXX: check nodes data identity here
         # subprocess.check_call(['docker-compose','down'])
 
@@ -68,9 +72,6 @@ def test_neighbor_restart(self):
         self.assertCommits(aggs)
         self.assertIsolation(aggs)
 
-        counts = self.client.insert_counts()
-        self.assertEqual(counts[0], counts[1])
-
 
     def test_node_crash(self):
         print('### test_node_crash ###')
@@ -84,9 +85,6 @@ def test_node_crash(self):
         self.assertCommits(aggs)
         self.assertIsolation(aggs)
 
-        counts = self.client.insert_counts()
-        self.assertEqual(counts[0], counts[1])
-
 
     def test_partition_referee(self):
         print('### test_partition_referee ###')
@@ -100,8 +98,6 @@ def test_partition_referee(self):
         self.assertCommits(aggs)
         self.assertIsolation(aggs)
 
-        counts = self.client.insert_counts()
-        self.assertEqual(counts[0], counts[1])
 
     def test_double_failure_referee(self):
         print('### test_double_failure_referee ###')
@@ -124,8 +120,6 @@ def test_double_failure_referee(self):
         self.assertCommits(aggs)
         self.assertIsolation(aggs)
 
-        counts = self.client.insert_counts()
-        self.assertEqual(counts[0], counts[1])
 
     def test_saved_referee_decision(self):
         print('### test_saved_referee_decision ###')
@@ -198,8 +192,6 @@ def test_saved_referee_decision(self):
 
         self.assertEqual(decisions_count, 0)
 
-        counts = self.client.insert_counts()
-        self.assertEqual(counts[0], counts[1])
 
     def test_winner_restart(self):
         print('### test_winner_restart ###')
@@ -228,9 +220,6 @@ def test_winner_restart(self):
         docker_api.containers.get('node1').start()
         self.awaitCommit(0)
 
-        counts = self.client.insert_counts()
-        self.assertEqual(counts[0], counts[1])
-
 
 if __name__ == '__main__':
     unittest.main()