/*

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code 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

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

/* @test

 * @summary it is new version of old test which was

 *          /src/share/test/serialization/psiotest.java

 *          Test pickling and unpickling an object with derived classes

 *          and using a read special to serialize the "middle" class,

 *          which raises NotSerializableException inside writeObject()

 *          and readObject() methods.

 */

import java.io.*;

public class CheckForException {

    public static void main (String argv[]) {

        System.err.println("\nRegression test of " +

                           "serialization/deserialization of " +

                           "complex objects which raise " +

                           "NotSerializableException inside " +

                           "writeObject() and readObject() methods.\n");

        FileInputStream istream = null;

        try {

            FileOutputStream ostream = new FileOutputStream("psiotest3.tmp");

            ObjectOutputStream p = new ObjectOutputStream(ostream);

            /* Catch the expected exception and

             * complain if it does not occur.

             */

            TryPickleClass npc = new TryPickleClass();

            NotSerializableException we = null;

            try {

                // Two objects of the same class that are used

                // in cleanup test below

                p.writeObject("test");

                p.writeObject("test2");

                p.writeObject(npc);

            } catch (NotSerializableException e) {

                we = e;

            }

            if (we == null) {

                System.err.println("\nTEST FAILED: Write of NoPickleClass " +

                    "should have raised an exception");

                throw new Error();

            }

            p.flush();

            ostream.close();

            istream = new FileInputStream("psiotest3.tmp");

            ObjectInputStream q = new ObjectInputStream(istream);

            /* Catch the expected exception and

             * complain if it does not occur.

             */

            TryPickleClass npc_u;

            NotSerializableException re = null;

            try {

                // Read the two objects, neither has a cleanup method

                q.readObject();

                q.readObject();

                npc_u = (TryPickleClass)q.readObject();

            } catch (NotSerializableException e) {

                re = e;

            }

            if (re == null) {

                System.err.println("\nTEST FAILED: Read of NoPickleClass " +

                   "should have raised an exception");

                throw new Error();

            }

            istream.close();

            System.err.println("\nTEST PASSED");

        } catch (Exception e) {

            System.err.print("TEST FAILED: ");

            e.printStackTrace();

            throw new Error();

        }

    }

}

class PickleClass implements java.io.Serializable {

    int ii = 17;

    transient int tmp[];

    private void writeObject(ObjectOutputStream pw) throws IOException {

        pw.writeUTF("PickleClass");

        pw.writeInt(ii);

    }

    private void readObject(ObjectInputStream pr) throws IOException {

        tmp = new int[32];

        pr.readUTF();

        ii = pr.readInt();

    }

    private void readObjectCleanup(ObjectInputStream pr) {

        System.err.println("\nPickleClass cleanup correctly called on abort.");

        if (tmp != null) {

            tmp = null;

        }

    }

}

class NoPickleClass extends PickleClass {

    private void writeObject(ObjectOutputStream pw)

        throws NotSerializableException

    {

        throw new NotSerializableException("NoPickleClass");

    }

    private void readObject(ObjectInputStream pr)

            throws NotSerializableException

    {

            throw new NotSerializableException("NoPickleClass");

    }

}

class TryPickleClass  extends NoPickleClass {

    int i = 7;

    transient int tmp[];

    private void writeObject(ObjectOutputStream pw) throws IOException {

            pw.writeInt(i);

    }

    private void readObject(ObjectInputStream pr) throws IOException {

            tmp = new int[32];

            i = pr.readInt();

    }

    private void readObjectCleanup(ObjectInputStream pr) {

            System.err.println("\nCleanup called on abort");

            if (tmp != null) {

                tmp = null;

        }

    }

}