/*************************************************************************
/* ObjectOutputStreamTest.java -- Tests ObjectOutputStream class
/*
/* Copyright (c) 1998 by Geoffrey C. Berry (gcb@cs.duke.edu)
/*
/* This program is free software; you can redistribute it and/or modify
/* it under the terms of the GNU General Public License as published 
/* by the Free Software Foundation, version 2. (see COPYING)
/*
/* This program is distributed in the hope that it will be useful, but
/* WITHOUT ANY WARRANTY; without even the implied warranty of
/* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
/* GNU General Public License for more details.
/*
/* You should have received a copy of the GNU General Public License
/* along with this program; if not, write to the Free Software Foundation
/* Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307 USA
/*************************************************************************/

import java.io.ByteArrayOutputStream;
import java.io.Externalizable;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectOutput;
import java.io.ObjectOutputStream;
import java.io.ObjectStreamException;
import java.io.Serializable;

public class ObjectOutputStreamTest
{
  public static void error()
  {
    System.out.print( "ERROR: " );
  }
  
  public static void pass()
  {
    System.out.print( "PASSED: " );
  }
  
  public static void fail()
  {
    System.out.print( "FAILED: " );
  }

  public static void pass( boolean exp_pass )
  {
    if( exp_pass )
      pass();
    else
      System.out.print( "XPASSED: " );
  }
  
  public static void fail( boolean exp_pass )
  {
    if( exp_pass )
      fail();
    else
      System.out.print( "XFAIL: " );
  }

  public static void testSerial( Object obj, String filename )
  {
    if( writeMode )
    {
      try
      {
	ObjectOutputStream oos =
	  new ObjectOutputStream( new FileOutputStream( filename ) );
	oos.writeObject( obj );
	oos.close();
      }
      catch( ObjectStreamException e )
      {}
      catch( IOException ioe )
      {
	ioe.printStackTrace();
      }
    }
    else
    { 
      ByteArrayOutputStream bytes = new ByteArrayOutputStream();

      try
      {
	ObjectOutputStream oos = new ObjectOutputStream( bytes );
	oos.writeObject( obj );
	oos.close();
      }
      catch( ObjectStreamException e )
      {}
      catch( IOException ioe )
      {
	fail();
	return;
      }
    
      byte[] jcl_bytes = bytes.toByteArray();
      int data;
      
      FileInputStream jdk_file;
      try
      {
	jdk_file = new FileInputStream( filename );

	for( int i=0; i < jcl_bytes.length; i++ )
	{
	  data = jdk_file.read();
	  
	  if( data == -1 )
	  {
	    fail();
	    return;
	  }
	  
	  if( (byte)data != jcl_bytes[i] )
	  {
	    fail();
	    return;
	  }
	}

	if( jdk_file.read() != -1 )
	{
	  fail();
	  return;
	}
      }
      catch( IOException e )
      {
	error();
	return;
      }
    
      pass();
    }
  }
  
  
  public static void main( String[] args )
  {
    writeMode = (args.length != 0);

    testSerial( new OOSNotSerial(), "notserial.data" );
    System.out.println( "Non-serializable class" );

    testSerial( new OOSBadField( 1, 2, new OOSNotSerial() ),
		"notserialfield.data" );
    System.out.println( "Object with non-serializable field" );

    testSerial( new OOSCallDefault( 1, 3.14, "test" ),
		"calldefault.data" );
    System.out.println( "Object calling defaultWriteObject()" );

    testSerial( new OOSNoCallDefault( 17, "no\ndefault", false ),
		"nocalldefault.data" );
    System.out.println( "Object not calling defaultWriteObject()" );

    testSerial( new OOSExtern( -1, "", true ), "external.data" );
    System.out.println( "Externalizable class" );

    testSerial( new HairyGraph(), "graph.data" );
    System.out.println( "Graph of objects with circular references" );
  }

  
  public static boolean writeMode;
}


class OOSNotSerial {}

class OOSBadField implements Serializable
{
  int x;
  int y;
  OOSNotSerial o;

  OOSBadField( int X, int Y, OOSNotSerial O )
  {
    x = X;
    y = Y;
    o = O;
  }
}


class OOSCallDefault implements Serializable
{
  int x;
  double y;
  transient String s;

  OOSCallDefault( int X, double Y, String S )
  {
    x = X;
    y = Y;
    s = S;
  }

  private void writeObject( ObjectOutputStream oos ) throws IOException
  {
    oos.writeObject( s );
    oos.defaultWriteObject();
    oos.writeObject( s );
  }
}

class OOSNoCallDefault implements Serializable
{
  int x;
  String s;
  boolean b;
  
  OOSNoCallDefault( int X, String S, boolean B )
  {
    x = X;
    s = S;
    b = B;
  }

  private void writeObject( ObjectOutputStream oos ) throws IOException
  {
    oos.writeInt( x );
    oos.writeObject( s );
    oos.writeBoolean( b );
  }
}

class OOSExtern extends OOSNoCallDefault implements Externalizable
{
  OOSExtern( int X, String S, boolean B )
  {
    super( X, S, B );
  }

  public void writeExternal( ObjectOutput oo ) throws IOException
  {
    oo.writeInt( super.x );
    oo.writeObject( super.s );
    oo.writeBoolean( super.b );
  }
  
  public void readExternal( ObjectInput oi ) {}
}

class GraphNode implements Serializable
{
  GraphNode a;
  GraphNode b;
  GraphNode c;
  GraphNode d;
}


class HairyGraph implements Serializable
{
  GraphNode A;
  GraphNode B;
  GraphNode C;
  GraphNode D;
  
  HairyGraph()
  {
    A = new GraphNode();
    B = new GraphNode();
    C = new GraphNode();
    D = new GraphNode();

    A.a = B;
    A.b = C;
    A.c = D;
    A.d = A;
    
    B.a = C;
    B.b = D;
    B.c = A;
    B.d = B;

    C.a = D;
    C.b = A;
    C.c = B;
    C.d = C;
    
    D.a = A;
    D.b = B;
    D.c = C;
    D.d = D;
  }
}