Test for trace_id on query

PYTHON-302
datastax · May 12, 2015 · bc3ea9b · bc3ea9b
1 parent 58536f5
commit bc3ea9b
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/integration/standard/test_query.py b/tests/integration/standard/test_query.py
@@ -73,6 +73,21 @@ def test_trace_prints_okay(self):
 
         cluster.shutdown()
 
+    def test_trace_id_to_query(self):
+        cluster = Cluster(protocol_version=PROTOCOL_VERSION)
+        session = cluster.connect()
+
+        query = "SELECT * FROM system.local"
+        statement = SimpleStatement(query)
+        self.assertIsNone(statement.trace_id)
+        future = session.execute_async(statement, trace=True)
+
+        # query should have trace_id, even before trace is obtained
+        future.result()
+        self.assertIsNotNone(statement.trace_id)
+
+        cluster.shutdown()
+
     def test_trace_ignores_row_factory(self):
         cluster = Cluster(protocol_version=PROTOCOL_VERSION)
         session = cluster.connect()