// Path: cassandra-unit/src/main/java/org/cassandraunit/dataset/CQLDataSet.java
// public interface CQLDataSet {
// 
//     List<String> getCQLStatements();
// 
//     String getKeyspaceName();
// 
//     boolean isKeyspaceCreation();
// 
//     boolean isKeyspaceDeletion();
// }
// 
// Path: cassandra-unit/src/main/java/org/cassandraunit/utils/CqlOperations.java
// public class CqlOperations {
// 
//     private static final Logger log = LoggerFactory.getLogger(CqlOperations.class);
// 
//     public static Consumer<String> execute(CqlSession session) {
//         return query -> {
//             log.debug("executing : {}", query);
//             session.execute(query);
//         };
//     }
// 
//     public static Consumer<String> use(CqlSession session) {
//         return keyspace -> session.execute("USE " + keyspace);
//     }
// 
//     public static Consumer<String> truncateTable(CqlSession session) {
//         return fullyQualifiedTable -> session.execute("truncate table " + fullyQualifiedTable);
//     }
// 
//     public static Consumer<String> dropKeyspace(CqlSession session) {
//         return keyspace -> execute(session).accept("DROP KEYSPACE IF EXISTS " + keyspace);
//     }
// 
//     public static Consumer<String> createKeyspace(CqlSession session) {
//         return keyspace -> execute(session).accept("CREATE KEYSPACE IF NOT EXISTS " + keyspace + " WITH replication={'class' : 'SimpleStrategy', 'replication_factor':1} AND durable_writes = false");
//     }
// }


import com.datastax.oss.driver.api.core.CqlSession;
import org.cassandraunit.dataset.CQLDataSet;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import static org.cassandraunit.utils.CqlOperations.*;
package org.cassandraunit;



/**
 * @author Marcin Szymaniuk
 * @author Jeremy Sevellec
 */
public class CQLDataLoader {

    private static final Logger log = LoggerFactory.getLogger(CQLDataLoader.class);
    public static final String DEFAULT_KEYSPACE_NAME = "cassandraunitkeyspace";

    public CqlSession getSession() {
        return session;
    }

    private final CqlSession session;

    public CQLDataLoader(CqlSession session) {
        this.session = session;
    }

// Path: cassandra-unit/src/main/java/org/cassandraunit/dataset/CQLDataSet.java
// public interface CQLDataSet {
// 
//     List<String> getCQLStatements();
// 
//     String getKeyspaceName();
// 
//     boolean isKeyspaceCreation();
// 
//     boolean isKeyspaceDeletion();
// }
// 
// Path: cassandra-unit/src/main/java/org/cassandraunit/utils/CqlOperations.java
// public class CqlOperations {
// 
//     private static final Logger log = LoggerFactory.getLogger(CqlOperations.class);
// 
//     public static Consumer<String> execute(CqlSession session) {
//         return query -> {
//             log.debug("executing : {}", query);
//             session.execute(query);
//         };
//     }
// 
//     public static Consumer<String> use(CqlSession session) {
//         return keyspace -> session.execute("USE " + keyspace);
//     }
// 
//     public static Consumer<String> truncateTable(CqlSession session) {
//         return fullyQualifiedTable -> session.execute("truncate table " + fullyQualifiedTable);
//     }
// 
//     public static Consumer<String> dropKeyspace(CqlSession session) {
//         return keyspace -> execute(session).accept("DROP KEYSPACE IF EXISTS " + keyspace);
//     }
// 
//     public static Consumer<String> createKeyspace(CqlSession session) {
//         return keyspace -> execute(session).accept("CREATE KEYSPACE IF NOT EXISTS " + keyspace + " WITH replication={'class' : 'SimpleStrategy', 'replication_factor':1} AND durable_writes = false");
//     }
// }


// Path: cassandra-unit/src/main/java/org/cassandraunit/CQLDataLoader.java
import com.datastax.oss.driver.api.core.CqlSession;
import org.cassandraunit.dataset.CQLDataSet;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import static org.cassandraunit.utils.CqlOperations.*;

package org.cassandraunit;



/**
 * @author Marcin Szymaniuk
 * @author Jeremy Sevellec
 */
public class CQLDataLoader {

    private static final Logger log = LoggerFactory.getLogger(CQLDataLoader.class);
    public static final String DEFAULT_KEYSPACE_NAME = "cassandraunitkeyspace";

    public CqlSession getSession() {
        return session;
    }

    private final CqlSession session;

    public CQLDataLoader(CqlSession session) {
        this.session = session;
    }

    public void load(CQLDataSet dataSet) {