Updating CQL test to insert and return multiple rows