diff --git a/spring-cql/src/test/java/org/springframework/cassandra/config/CassandraCqlClusterFactoryBeanUnitTests.java b/spring-cql/src/test/java/org/springframework/cassandra/config/CassandraCqlClusterFactoryBeanUnitTests.java
index 580cedbf5..37998822d 100644
--- a/spring-cql/src/test/java/org/springframework/cassandra/config/CassandraCqlClusterFactoryBeanUnitTests.java
+++ b/spring-cql/src/test/java/org/springframework/cassandra/config/CassandraCqlClusterFactoryBeanUnitTests.java
@@ -85,22 +85,6 @@ public class CassandraCqlClusterFactoryBeanUnitTests {
assertThat(bean.getObject().isClosed(), is(true));
}
- /**
- * @see DATACASS-226
- * @throws Exception
- */
- @Test
- public void shouldSetCompressionType() throws Exception {
-
- CompressionType compressionType = CompressionType.SNAPPY;
-
- CassandraCqlClusterFactoryBean bean = new CassandraCqlClusterFactoryBean();
- bean.setCompressionType(compressionType);
- bean.afterPropertiesSet();
-
- assertThat(getConfiguration(bean).getProtocolOptions().getCompression(), is(Compression.SNAPPY));
- }
-
/**
* @see DATACASS-217
* @throws Exception
@@ -114,7 +98,23 @@ public class CassandraCqlClusterFactoryBeanUnitTests {
bean.setCompressionType(compressionType);
bean.afterPropertiesSet();
- assertThat(getConfiguration(bean).getProtocolOptions().getCompression(), is(Compression.LZ4));
+ assertThat(getProtocolOptions(bean).getCompression(), is(Compression.LZ4));
+ }
+
+ /**
+ * @see DATACASS-226
+ * @throws Exception
+ */
+ @Test
+ public void shouldSetSnappyCompressionType() throws Exception {
+
+ CompressionType compressionType = CompressionType.SNAPPY;
+
+ CassandraCqlClusterFactoryBean bean = new CassandraCqlClusterFactoryBean();
+ bean.setCompressionType(compressionType);
+ bean.afterPropertiesSet();
+
+ assertThat(getProtocolOptions(bean).getCompression(), is(Compression.SNAPPY));
}
/**