def reserveConnection_nonExistantDataSourceLookedUp_shouldFail(self):
System.setProperty('java.naming.factory.initial', 'com.vaadin.addon.sqlcontainer.connection.MockInitialContextFactory')
context = EasyMock.createMock(Context)
context.lookup('foo')
EasyMock.expectLastCall().andThrow(NamingException('fail'))
MockInitialContextFactory.setMockContext(context)
EasyMock.replay(context)
pool = self.J2EEConnectionPool('foo')
pool.reserveConnection()
EasyMock.verify(context)
def reserveConnection_dataSourceLookedUp_shouldReturnValidConnection(self):
connection = EasyMock.createMock(Connection)
connection.setAutoCommit(False)
EasyMock.expectLastCall()
connection.close()
EasyMock.expectLastCall()
ds = EasyMock.createMock(DataSource)
ds.getConnection()
EasyMock.expectLastCall().andReturn(connection)
System.setProperty('java.naming.factory.initial', 'com.vaadin.data.util.sqlcontainer.connection.MockInitialContextFactory')
context = EasyMock.createMock(Context)
context.lookup('testDataSource')
EasyMock.expectLastCall().andReturn(ds)
MockInitialContextFactory.setMockContext(context)
EasyMock.replay(context, connection, ds)
pool = self.J2EEConnectionPool('testDataSource')
c = pool.reserveConnection()
Assert.assertEquals(connection, c)
pool.releaseConnection(c)
EasyMock.verify(context, connection, ds)