com.github.smokestack.jdbc.MockDataSource Maven / Gradle / Ivy

Go to download
/**
 * 
 */
package com.github.smokestack.jdbc;

import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

import javax.sql.DataSource;

import org.apache.commons.lang.builder.ReflectionToStringBuilder;
import org.apache.commons.lang.builder.ToStringStyle;

/**
 * @author bbrag
 *
 */
public class MockDataSource implements DataSource {

	/**
	 * Public access to instance
	 */
    public static final MockDataSource instance = new MockDataSource();
    
	private List mockConnections=new ArrayList();

	private PrintWriter logWriter;

	private int loginTimeout;

	private MockDataSource() {
		super();
	}

    public static synchronized MockDataSource getInstance() {
        return instance;
    }
	
	/* (non-Javadoc)
	 * @see javax.sql.DataSource#getConnection()
	 */
	public Connection getConnection() throws SQLException {
		_getConnection();
		MockConnection c=new MockConnection();
		mockConnections.add(c);
		return c;
	}

	public MockConnection _getConnection() throws SQLException {
		return null;
	}

	/* (non-Javadoc)
	 * @see javax.sql.DataSource#getConnection(java.lang.String, java.lang.String)
	 */
	public Connection getConnection(String username, String password)
			throws SQLException {
		_getConnection(username, password);
		MockConnection c=new MockConnection();
		mockConnections.add(c);
		return c;
	}

	public MockConnection _getConnection(String username, String password)
			throws SQLException {
		return null;
	}

	/* (non-Javadoc)
	 * @see javax.sql.DataSource#getLogWriter()
	 */
	public PrintWriter getLogWriter() throws SQLException {
		_getLogWriter();
		return this.logWriter;
	}

	public PrintWriter _getLogWriter() throws SQLException {
		return null;
	}

	/* (non-Javadoc)
	 * @see javax.sql.DataSource#getLoginTimeout()
	 */
	public int getLoginTimeout() throws SQLException {
		_getLoginTimeout();
		return loginTimeout;
	}

	public int _getLoginTimeout() throws SQLException {
		return -1;
	}

	/* (non-Javadoc)
	 * @see javax.sql.DataSource#setLogWriter(java.io.PrintWriter)
	 */
	public void setLogWriter(PrintWriter out) throws SQLException {
		_setLogWriter(out);
		this.logWriter = out;
	}

	public void _setLogWriter(PrintWriter out) throws SQLException {
	}

	/* (non-Javadoc)
	 * @see javax.sql.DataSource#setLoginTimeout(int)
	 */
	public void setLoginTimeout(int seconds) throws SQLException {
		_setLoginTimeout(seconds);
		this.loginTimeout = seconds;
	}

	public void _setLoginTimeout(int seconds) throws SQLException {
	}

	public List getMockConnections() {
		return mockConnections;
	}

	public void reset() {
		mockConnections=new ArrayList();
		
	}
	
	@Override
	public String toString(){
		return ReflectionToStringBuilder.toString(this, ToStringStyle.MULTI_LINE_STYLE);
	}
}