Insert.java

package dbfit.fixture;

import dbfit.api.DBEnvironment;
import dbfit.api.DbEnvironmentFactory;
import dbfit.api.DbObject;
import dbfit.api.DbTable;

import java.sql.SQLException;

public class Insert extends DbObjectExecutionFixture {
    private DBEnvironment environment;
    private String tableName;

    public Insert() {
        environment = DbEnvironmentFactory.getDefaultEnvironment();
    }

    public Insert(DBEnvironment dbEnvironment) {
        this.environment = dbEnvironment;
    }

    public Insert(DBEnvironment dbEnvironment, String tableName) {
        this.tableName = tableName;
        this.environment = dbEnvironment;
    }

    @Override
    protected DbObject getTargetDbObject() throws SQLException {
        if ((tableName == null || tableName.trim().length() == 0) && args.length > 0) {
            tableName = args[0];
        };
        return new DbTable(environment, tableName);
    }
}