From 5357c3426d1703cdacd61f7652082bfa543f59c9 Mon Sep 17 00:00:00 2001
From: Dmitry Kropachev <dmitry.kropachev@gmail.com>
Date: Tue, 13 Aug 2024 06:17:04 -0400
Subject: [PATCH] Temporary disable AES256ColumnEncryptionPolicy

There was some seriouse issues in policy implementation in `3.27.0`.
We want to inspect the feature before making it available.
---
 cassandra/cluster.py                                          | 2 ++
 tests/integration/standard/column_encryption/test_policies.py | 1 +
 tests/unit/column_encryption/test_policies.py                 | 1 +
 3 files changed, 4 insertions(+)

diff --git a/cassandra/cluster.py b/cassandra/cluster.py
index 17dcb2b7c7..754325e4e4 100644
--- a/cassandra/cluster.py
+++ b/cassandra/cluster.py
@@ -2686,6 +2686,8 @@ def __init__(self, cluster, hosts, keyspace=None):
                     {"column_encryption_policy": self.cluster.column_encryption_policy})
             except AttributeError:
                 log.info("Unable to set column encryption policy for session")
+            raise Exception(
+                "column_encryption_policy is temporary disabled, until https://github.com/scylladb/python-driver/issues/365 is sorted out")
 
         if self.cluster.monitor_reporting_enabled:
             cc_host = self.cluster.get_control_connection_host()
diff --git a/tests/integration/standard/column_encryption/test_policies.py b/tests/integration/standard/column_encryption/test_policies.py
index dea6b6d39e..8af27ab87a 100644
--- a/tests/integration/standard/column_encryption/test_policies.py
+++ b/tests/integration/standard/column_encryption/test_policies.py
@@ -25,6 +25,7 @@
 def setup_module():
     use_singledc()
 
+@unittest.skip("Skip until https://github.com/scylladb/python-driver/issues/365 is sorted out")
 class ColumnEncryptionPolicyTest(unittest.TestCase):
 
     def _recreate_keyspace(self, session):
diff --git a/tests/unit/column_encryption/test_policies.py b/tests/unit/column_encryption/test_policies.py
index 38136c69d4..27e7c62ce7 100644
--- a/tests/unit/column_encryption/test_policies.py
+++ b/tests/unit/column_encryption/test_policies.py
@@ -19,6 +19,7 @@
 from cassandra.column_encryption.policies import AES256ColumnEncryptionPolicy, \
     AES256_BLOCK_SIZE_BYTES, AES256_KEY_SIZE_BYTES
 
+@unittest.skip("Skip until https://github.com/scylladb/python-driver/issues/365 is sorted out")
 class AES256ColumnEncryptionPolicyTest(unittest.TestCase):
 
     def _random_block(self):