/*
* 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 java.util.Arrays;
import java.util.LinkedList;
import java.util.List;
import junit.framework.TestCase;
import org.apache.commons.pool.BasePoolableObjectFactory;
import org.apache.commons.pool.PoolableObjectFactory;
/**
* @author Dirk Verbeeck
* @author Sandy McArthur
* @version $Revision: 1222396 $ $Date: 2011-12-22 14:02:25 -0500 (Thu, 22 Dec 2011) $
*/
public class TestSoftRefOutOfMemory extends TestCase {
private SoftReferenceObjectPool<String> pool;
public TestSoftRefOutOfMemory(String testName) {
super(testName);
}
@Override
public void tearDown() throws Exception {
if (pool != null) {
pool.close();
pool = null;
}
System.gc();
}
public void testOutOfMemory() throws Exception {
pool = new SoftReferenceObjectPool<String>(new SmallPoolableObjectFactory());
String obj = pool.borrowObject();
assertEquals("1", obj);
pool.returnObject(obj);
obj = null;
assertEquals(1, pool.getNumIdle());
final List<byte[]> garbage = new LinkedList<byte[]>();
final Runtime runtime = Runtime.getRuntime();
while (pool.getNumIdle() > 0) {
try {
long freeMemory = runtime.freeMemory();
if (freeMemory > Integer.MAX_VALUE) {
freeMemory = Integer.MAX_VALUE;
}
garbage.add(new byte[Math.min(1024 * 1024, (int)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<String>(new SmallPoolableObjectFactory());
for (int i = 0 ; i < 1000 ; i++) {
pool.addObject();
}
String obj = pool.borrowObject();
assertEquals("1000", obj);
pool.returnObject(obj);
obj = null;
assertEquals(1000, pool.getNumIdle());
final List<byte[]> garbage = new LinkedList<byte[]>();
final Runtime runtime = Runtime.getRuntime();
while (pool.getNumIdle() > 0) {
try {
long freeMemory = runtime.freeMemory();
if (freeMemory > Integer.MAX_VALUE) {
freeMemory = Integer.MAX_VALUE;
}
garbage.add(new byte[Math.min(1024 * 1024, (int)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<String>(new LargePoolableObjectFactory(1000000));
String obj = pool.borrowObject();
assertTrue(obj.startsWith("1."));
pool.returnObject(obj);
obj = null;
assertEquals(1, pool.getNumIdle());
final List<byte[]> garbage = new LinkedList<byte[]>();
final Runtime runtime = Runtime.getRuntime();
while (pool.getNumIdle() > 0) {
try {
long freeMemory = runtime.freeMemory();
if (freeMemory > Integer.MAX_VALUE) {
freeMemory = Integer.MAX_VALUE;
}
garbage.add(new byte[Math.min(1024 * 1024, (int)freeMemory/2)]);
} catch (OutOfMemoryError oome) {
System.gc();
}
System.gc();
}
garbage.clear();
System.gc();
obj = pool.borrowObject();
assertTrue(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<String>(new BasePoolableObjectFactory<String>() {
@Override
public String makeObject() throws Exception {
throw new OutOfMemoryError();
}
});
try {
pool.borrowObject();
fail("Expected out of memory.");
}
catch (OutOfMemoryError ex) {
// expected
}
pool.close();
pool = new SoftReferenceObjectPool<String>(new BasePoolableObjectFactory<String>() {
@Override
public String makeObject() throws Exception {
return new String();
}
@Override
public boolean validateObject(String obj) {
throw new OutOfMemoryError();
}
});
try {
pool.borrowObject();
fail("Expected out of memory.");
}
catch (OutOfMemoryError ex) {
// expected
}
pool.close();
pool = new SoftReferenceObjectPool<String>(new BasePoolableObjectFactory<String>() {
@Override
public String makeObject() throws Exception {
return new String();
}
@Override
public boolean validateObject(String obj) {
throw new IllegalAccessError();
}
@Override
public void destroyObject(String obj) throws Exception {
throw new OutOfMemoryError();
}
});
try {
pool.borrowObject();
fail("Expected out of memory.");
}
catch (OutOfMemoryError ex) {
// expected
}
pool.close();
}
public static class SmallPoolableObjectFactory implements PoolableObjectFactory<String> {
private int counter = 0;
public String 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(String obj) {
return true;
}
public void activateObject(String obj) { }
public void passivateObject(String obj) { }
public void destroyObject(String obj) { }
}
public static class LargePoolableObjectFactory implements PoolableObjectFactory<String> {
private String buffer;
private int counter = 0;
public LargePoolableObjectFactory(int size) {
char[] data = new char[size];
Arrays.fill(data, '.');
buffer = new String(data);
}
public String makeObject() {
counter++;
return String.valueOf(counter) + buffer;
}
public boolean validateObject(String obj) {
return true;
}
public void activateObject(String obj) { }
public void passivateObject(String obj) { }
public void destroyObject(String obj) { }
}
}