/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.commons.pool.impl;
import junit.framework.TestCase;
import junit.framework.TestSuite;
import org.apache.commons.pool.BasePoolableObjectFactory;
import org.apache.commons.pool.PoolableObjectFactory;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;
/**
* @author Dirk Verbeeck
* @author Sandy McArthur
* @version $Revision: 480413 $ $Date: 2006-11-29 00:16:05 -0500 (Wed, 29 Nov 2006) $
*/
public class TestSoftRefOutOfMemory extends TestCase {
private SoftReferenceObjectPool pool;
public TestSoftRefOutOfMemory(String testName) {
super(testName);
}
public static TestSuite suite() {
return new TestSuite(TestSoftRefOutOfMemory.class);
}
public void tearDown() throws Exception {
if (pool != null) {
pool.close();
pool = null;
}
System.gc();
}
public void testOutOfMemory() throws Exception {
pool = new SoftReferenceObjectPool(new SmallPoolableObjectFactory());
Object obj = pool.borrowObject();
assertEquals("1", obj);
pool.returnObject(obj);
obj = null;
assertEquals(1, pool.getNumIdle());
final List garbage = new LinkedList();
final Runtime runtime = Runtime.getRuntime();
while (pool.getNumIdle() > 0) {
try {
garbage.add(new byte[Math.min(1024 * 1024, (int)runtime.freeMemory()/2)]);
} catch (OutOfMemoryError oome) {
System.gc();
}
System.gc();
}
garbage.clear();
System.gc();
obj = pool.borrowObject();
assertEquals("2", obj);
pool.returnObject(obj);
obj = null;
assertEquals(1, pool.getNumIdle());
}
public void testOutOfMemory1000() throws Exception {
pool = new SoftReferenceObjectPool(new SmallPoolableObjectFactory());
for (int i = 0 ; i < 1000 ; i++) {
pool.addObject();
}
Object obj = pool.borrowObject();
assertEquals("1000", obj);
pool.returnObject(obj);
obj = null;
assertEquals(1000, pool.getNumIdle());
final List garbage = new LinkedList();
final Runtime runtime = Runtime.getRuntime();
while (pool.getNumIdle() > 0) {
try {
garbage.add(new byte[Math.min(1024 * 1024, (int)runtime.freeMemory()/2)]);
} catch (OutOfMemoryError oome) {
System.gc();
}
System.gc();
}
garbage.clear();
System.gc();
obj = pool.borrowObject();
assertEquals("1001", obj);
pool.returnObject(obj);
obj = null;
assertEquals(1, pool.getNumIdle());
}
public void testOutOfMemoryLarge() throws Exception {
pool = new SoftReferenceObjectPool(new LargePoolableObjectFactory(1000000));
Object obj = pool.borrowObject();
assertTrue(((String)obj).startsWith("1."));
pool.returnObject(obj);
obj = null;
assertEquals(1, pool.getNumIdle());
final List garbage = new LinkedList();
final Runtime runtime = Runtime.getRuntime();
while (pool.getNumIdle() > 0) {
try {
garbage.add(new byte[Math.min(1024 * 1024, (int)runtime.freeMemory()/2)]);
} catch (OutOfMemoryError oome) {
System.gc();
}
System.gc();
}
garbage.clear();
System.gc();
obj = pool.borrowObject();
assertTrue(((String)obj).startsWith("2."));
pool.returnObject(obj);
obj = null;
assertEquals(1, pool.getNumIdle());
}
/**
* Makes sure an {@link OutOfMemoryError} isn't swallowed.
*/
public void testOutOfMemoryError() throws Exception {
pool = new SoftReferenceObjectPool(new BasePoolableObjectFactory() {
public Object makeObject() throws Exception {
throw new OutOfMemoryError();
}
});
try {
pool.borrowObject();
fail("Expected out of memory.");
}
catch (OutOfMemoryError ex) {
// expected
}
pool.close();
pool = new SoftReferenceObjectPool(new BasePoolableObjectFactory() {
public Object makeObject() throws Exception {
return new Object();
}
public boolean validateObject(Object obj) {
throw new OutOfMemoryError();
}
});
pool.returnObject(pool.borrowObject());
try {
pool.borrowObject();
fail("Expected out of memory.");
}
catch (OutOfMemoryError ex) {
// expected
}
pool.close();
}
public static class SmallPoolableObjectFactory implements PoolableObjectFactory {
private int counter = 0;
public Object makeObject() {
counter++;
// It seems that as of Java 1.4 String.valueOf may return an
// intern()'ed String this may cause problems when the tests
// depend on the returned object to be eventually garbaged
// collected. Either way, making sure a new String instance
// is returned eliminated false failures.
return new String(String.valueOf(counter));
}
public boolean validateObject(Object obj) {
return true;
}
public void activateObject(Object obj) { }
public void passivateObject(Object obj) { }
public void destroyObject(Object obj) { }
}
public static class LargePoolableObjectFactory implements PoolableObjectFactory {
private String buffer;
private int counter = 0;
public LargePoolableObjectFactory(int size) {
char[] data = new char[size];
Arrays.fill(data, '.');
buffer = new String(data);
}
public Object makeObject() {
counter++;
return String.valueOf(counter) + buffer;
}
public boolean validateObject(Object obj) {
return true;
}
public void activateObject(Object obj) { }
public void passivateObject(Object obj) { }
public void destroyObject(Object obj) { }
}
}