package org.apache.ojb.ejb.pb;
/* Copyright 2004-2005 The Apache Software Foundation
*
* Licensed 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.
*/
import javax.ejb.EJBHome;
import javax.naming.Context;
import javax.rmi.PortableRemoteObject;
import java.rmi.RemoteException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import junit.framework.TestCase;
import org.apache.ojb.ejb.ArticleVO;
import org.apache.ojb.ejb.ContextHelper;
import org.apache.ojb.ejb.VOHelper;
/**
* Test client using the {@link org.apache.ojb.ejb.pb.PBSessionBean}.
*
* @author <a href="mailto:armin@codeAuLait.de">Armin Waibel</a>
* @version $Id: PBClient.java,v 1.5.2.2 2005/12/21 22:21:38 tomdz Exp $
*/
public class PBClient extends TestCase
{
PBSessionRemote bean;
static int loops = 500;
public PBClient(String s)
{
super(s);
}
public PBClient()
{
super(PBClient.class.getName());
}
public static void main(String[] args)
{
loops = args.length > 0 ? new Integer(args[0]).intValue(): 500;
junit.textui.TestRunner.main(new String[]{PBClient.class.getName()});
}
protected void setUp() throws Exception
{
super.setUp();
init();
}
protected void init()
{
Context ctx = ContextHelper.getContext();
try
{
Object object = PortableRemoteObject.narrow(ctx.lookup(PBSessionHome.JNDI_NAME), EJBHome.class);
bean = ((PBSessionHome) object).create();
}
catch (Exception e)
{
e.printStackTrace();
}
}
public void testInsertDelete() throws RemoteException
{
int articlesBefore = bean.getArticleCount();
int personsBefore = bean.getPersonCount();
List articles = VOHelper.createNewArticleList(10);
List persons = VOHelper.createNewPersonList(5);
articles = bean.storeObjects(articles);
persons = bean.storeObjects(persons);
int articlesAfterStore = bean.getArticleCount();
int personsAfterStore = bean.getPersonCount();
assertEquals("Storing of articles failed", articlesBefore + 10, articlesAfterStore);
assertEquals("Storing of persons faile", personsBefore + 5, personsAfterStore);
bean.deleteObjects(articles);
bean.deleteObjects(persons);
int articlesAfterDelete = bean.getArticleCount();
int personsAfterDelete = bean.getPersonCount();
assertEquals("Deleting of articles failed", articlesAfterStore - 10, articlesAfterDelete);
assertEquals("Deleting of persons failed", personsAfterStore - 5, personsAfterDelete);
}
public void testServerSideMethods() throws RemoteException
{
boolean result = bean.allInOne(VOHelper.createNewArticleList(10), VOHelper.createNewPersonList(5));
assertTrue("Something happened on sever side test method - 'allInOne(...)'", result);
}
public void testStress() throws Exception
{
System.out.println("## PB-api testStress");
System.out.println("Stress test will be done with " + loops + " loops");
System.out.println("# Store #");
for (int i = 0; i < loops; i++)
{
bean.storeObjects(VOHelper.createNewArticleList(1));
if(i%10==0)System.out.print(".");
if(i%400==0)System.out.println();
}
Collection col = bean.getAllObjects(ArticleVO.class);
int i =0;
System.out.println("\n# Delete #");
for (Iterator iterator = col.iterator(); iterator.hasNext();)
{
ArticleVO article = (ArticleVO) iterator.next();
List del = new ArrayList();
del.add(article);
bean.deleteObjects(del);
if(++i%10==0)System.out.print(".");
if(i%400==0)System.out.println();
}
System.out.println("");
System.out.println("## PB-api testStress END ##");
}
public void testStress2() throws Exception
{
System.out.println("## PB-api testStress2");
System.out.println("# Store " + loops + " objects");
List newObjects = null;
for (int i = 0; i < loops; i++)
{
newObjects = VOHelper.createNewArticleList(loops);
}
bean.storeObjects(newObjects);
List lookupObjects = new ArrayList(bean.getAllObjects(ArticleVO.class));
System.out.println("# Delete " + loops + " objects");
bean.deleteObjects(lookupObjects);
System.out.println("## PB-api testStress2 END ##");
}
}