com.mockrunner.test.jdbc.JDBCMockObjectFactoryTest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mockrunner-jdk1.6-j2ee1.3 Show documentation
Mockrunner is a lightweight framework for unit testing applications in the J2EE environment. It supports servlets, filters, tag classes and Struts actions. It includes a JDBC a JMS and a JCA test framework and can be used to test EJB based applications.
The newest version!
package com.mockrunner.test.jdbc;

import java.sql.Connection;
import java.sql.Driver;
import java.sql.DriverManager;
import java.sql.DriverPropertyInfo;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.Properties;

import junit.framework.TestCase;

import com.mockrunner.mock.jdbc.JDBCMockObjectFactory;
import com.mockrunner.mock.jdbc.MockConnection;
import com.mockrunner.mock.jdbc.MockDataSource;
import com.mockrunner.mock.jdbc.MockDriver;

public class JDBCMockObjectFactoryTest extends TestCase
{
    public void testCreate() throws Exception
    {
        JDBCMockObjectFactory factory = new JDBCMockObjectFactory();
        assertTrue(factory.getMockDriver().connect("", null) == factory.getMockDataSource().getConnection());
        assertNotNull(DriverManager.getConnection("test"));
        assertTrue(factory.getMockDriver().connect("", null) == DriverManager.getConnection("test"));
        Enumeration drivers = DriverManager.getDrivers();
        drivers.nextElement();
        assertFalse(drivers.hasMoreElements());
    }
    
    public void testRestoreDivers() throws Exception
    {
        DriverManager.registerDriver(new TestDriver());
        DriverManager.registerDriver(new TestDriver());
        ArrayList oldDrivers = new ArrayList();
        Enumeration drivers = DriverManager.getDrivers();
        while(drivers.hasMoreElements())
        {
            oldDrivers.add(drivers.nextElement());
        }
        JDBCMockObjectFactory factory = new JDBCMockObjectFactory();
        new JDBCMockObjectFactory();
        new JDBCMockObjectFactory();
        drivers = DriverManager.getDrivers();
        assertTrue(drivers.nextElement() instanceof MockDriver);
        assertFalse(drivers.hasMoreElements());
        factory.restoreDrivers();
        drivers = DriverManager.getDrivers();
        int numberDrivers = 0;
        while(drivers.hasMoreElements())
        {
            numberDrivers++;
            assertTrue(oldDrivers.contains(drivers.nextElement()));
        }
        factory.restoreDrivers();
        factory.restoreDrivers();
        factory.restoreDrivers();
        int newNumberDrivers = 0;
        drivers = DriverManager.getDrivers();
        while(drivers.hasMoreElements())
        {
            newNumberDrivers++;
            assertTrue(oldDrivers.contains(drivers.nextElement()));
        }
        assertTrue(numberDrivers == newNumberDrivers);
        factory.registerMockDriver();
        drivers = DriverManager.getDrivers();
        numberDrivers = 0;
        while(drivers.hasMoreElements())
        {
            numberDrivers++;
            assertTrue(drivers.nextElement() instanceof MockDriver);
        }
        assertTrue(numberDrivers == 1);
        factory.registerMockDriver();
        factory.registerMockDriver();
        factory.restoreDrivers();
        factory.restoreDrivers();
        drivers = DriverManager.getDrivers();
        numberDrivers = 0;
        while(drivers.hasMoreElements())
       {
           numberDrivers++;
           assertTrue(oldDrivers.contains(drivers.nextElement()));
       }
       assertTrue(numberDrivers == newNumberDrivers);
    }
    
    public void testOverrideCreate()
    {
        JDBCMockObjectFactory factory = new TestJDBCMockObjectFactory();
        assertNotSame(factory.getMockConnection().getClass(), MockConnection.class);
        assertNotSame(factory.getMockDataSource().getClass(), MockDataSource.class);
        assertNotSame(factory.getMockDriver().getClass(), MockDriver.class);
    }
    
    public static class TestDriver implements Driver
    {
    
        public boolean acceptsURL(String url) throws SQLException
        {
            return false;
        }

        public Connection connect(String url, Properties info) throws SQLException
        {
            return null;
        }

        public int getMajorVersion()
        {
            return 0;
        }

        public int getMinorVersion()
        {
            return 0;
        }

        public DriverPropertyInfo[] getPropertyInfo(String url, Properties info) throws SQLException
        {
            return null;
        }

        public boolean jdbcCompliant()
        {
            return false;
        }
    }
    
    public static class TestJDBCMockObjectFactory extends JDBCMockObjectFactory
    {
        public MockConnection createMockConnection()
        {
            return new MockConnection() {};
        }

        public MockDataSource createMockDataSource()
        {
            return new MockDataSource() {};
        }

        public MockDriver createMockDriver()
        {
            return new MockDriver() {};
        }
    }
}