PYTHON-118: Set 60 sec timeout for SELECT and CREATE TYPE queries to avoid test failure on slow systems.

This commit is contained in:
Irina Kaplounova
2014-08-18 09:53:51 -07:00
parent 0bf4ee1fc6
commit c323d4ba70

View File

@@ -275,7 +275,7 @@ class TypeTests(unittest.TestCase):
raise unittest.SkipTest("The tuple type was introduced in Cassandra 2.1") raise unittest.SkipTest("The tuple type was introduced in Cassandra 2.1")
MAX_TEST_LENGTH = 16384 MAX_TEST_LENGTH = 16384
timeout = 60 EXTENDED_QUERY_TIMEOUT = 60
c = Cluster(protocol_version=PROTOCOL_VERSION) c = Cluster(protocol_version=PROTOCOL_VERSION)
s = c.connect() s = c.connect()
@@ -285,8 +285,9 @@ class TypeTests(unittest.TestCase):
s.set_keyspace("test_udt_sizes") s.set_keyspace("test_udt_sizes")
# create the seed udt, increase timeout to avoid the query failure on slow systems # create the seed udt, increase timeout to avoid the query failure on slow systems
s.execute("CREATE TYPE lengthy_udt ({})".format(', '.join(['v_{} int'.format(i) s.execute("CREATE TYPE lengthy_udt ({})"
for i in range(MAX_TEST_LENGTH)])), timeout=timeout) .format(', '.join(['v_{} int'.format(i)
for i in range(MAX_TEST_LENGTH)])), timeout=EXTENDED_QUERY_TIMEOUT)
# create a table with multiple sizes of nested udts # create a table with multiple sizes of nested udts
# no need for all nested types, only a spot checked few and the largest one # no need for all nested types, only a spot checked few and the largest one
@@ -308,7 +309,7 @@ class TypeTests(unittest.TestCase):
s.execute("INSERT INTO mytable (k, v) VALUES (0, %s)", (created_udt,)) s.execute("INSERT INTO mytable (k, v) VALUES (0, %s)", (created_udt,))
# verify udt was written and read correctly, increase timeout to avoid the query failure on slow systems # verify udt was written and read correctly, increase timeout to avoid the query failure on slow systems
result = s.execute("SELECT v FROM mytable WHERE k=0", timeout=timeout)[0] result = s.execute("SELECT v FROM mytable WHERE k=0", timeout=EXTENDED_QUERY_TIMEOUT)[0]
self.assertEqual(created_udt, result.v) self.assertEqual(created_udt, result.v)
def nested_udt_helper(self, udts, i): def nested_udt_helper(self, udts, i):