package org.jfox.ejb.examples.stateful;

import java.io.ObjectOutputStream;
import java.io.FileOutputStream;
import java.io.ObjectInputStream;
import java.io.FileInputStream;
import javax.naming.Context;
import javax.naming.InitialContext;
import javax.ejb.EJBMetaData;
import javax.ejb.HomeHandle;
import javax.ejb.Handle;

import junit.framework.TestCase;

/**
 * 
 * @author <a href="mailto:young_yy@hotmail.com">Young Yang</a>
 */

public class WorldTest extends TestCase {
  private static WorldHome home = null;
  private static World world = null;

  static {
    try {
      Context ctx = new InitialContext();
      Object obj = ctx.lookup("" + WorldHome.class.getName().replace('.','/'));
      home = (WorldHome)javax.rmi.PortableRemoteObject.narrow(obj,WorldHome.class);
//      ((ExtendedEJBHome)home).useProtocol("soap");
      world = home.create();
    }
    catch(Exception e){
      e.printStackTrace();
    }
  }

  protected void setUp() throws Exception {
  }

  public void testToString() {
    System.out.println(home.toString());
    System.out.println(world);
  }

  public void testGetEJBMetaData() {
    try {
      System.out.print("test getEJBMetaData()...");
      EJBMetaData meta = home.getEJBMetaData();
      System.out.println("EJBMetaData: " + meta);
    }
    catch(Exception e){
      fail(e.toString());
    }
  }

  public void testGetWord() {
    try {
      assertTrue(world.getWord().equals("Hello,World"));
    }
    catch(Exception e){
      e.printStackTrace();
      fail(e.toString());
    }
  }

  public void testSetWord() {
    try {
      System.out.print("test setWord()...");
      world.setWord("Hello,All");
      // for test passivate & active
//      Thread.sleep(70000L);
      assertTrue(world.getWord().equals("Hello,All"));
      System.out.println("OK!");
    }
    catch(Exception e){
      e.printStackTrace();
      fail(e.toString());
    }

  }

  public void testHomeHandle() {
    try {
      System.out.print("test getHomeHandle()...");
      HomeHandle homeHandle = home.getHomeHandle();
      writeHandle(homeHandle);
      homeHandle = (HomeHandle)readHandle();
      home = (WorldHome)homeHandle.getEJBHome();
      world = home.create();
      assertTrue(world.getWord().equals("Hello,World"));
      System.out.println("OK!");
    }
    catch(Exception e){
      fail(e.toString());
    }
  }

  public void testHandle() {
    try {
      System.out.print("test getHandle()...");
      Handle handle = world.getHandle();
      writeHandle(handle);
      handle = (Handle)readHandle();
      world = (World)handle.getEJBObject();
      assertTrue(world.getWord().equals("Hello,World"));
      System.out.println("OK!");
    }
    catch(Exception e){
      fail(e.toString());
    }
  }

  private void writeHandle(Object obj) throws Exception {
    ObjectOutputStream out = new ObjectOutputStream(new FileOutputStream(this.getClass() + ".ser"));
    out.writeObject(obj);
    out.close();
  }

  private Object readHandle() throws Exception {
    ObjectInputStream in = new ObjectInputStream(new FileInputStream(this.getClass() + ".ser"));
    Object handle = in.readObject();
    in.close();
    return handle;
  }

  public static void main(String[] args) {
    junit.textui.TestRunner.run(WorldTest.class);
  }

}