diff --git a/system_test/e2e/auto_clustering.py b/system_test/e2e/auto_clustering.py index 5c4ee6e2..160bac3e 100644 --- a/system_test/e2e/auto_clustering.py +++ b/system_test/e2e/auto_clustering.py @@ -9,7 +9,7 @@ import os import unittest -from helpers import Node, Cluster, d_, write_random_file, deprovision_node, random_string, TIMEOUT +from helpers import Node, Cluster, d_, write_random_file, deprovision_node, random_string, poll_query, TIMEOUT RQLITED_PATH = os.environ['RQLITED_PATH'] @@ -253,8 +253,7 @@ class TestAutoClusteringKVStores(unittest.TestCase): n3.wait_for_leader() n3.wait_for_all_applied() self.assertEqual(n3.disco_mode(), mode) - j = n3.query('SELECT * FROM foo', level='none') - self.assertEqual(j, d_("{'results': [{'values': [[1, 'fiona']], 'types': ['integer', 'text'], 'columns': ['id', 'name']}]}")) + poll_query(n3, 'SELECT * FROM foo', d_("{'results': [{'values': [[1, 'fiona']], 'types': ['integer', 'text'], 'columns': ['id', 'name']}]}")) # Add a fifth node, this time running in non-voter mode. Should join fine. n4 = Node(RQLITED_PATH, '4', raft_voter=False) @@ -262,8 +261,7 @@ class TestAutoClusteringKVStores(unittest.TestCase): n4.wait_for_leader() n4.wait_for_all_applied() self.assertEqual(n4.disco_mode(), mode) - j = n4.query('SELECT * FROM foo', level='none') - self.assertEqual(j, d_("{'results': [{'values': [[1, 'fiona']], 'types': ['integer', 'text'], 'columns': ['id', 'name']}]}")) + poll_query(n4, 'SELECT * FROM foo', d_("{'results': [{'values': [[1, 'fiona']], 'types': ['integer', 'text'], 'columns': ['id', 'name']}]}")) deprovision_node(n1) deprovision_node(n2)