<!-- Prompt runner test target, runs test.class from test.properties (default is LRG). -->

    <target name="test.prompt">

        <antcall target="run.test.prompt" inheritRefs="true">

            <param name="TEST_CLASS" value="${test.class}"/>

        </antcall>

    </target>

    <!-- Runs performance tests. -->

    <target name="test.performance">

        <antcall target="run.test.prompt" inheritRefs="true">

            <param name="TEST_CLASS" value="org.eclipse.persistence.testing.tests.performance.PerformanceTestModel"/>

        </antcall>

    </target>

    <!-- Launches the testing browser -->

    <target name="test.browser">

        <echo>Running test browser with ${test.properties}</echo>

        <mkdir dir="${eclipselink.core.test}/${run.dir}"/>

        <mkdir dir="${eclipselink.core.test}/${report.dir}"/>

        <java classname="org.eclipse.persistence.testing.framework.ui.TestingBrowserFrame"

             fork="true"

             maxmemory="512m"

             failonerror="true"

             dir="${eclipselink.core.test}/${run.dir}">

            <jvmarg value="-Declipselink.logging.level=${logging.level}"/>

            <jvmarg value="-Dtest.properties=${test.properties.run}"/>

            <jvmarg value="-Djava.security.manager"/>

            <jvmarg value="-Djava.security.policy=./java.policy.allpermissions"/>

            <classpath>

                <path refid="run.classpath"/>

            </classpath>

        </java>

    </target>

    <!-- Runs the test using the PromptTestRunner -->

    <target name="run.test.prompt">

        <echo>Running test with ${test.properties} on ${db.url}</echo>

        <mkdir dir="${eclipselink.core.test}/${run.dir}"/>

        <mkdir dir="${eclipselink.core.test}/${report.dir}"/>

        <java classname="org.eclipse.persistence.testing.framework.PromptTestRunner"

             fork="true"

             maxmemory="512m"

             failonerror="true"

             dir="${eclipselink.core.test}/${run.dir}">

            <jvmarg value="-Declipselink.logging.level=${logging.level}"/>

            <jvmarg value="-Dtest.properties=${test.properties.run}"/>

            <jvmarg value="-Djava.security.manager"/>

            <jvmarg value="-Djava.security.policy=./java.policy.allpermissions"/>

            <arg value="pfile=${test.properties.run}" />

            <arg value="testmodel=${TEST_CLASS}" />

            <arg value="regressionlog=../${report.dir}/test.log" />

            <arg value="saveresults=true" />

            <classpath>

                <path refid="run.classpath"/>

            </classpath>

        </java>

    </target>

            // Trim the machine name only because of network issues.

            if (this.machine.indexOf(".") != -1) {

                this.machine = this.machine.substring(0, this.machine.indexOf("."));

            }

/*******************************************************************************

 * Copyright (c) 1998, 2008 Oracle. All rights reserved.

 * This program and the accompanying materials are made available under the 

 * terms of the Eclipse Public License v1.0 and Eclipse Distribution License v. 1.0 

 * which accompanies this distribution. 

 * The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html

 * and the Eclipse Distribution License is available at 

 * http://www.eclipse.org/org/documents/edl-v10.php.

 *

 * Contributors:

 *     Oracle - initial API and implementation from Oracle TopLink

 ******************************************************************************/  

package org.eclipse.persistence.testing.framework;

import java.util.ArrayList;

import java.util.List;

import java.lang.ref.WeakReference;

/**

 * Generic test for testing for memory leaks.

 */

public abstract class MemoryLeakTestCase extends TestCase {

    /** Stores objects that should garbage collect. */

    protected List<WeakReference> weakList;

    /** Set number of objects allowed not to garbage collect. */

    protected int threshold = 0;

    /**

     * Add a weak reference to the object to ensure it garbage collects.

     */

    public void addWeakReference(Object reference) {

        this.weakList.add(new WeakReference(reference));

    }

    /**

     * Add a weak reference to the object to ensure it garbage collects.

     */

    public void addWeakReferences(List references) {

        for (Object ref : references) {

            addWeakReference(ref);

        }

    }

    public int getThreshold() {

        return threshold;

    }

    public void setThreshold(int threshold) {

        this.threshold = threshold;

    }

    public void setup() {

       this.weakList = new ArrayList();

       getSession().getIdentityMapAccessor().initializeAllIdentityMaps();

    }

    public void verify() {

        forceGC();

        int count = 0;

        for (WeakReference ref : this.weakList) {

            if (ref.get() != null) {

                count++;

            }

        }

        if (count > this.threshold) {

            throwError("Objects did not garbage collect: " + count);

        }

    }

    /**

     * Force a garbage collection.

     */

    public void forceGC() {

        WeakReference ref = new WeakReference(new Object());

        for (int loops = 0; loops < 10; loops++) {

            List junk = new ArrayList (50000);

            for (int i = 0; i < 50000; i++) {

                junk.add(new java.math.BigDecimal(i));

            }

            // Force garbage collection.

            System.gc();

            System.runFinalization();

        }

        // Check if a garbage collect really occurred.

        if (ref.get() != null) {

            System.out.println("WARNING: gc did not occur");

        }

    }    

}

        performanceTests.addElement("org.eclipse.persistence.testing.tests.jpa.memory.JPAMemoryTestModel");

/*******************************************************************************

 * Copyright (c) 1998, 2008 Oracle. All rights reserved.

 * This program and the accompanying materials are made available under the 

 * terms of the Eclipse Public License v1.0 and Eclipse Distribution License v. 1.0 

 * which accompanies this distribution. 

 * The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html

 * and the Eclipse Distribution License is available at 

 * http://www.eclipse.org/org/documents/edl-v10.php.

 *

 * Contributors:

 *     Oracle - initial API and implementation from Oracle TopLink

 ******************************************************************************/  

 package org.eclipse.persistence.testing.tests.jpa.memory;

import java.util.List;

import java.util.HashMap;

import java.util.Map;

import javax.persistence.EntityManager;

import javax.persistence.Query;

import javax.persistence.spi.PersistenceProvider;

import org.eclipse.persistence.testing.models.performance.*;

import org.eclipse.persistence.internal.helper.Helper;

import org.eclipse.persistence.jpa.JpaEntityManager;

import org.eclipse.persistence.jpa.config.CacheUsage;

import org.eclipse.persistence.testing.framework.*;

/**

 * Memory tests that test for memory leaks.

 */

public class JPAMemoryLeakModel extends TestModel {

    public JPAMemoryLeakModel() {

        setDescription("Memory tests that test for memory leaks.");

    }

    public void addRequiredSystems() {

        addRequiredSystem(new EmployeeSystem());

    }

    public void addTests() {

        addTest(buildReadTest());

        addTest(buildInsertTest());

        addTest(buildUpdateTest());

    }

    /**

     * Create/populate database.

     */

    public void setup() {

        // Populate database (through TopLink).

        for (int j = 0; j < 100; j++) {

            Employee empInsert = new Employee();

            empInsert.setFirstName("Brendan");

            empInsert.setMale();

            empInsert.setLastName("" + j + "");

            empInsert.setSalary(100000);

            EmploymentPeriod employmentPeriod = new EmploymentPeriod();

            java.sql.Date startDate = Helper.dateFromString("1901-12-31");

            java.sql.Date endDate = Helper.dateFromString("1895-01-01");

            employmentPeriod.setEndDate(startDate);

            employmentPeriod.setStartDate(endDate);

            empInsert.setPeriod(employmentPeriod);

            empInsert.setAddress(new Address());

            empInsert.getAddress().setCity("Nepean");

            empInsert.getAddress().setPostalCode("N5J2N5");

            empInsert.getAddress().setProvince("ON");

            empInsert.getAddress().setStreet("1111 Mountain Blvd. Floor 13, suite " + j);

            empInsert.getAddress().setCountry("Canada");

            empInsert.addPhoneNumber(new PhoneNumber("Work Fax", "613", "2255943"));

            empInsert.addPhoneNumber(new PhoneNumber("Home", "613", "2224599"));

            getDatabaseSession().insertObject(empInsert);

        }

        setupProvider();

        getSession().logMessage(getExecutor().getEntityManagerFactory().getClass().toString());

        System.out.println(getExecutor().getEntityManagerFactory().getClass().toString());

    }

    /**

     * Setup the JPA provider.

     */

    public void setupProvider() {

        // Configure provider to be TopLink.

        String providerClass = "org.eclipse.persistence.jpa.PersistenceProvider";

        PersistenceProvider provider = null;

        try {

            provider = (PersistenceProvider)Class.forName(providerClass).newInstance();

        } catch (Exception error) {

            throw new TestProblemException("Failed to create persistence provider.", error);

        }

        Map properties = getPersistenceProperties();

        getExecutor().setEntityManagerFactory(provider.createEntityManagerFactory("performance", properties));

    }

    /**

     * Build the persistence properties.

     */

    public Map getPersistenceProperties() {    

        Map properties = new HashMap();

        properties.put("eclipselink.jdbc.driver", getSession().getLogin().getDriverClassName());

        properties.put("eclipselink.jdbc.url", getSession().getLogin().getConnectionString());

        properties.put("eclipselink.jdbc.user", getSession().getLogin().getUserName());

        properties.put("eclipselink.jdbc.password", getSession().getLogin().getPassword());

        properties.put("eclipselink.logging.level", getSession().getSessionLog().getLevelString());

        properties.put("eclipselink.jdbc.cache-statements", "true");

        return properties;

    }

    /**

     * Test that inserts allow the garbage collection of the persisted objects.

     */

    public TestCase buildInsertTest() {

        MemoryLeakTestCase test = new MemoryLeakTestCase() {

            public void test() {

                EntityManager manager = createEntityManager();

                ((JpaEntityManager)manager).getUnitOfWork().getParent().getIdentityMapAccessor().initializeAllIdentityMaps();

                manager.getTransaction().begin();

                for (int count = 0; count < 500; count++) {

                    Employee employee = new Employee();

                    employee.setFirstName("NewGuy");

                    employee.setLastName("Smith");

                    manager.persist(employee);

                    addWeakReference(employee);

                }

                manager.getTransaction().commit();

                Query query = manager.createQuery("Select e from Employee e");

                query.setHint("eclipselink.return-shared", true);

                query.setHint("eclipselink.cache-usage", CacheUsage.CheckCacheOnly);

                addWeakReferences(query.getResultList());

                addWeakReference(manager);

                addWeakReference(((JpaEntityManager)manager).getUnitOfWork());

                addWeakReference(((JpaEntityManager)manager).getSession());

                manager.close();

            }

            public void reset() {

                getSession().executeNonSelectingSQL("Delete from PHONE where P_NUMBER = '9991111'");

                getSession().executeNonSelectingSQL("Delete from EMPLOYEE where F_NAME = 'NewGuy'");

                getSession().executeNonSelectingSQL("Delete from ADDRESS where STREET = 'Hasting Perf'");

            }

        };

        test.setName("InsertMemoryLeakTest");

        test.setThreshold(100);

        return test;

    }

    /**

     * Test that update allow the garbage collection of the objects.

     */

    public TestCase buildUpdateTest() {

        MemoryLeakTestCase test = new MemoryLeakTestCase() {

            public void test() {

                EntityManager manager = createEntityManager();

                ((JpaEntityManager)manager).getUnitOfWork().getParent().getIdentityMapAccessor().initializeAllIdentityMaps();

                manager.getTransaction().begin();

                Query query = manager.createQuery("Select e from Employee e");

                List<Employee> employees = query.getResultList();

                for (Employee employee : employees) {

                    employee.setFirstName("UpdatedGuy");

                    addWeakReference(employee);

                }

                manager.getTransaction().commit();

                query = manager.createQuery("Select e from Employee e");

                query.setHint("eclipselink.return-shared", true);

                addWeakReferences(query.getResultList());

                addWeakReference(manager);

                addWeakReference(((JpaEntityManager)manager).getUnitOfWork());

                addWeakReference(((JpaEntityManager)manager).getUnitOfWork().getParent());

                manager.close();

            }

        };

        test.setName("UpdateMemoryLeakTest");

        test.setThreshold(100);

        return test;

    }

    /**

     * Test that read allow the garbage collection of the objects.

     */

    public TestCase buildReadTest() {

        MemoryLeakTestCase test = new MemoryLeakTestCase() {

            public void test() {

                EntityManager manager = createEntityManager();

                ((JpaEntityManager)manager).getUnitOfWork().getParent().getIdentityMapAccessor().initializeAllIdentityMaps();

                Query query = manager.createQuery("Select e from Employee e");

                query.setHint("eclipselink.return-shared", true);

                addWeakReferences(query.getResultList());

                addWeakReference(query);

                query = manager.createQuery("Select e from Employee e");

                addWeakReferences(query.getResultList());

                addWeakReference(query);

                addWeakReference(manager);

                addWeakReference(((JpaEntityManager)manager).getUnitOfWork());

                addWeakReference(((JpaEntityManager)manager).getSession());

                manager.close();

            }

        };

        test.setName("ReadMemoryLeakTest");

        test.setThreshold(100);

        return test;

    }

}

/*******************************************************************************

 * Copyright (c) 1998, 2008 Oracle. All rights reserved.

 * This program and the accompanying materials are made available under the 

 * terms of the Eclipse Public License v1.0 and Eclipse Distribution License v. 1.0 

 * which accompanies this distribution. 

 * The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html

 * and the Eclipse Distribution License is available at 

 * http://www.eclipse.org/org/documents/edl-v10.php.

 *

 * Contributors:

 *     Oracle - initial API and implementation from Oracle TopLink

 ******************************************************************************/  

 package org.eclipse.persistence.testing.tests.jpa.memory;

import org.eclipse.persistence.testing.framework.*;

/**

 * Memory tests that compare JPA memory.

 */

public class JPAMemoryTestModel extends TestModel {

    public JPAMemoryTestModel() {

        setDescription("Memory tests that compare JPA memory usage.");

        addTest(new JPAMemoryLeakModel());

        addTest(new JPAMemoryRegressionModel());

    }

}

/*******************************************************************************

 * Copyright (c) 1998, 2008 Oracle. All rights reserved.

 * This program and the accompanying materials are made available under the 

 * terms of the Eclipse Public License v1.0 and Eclipse Distribution License v. 1.0 

 * which accompanies this distribution. 

 * The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html

 * and the Eclipse Distribution License is available at 

 * http://www.eclipse.org/org/documents/edl-v10.php.

 *

 * Contributors:

 *     Oracle - initial API and implementation from Oracle TopLink

 ******************************************************************************/  

 package org.eclipse.persistence.testing.tests.jpa.memory;

import java.util.HashMap;

import java.util.Map;

import javax.persistence.spi.PersistenceProvider;

import org.eclipse.persistence.testing.models.performance.*;

import org.eclipse.persistence.internal.helper.Helper;

import org.eclipse.persistence.testing.framework.*;

/**

 * Memory tests that compare JPA memory usage.

 */

public class JPAMemoryRegressionModel extends TestModel {

    public JPAMemoryRegressionModel() {

        setDescription("Memory tests that compare JPA memory usage.");

    }

    public void addRequiredSystems() {

        addRequiredSystem(new EmployeeSystem());

    }

    public void addTests() {

        //addTest(new ServerSessionMemoryRegressionTest());

        //addTest(new ReadMemoryRegressionTest());

        //addTest(new WriteMemoryRegressionTest());

    }

    /**

     * Create/populate database.

     */

    public void setup() {

        // Populate database (through EclipseLink).

        for (int j = 0; j < 100; j++) {

            Employee empInsert = new Employee();

            empInsert.setFirstName("Brendan");

            empInsert.setMale();

            empInsert.setLastName("" + j + "");

            empInsert.setSalary(100000);

            EmploymentPeriod employmentPeriod = new EmploymentPeriod();

            java.sql.Date startDate = Helper.dateFromString("1901-12-31");

            java.sql.Date endDate = Helper.dateFromString("1895-01-01");

            employmentPeriod.setEndDate(startDate);

            employmentPeriod.setStartDate(endDate);

            empInsert.setPeriod(employmentPeriod);

            empInsert.setAddress(new Address());

            empInsert.getAddress().setCity("Nepean");

            empInsert.getAddress().setPostalCode("N5J2N5");

            empInsert.getAddress().setProvince("ON");

            empInsert.getAddress().setStreet("1111 Mountain Blvd. Floor 13, suite " + j);

            empInsert.getAddress().setCountry("Canada");

            empInsert.addPhoneNumber(new PhoneNumber("Work Fax", "613", "2255943"));

            empInsert.addPhoneNumber(new PhoneNumber("Home", "613", "2224599"));

            getDatabaseSession().insertObject(empInsert);

        }

        setupProvider();

        getSession().logMessage(getExecutor().getEntityManagerFactory().getClass().toString());

        System.out.println(getExecutor().getEntityManagerFactory().getClass().toString());

    }

    /**

     * Setup the JPA provider.

     */

    public void setupProvider() {

        // Configure provider to be EclipseLink.

        String providerClass = "org.eclipse.persistence.jpa.PersistenceProvider";

        PersistenceProvider provider = null;

        try {

            provider = (PersistenceProvider)Class.forName(providerClass).newInstance();

        } catch (Exception error) {

            throw new TestProblemException("Failed to create persistence provider.", error);

        }

        Map properties = getPersistenceProperties();

        getExecutor().setEntityManagerFactory(provider.createEntityManagerFactory("performance", properties));

    }

    /**

     * Build the persistence properties.

     */

    public Map getPersistenceProperties() {    

        Map properties = new HashMap();

        properties.put("eclipselink.jdbc.driver", getSession().getLogin().getDriverClassName());

        properties.put("eclipselink.jdbc.url", getSession().getLogin().getConnectionString());

        properties.put("eclipselink.jdbc.user", getSession().getLogin().getUserName());

        properties.put("eclipselink.jdbc.password", getSession().getLogin().getPassword());

        properties.put("eclipselink.logging.level", getSession().getSessionLog().getLevelString());

        properties.put("eclipselink.jdbc.cache-statements", "true");

        return properties;

    }    

}

/*******************************************************************************

 * Copyright (c) 1998, 2008 Oracle. All rights reserved.

 * This program and the accompanying materials are made available under the 

 * terms of the Eclipse Public License v1.0 and Eclipse Distribution License v. 1.0 

 * which accompanies this distribution. 

 * The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html

 * and the Eclipse Distribution License is available at 

 * http://www.eclipse.org/org/documents/edl-v10.php.

 *

 * Contributors:

 *     Oracle - initial API and implementation from Oracle TopLink

 ******************************************************************************/  

 package org.eclipse.persistence.testing.tests.jpa.memory;

import java.util.List;

import java.util.HashMap;

import java.util.Map;

import javax.persistence.EntityManager;

import javax.persistence.Query;

import javax.persistence.spi.PersistenceProvider;

import org.eclipse.persistence.testing.models.performance.*;

import org.eclipse.persistence.internal.helper.Helper;

import org.eclipse.persistence.jpa.JpaEntityManager;

import org.eclipse.persistence.jpa.config.CacheUsage;

import org.eclipse.persistence.testing.framework.*;

/**

 * Memory tests that test for memory leaks.

 */

public class JPAMemoryLeakModel extends TestModel {

    public JPAMemoryLeakModel() {

        setDescription("Memory tests that test for memory leaks.");

    }

    public void addRequiredSystems() {

        addRequiredSystem(new EmployeeSystem());

    }

    public void addTests() {

        addTest(buildReadTest());

        addTest(buildInsertTest());

        addTest(buildUpdateTest());

    }

    /**

     * Create/populate database.

     */

    public void setup() {

        // Populate database (through TopLink).

        for (int j = 0; j < 100; j++) {

            Employee empInsert = new Employee();

            empInsert.setFirstName("Brendan");

            empInsert.setMale();

            empInsert.setLastName("" + j + "");

            empInsert.setSalary(100000);

            EmploymentPeriod employmentPeriod = new EmploymentPeriod();

            java.sql.Date startDate = Helper.dateFromString("1901-12-31");

            java.sql.Date endDate = Helper.dateFromString("1895-01-01");

            employmentPeriod.setEndDate(startDate);

            employmentPeriod.setStartDate(endDate);

            empInsert.setPeriod(employmentPeriod);

            empInsert.setAddress(new Address());

            empInsert.getAddress().setCity("Nepean");

            empInsert.getAddress().setPostalCode("N5J2N5");

            empInsert.getAddress().setProvince("ON");

            empInsert.getAddress().setStreet("1111 Mountain Blvd. Floor 13, suite " + j);

            empInsert.getAddress().setCountry("Canada");

            empInsert.addPhoneNumber(new PhoneNumber("Work Fax", "613", "2255943"));

            empInsert.addPhoneNumber(new PhoneNumber("Home", "613", "2224599"));

            getDatabaseSession().insertObject(empInsert);

        }

        setupProvider();

        getSession().logMessage(getExecutor().getEntityManagerFactory().getClass().toString());

        System.out.println(getExecutor().getEntityManagerFactory().getClass().toString());

    }

    /**

     * Setup the JPA provider.

     */

    public void setupProvider() {

        // Configure provider to be TopLink.

        String providerClass = "org.eclipse.persistence.jpa.PersistenceProvider";

        PersistenceProvider provider = null;

        try {

            provider = (PersistenceProvider)Class.forName(providerClass).newInstance();

        } catch (Exception error) {

            throw new TestProblemException("Failed to create persistence provider.", error);

        }

        Map properties = getPersistenceProperties();

        getExecutor().setEntityManagerFactory(provider.createEntityManagerFactory("performance", properties));

    }

    /**

     * Build the persistence properties.

     */

    public Map getPersistenceProperties() {    

        Map properties = new HashMap();

        properties.put("eclipselink.jdbc.driver", getSession().getLogin().getDriverClassName());

        properties.put("eclipselink.jdbc.url", getSession().getLogin().getConnectionString());

        properties.put("eclipselink.jdbc.user", getSession().getLogin().getUserName());

        properties.put("eclipselink.jdbc.password", getSession().getLogin().getPassword());

        properties.put("eclipselink.logging.level", getSession().getSessionLog().getLevelString());

        properties.put("eclipselink.jdbc.cache-statements", "true");

        return properties;

    }

    /**

     * Test that inserts allow the garbage collection of the persisted objects.

     */

    public TestCase buildInsertTest() {

        MemoryLeakTestCase test = new MemoryLeakTestCase() {

            public void test() {

                EntityManager manager = createEntityManager();

                ((JpaEntityManager)manager).getUnitOfWork().getParent().getIdentityMapAccessor().initializeAllIdentityMaps();

                manager.getTransaction().begin();

                for (int count = 0; count < 500; count++) {

                    Employee employee = new Employee();

                    employee.setFirstName("NewGuy");

                    employee.setLastName("Smith");

                    manager.persist(employee);

                    addWeakReference(employee);

                }

                manager.getTransaction().commit();

                Query query = manager.createQuery("Select e from Employee e");

                query.setHint("eclipselink.return-shared", true);

                query.setHint("eclipselink.cache-usage", CacheUsage.CheckCacheOnly);

                addWeakReferences(query.getResultList());

                addWeakReference(manager);

                addWeakReference(((JpaEntityManager)manager).getUnitOfWork());

                addWeakReference(((JpaEntityManager)manager).getSession());

                manager.close();

            }

            public void reset() {

                getSession().executeNonSelectingSQL("Delete from PHONE where P_NUMBER = '9991111'");

                getSession().executeNonSelectingSQL("Delete from EMPLOYEE where F_NAME = 'NewGuy'");

                getSession().executeNonSelectingSQL("Delete from ADDRESS where STREET = 'Hasting Perf'");

            }

        };

        test.setName("InsertMemoryLeakTest");

        test.setThreshold(100);

        return test;

    }

    /**

     * Test that update allow the garbage collection of the objects.

     */

    public TestCase buildUpdateTest() {

        MemoryLeakTestCase test = new MemoryLeakTestCase() {

            public void test() {

                EntityManager manager = createEntityManager();

                ((JpaEntityManager)manager).getUnitOfWork().getParent().getIdentityMapAccessor().initializeAllIdentityMaps();

                manager.getTransaction().begin();

                Query query = manager.createQuery("Select e from Employee e");

                List<Employee> employees = query.getResultList();

                for (Employee employee : employees) {

                    employee.setFirstName("UpdatedGuy");

                    addWeakReference(employee);

                }

                manager.getTransaction().commit();

                query = manager.createQuery("Select e from Employee e");

                query.setHint("eclipselink.return-shared", true);

                addWeakReferences(query.getResultList());

                addWeakReference(manager);

                addWeakReference(((JpaEntityManager)manager).getUnitOfWork());

                addWeakReference(((JpaEntityManager)manager).getUnitOfWork().getParent());

                manager.close();

            }

        };

        test.setName("UpdateMemoryLeakTest");

        test.setThreshold(100);

        return test;

    }

    /**

     * Test that read allow the garbage collection of the objects.

     */

    public TestCase buildReadTest() {

        MemoryLeakTestCase test = new MemoryLeakTestCase() {

            public void test() {

                EntityManager manager = createEntityManager();

                ((JpaEntityManager)manager).getUnitOfWork().getParent().getIdentityMapAccessor().initializeAllIdentityMaps();

                Query query = manager.createQuery("Select e from Employee e");

                query.setHint("eclipselink.return-shared", true);

                addWeakReferences(query.getResultList());

                addWeakReference(query);

                query = manager.createQuery("Select e from Employee e");

                addWeakReferences(query.getResultList());

                addWeakReference(query);

                addWeakReference(manager);

                addWeakReference(((JpaEntityManager)manager).getUnitOfWork());

                addWeakReference(((JpaEntityManager)manager).getSession());

                manager.close();

            }

        };

        test.setName("ReadMemoryLeakTest");

        test.setThreshold(100);

        return test;

    }

}

/*******************************************************************************

 * Copyright (c) 1998, 2008 Oracle. All rights reserved.

 * This program and the accompanying materials are made available under the 

 * terms of the Eclipse Public License v1.0 and Eclipse Distribution License v. 1.0 

 * which accompanies this distribution. 

 * The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html

 * and the Eclipse Distribution License is available at 

 * http://www.eclipse.org/org/documents/edl-v10.php.

 *

 * Contributors:

 *     Oracle - initial API and implementation from Oracle TopLink

 ******************************************************************************/  

 package org.eclipse.persistence.testing.tests.jpa.memory;

import java.util.HashMap;

import java.util.Map;

import javax.persistence.spi.PersistenceProvider;

import org.eclipse.persistence.testing.models.performance.*;

import org.eclipse.persistence.internal.helper.Helper;

import org.eclipse.persistence.testing.framework.*;

/**

 * Memory tests that compare JPA memory usage.

 */

public class JPAMemoryRegressionModel extends TestModel {

    public JPAMemoryRegressionModel() {

        setDescription("Memory tests that compare JPA memory usage.");

    }

    public void addRequiredSystems() {

        addRequiredSystem(new EmployeeSystem());

    }

    public void addTests() {

        //addTest(new ServerSessionMemoryRegressionTest());

        //addTest(new ReadMemoryRegressionTest());

        //addTest(new WriteMemoryRegressionTest());

    }

    /**

     * Create/populate database.

     */

    public void setup() {

        // Populate database (through EclipseLink).

        for (int j = 0; j < 100; j++) {

            Employee empInsert = new Employee();

            empInsert.setFirstName("Brendan");

            empInsert.setMale();

            empInsert.setLastName("" + j + "");

            empInsert.setSalary(100000);

            EmploymentPeriod employmentPeriod = new EmploymentPeriod();

            java.sql.Date startDate = Helper.dateFromString("1901-12-31");

            java.sql.Date endDate = Helper.dateFromString("1895-01-01");

            employmentPeriod.setEndDate(startDate);

            employmentPeriod.setStartDate(endDate);

            empInsert.setPeriod(employmentPeriod);

            empInsert.setAddress(new Address());

            empInsert.getAddress().setCity("Nepean");

            empInsert.getAddress().setPostalCode("N5J2N5");

            empInsert.getAddress().setProvince("ON");

            empInsert.getAddress().setStreet("1111 Mountain Blvd. Floor 13, suite " + j);

            empInsert.getAddress().setCountry("Canada");

            empInsert.addPhoneNumber(new PhoneNumber("Work Fax", "613", "2255943"));

            empInsert.addPhoneNumber(new PhoneNumber("Home", "613", "2224599"));

            getDatabaseSession().insertObject(empInsert);

        }

        setupProvider();

        getSession().logMessage(getExecutor().getEntityManagerFactory().getClass().toString());

        System.out.println(getExecutor().getEntityManagerFactory().getClass().toString());

    }

    /**

     * Setup the JPA provider.

     */

    public void setupProvider() {

        // Configure provider to be EclipseLink.

        String providerClass = "org.eclipse.persistence.jpa.PersistenceProvider";

        PersistenceProvider provider = null;

        try {

            provider = (PersistenceProvider)Class.forName(providerClass).newInstance();

        } catch (Exception error) {

            throw new TestProblemException("Failed to create persistence provider.", error);

        }

        Map properties = getPersistenceProperties();

        getExecutor().setEntityManagerFactory(provider.createEntityManagerFactory("performance", properties));

    }

    /**

     * Build the persistence properties.

     */

    public Map getPersistenceProperties() {    

        Map properties = new HashMap();

        properties.put("eclipselink.jdbc.driver", getSession().getLogin().getDriverClassName());

        properties.put("eclipselink.jdbc.url", getSession().getLogin().getConnectionString());

        properties.put("eclipselink.jdbc.user", getSession().getLogin().getUserName());

        properties.put("eclipselink.jdbc.password", getSession().getLogin().getPassword());

        properties.put("eclipselink.logging.level", getSession().getSessionLog().getLevelString());

        properties.put("eclipselink.jdbc.cache-statements", "true");

        return properties;

    }    

}

/*******************************************************************************

 * Copyright (c) 1998, 2008 Oracle. All rights reserved.

 * This program and the accompanying materials are made available under the 

 * terms of the Eclipse Public License v1.0 and Eclipse Distribution License v. 1.0 

 * which accompanies this distribution. 

 * The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html

 * and the Eclipse Distribution License is available at 

 * http://www.eclipse.org/org/documents/edl-v10.php.

 *

 * Contributors:

 *     Oracle - initial API and implementation from Oracle TopLink

 ******************************************************************************/  

 package org.eclipse.persistence.testing.tests.jpa.memory;

import org.eclipse.persistence.testing.framework.*;

/**

 * Memory tests that compare JPA memory.

 */

public class JPAMemoryTestModel extends TestModel {

    public JPAMemoryTestModel() {

        setDescription("Memory tests that compare JPA memory usage.");

        addTest(new JPAMemoryLeakModel());

        addTest(new JPAMemoryRegressionModel());

    }

}

        addTest(new TopLinkJPAPerformanceRegressionModel());

<listEntry value="&lt;?xml version=&quot;1.0&quot; encoding=&quot;UTF-8&quot; standalone=&quot;no&quot;?&gt;&#13;&#10;&lt;runtimeClasspathEntry internalArchive=&quot;/eclipselink.jpa.test/eclipselink-xml-extended-model.jar&quot; path=&quot;3&quot; type=&quot;2&quot;/&gt;&#13;&#10;"/>

<listEntry value="&lt;?xml version=&quot;1.0&quot; encoding=&quot;UTF-8&quot; standalone=&quot;no&quot;?&gt;&#13;&#10;&lt;runtimeClasspathEntry internalArchive=&quot;/eclipselink.jpa.test/jpa-performance.jar&quot; path=&quot;3&quot; type=&quot;2&quot;/&gt;&#13;&#10;"/>