org.cassandraunit.dataset.yaml.ClassPathYamlDataSet Maven / Gradle / Ivy

Go to download

package org.cassandraunit.dataset.yaml;

import org.cassandraunit.dataset.DataSet;

import java.io.InputStream;

public class ClassPathYamlDataSet extends AbstractYamlDataSet implements DataSet {

    public ClassPathYamlDataSet(String dataSetLocation) {
        super(dataSetLocation);
    }

    @Override
    protected InputStream getInputDataSetLocation(String dataSetLocation) {
        InputStream inputDataSetLocation = this.getClass().getResourceAsStream("/" + dataSetLocation);
        return inputDataSetLocation;
    }

}