Add and modify test cases for Cassandra driver