com.mockrunner.mock.connector.cci.MockConnectionFactory Maven / Gradle / Ivy

Go to download
package com.mockrunner.mock.connector.cci;

import javax.naming.NamingException;
import javax.naming.Reference;
import javax.resource.ResourceException;
import javax.resource.cci.Connection;
import javax.resource.cci.ConnectionFactory;
import javax.resource.cci.ConnectionSpec;
import javax.resource.cci.RecordFactory;
import javax.resource.cci.ResourceAdapterMetaData;

/**
 * Mock implementation of ConnectionFactory
 */
public class MockConnectionFactory implements ConnectionFactory
{
    private Connection connection;
    private RecordFactory recordFactory;
    private ResourceAdapterMetaData metaData;
    private Reference reference;

    public MockConnectionFactory()
    {
        metaData = new MockResourceAdapterMetaData();
        recordFactory  = new MockRecordFactory();
    }

    public void setConnection(Connection connection)
    {
        this.connection = connection;
    }

    public Connection getConnection() throws ResourceException
    {
        return connection;
    }

    public MockConnection getMockConnection()
    {
        if(connection instanceof MockConnection)
        {
            return (MockConnection)connection;
        }
        return null;
    }

    public Connection getConnection(ConnectionSpec cs) throws ResourceException
    {
        return connection;
    }

    public RecordFactory getRecordFactory() throws ResourceException
    {
        return recordFactory;
    }

    public ResourceAdapterMetaData getMetaData() throws ResourceException
    {
        return metaData;
    }

    public void setReference(Reference reference)
    {
        this.reference = reference;
    }

    public Reference getReference() throws NamingException
    {
        return reference;
    }
    
    /**
     * Sets the resource adapter meta data. If you do not set an explicit
     * ResourceAdapterMetaData object, a default {@link MockResourceAdapterMetaData} 
     * will be created.
     * @param metaData the ResourceAdapterMetaData
     */
    public void setMetaData(ResourceAdapterMetaData metaData)
    {
        this.metaData = metaData;
    }
    
    /**
     * Sets the record factory. If you do not set an explicit
     * RecordFactory, a default {@link MockRecordFactory} 
     * will be created.
     * @param recordFactory the RecordFactory
     */
    public void setRecordFactory(RecordFactory recordFactory)
    {
        this.recordFactory = recordFactory;
    }
}