From 03ca2f5aae45802264e82c675b881d60f1269407 Mon Sep 17 00:00:00 2001
From: GregBestland <Greg.Bestland@datastax.com>
Date: Wed, 15 Jun 2016 12:46:25 -0500
Subject: [PATCH] PYTHON-569 adding test for TA LBP with execution profiles

---
 tests/integration/standard/test_cluster.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/tests/integration/standard/test_cluster.py b/tests/integration/standard/test_cluster.py
index d5945cb1..62244b93 100644
--- a/tests/integration/standard/test_cluster.py
+++ b/tests/integration/standard/test_cluster.py
@@ -702,6 +702,23 @@ class ClusterTests(unittest.TestCase):
 
             self.assertEqual(rr2_queried_hosts, rr1_queried_hosts)
 
+    def test_ta_lbp(self):
+        """
+        Test that execution profiles containing token aware LBP can be added
+
+        @since 3.5
+        @jira_ticket PYTHON-569
+        @expected_result Queries can run
+
+        @test_category config_profiles
+        """
+        query = "select release_version from system.local"
+        ta1 = ExecutionProfile()
+        with Cluster() as cluster:
+            session = cluster.connect()
+            cluster.add_execution_profile("ta1", ta1)
+            rs = session.execute(query, execution_profile='ta1')
+
     def test_clone_shared_lbp(self):
         """
         Tests that profile load balancing policies are shared on clone