LazyDeserializeTest.java

/*
 * Copyright 2011 MyBatis.org.
 *
 * 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.
 */
package org.apache.ibatis.submitted.lazy_deserialize;

import java.io.PrintWriter;
import java.io.Reader;
import java.sql.Connection;
import java.sql.DriverManager;
import org.apache.ibatis.io.Resources;
import org.apache.ibatis.jdbc.ScriptRunner;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;
import org.junit.BeforeClass;
import org.junit.Test;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import org.apache.ibatis.session.Configuration;
import static org.junit.Assert.*;

/**
 *
 * @author Franta Mejta
 * @date 2011-04-06T10:58:55+0200
 */
public final class LazyDeserializeTest {

    private static final int FOO_ID = 1;
    private static final int BAR_ID = 10;
    private static SqlSessionFactory factory;

    public static Configuration getConfiguration() {
        return factory.getConfiguration();
    }

    @BeforeClass
    public static void setupClass() throws Exception {
        Connection conn = null;

        try {
            Class.forName("org.hsqldb.jdbcDriver");
            conn = DriverManager.getConnection("jdbc:hsqldb:mem:lazy_deserialize", "sa", "");

            Reader reader = Resources.getResourceAsReader("org/apache/ibatis/submitted/lazy_deserialize/CreateDB.sql");

            ScriptRunner runner = new ScriptRunner(conn);
            runner.setLogWriter(null);
            runner.setErrorLogWriter(new PrintWriter(System.err));
            runner.runScript(reader);
            conn.commit();
            reader.close();

            reader = Resources.getResourceAsReader("org/apache/ibatis/submitted/lazy_deserialize/ibatisConfig.xml");
            factory = new SqlSessionFactoryBuilder().build(reader);
            reader.close();
        } finally {
            if (conn != null) {
                conn.close();
            }
        }
    }

    @Test
    public void testLoadLazyDeserialize() throws Exception {
        final SqlSession session = factory.openSession();
        try {
            final Mapper mapper = session.getMapper(Mapper.class);
            final LazyObjectFoo foo = mapper.loadFoo(FOO_ID);

            final byte[] serializedFoo = this.serializeFoo(foo);
            final LazyObjectFoo deserializedFoo = this.deserializeFoo(serializedFoo);

            assertNotNull(deserializedFoo);
            assertEquals(Integer.valueOf(FOO_ID), deserializedFoo.getId());
            assertNotNull(deserializedFoo.getLazyObjectBar());
            assertEquals(Integer.valueOf(BAR_ID), deserializedFoo.getLazyObjectBar().getId());
        } finally {
            session.close();
        }
    }

    private byte[] serializeFoo(final LazyObjectFoo foo) throws Exception {
        final ByteArrayOutputStream bos = new ByteArrayOutputStream();
        final ObjectOutputStream oos = new ObjectOutputStream(bos);
        oos.writeObject(foo);
        oos.close();

        return bos.toByteArray();
    }

    private LazyObjectFoo deserializeFoo(final byte[] serializedFoo) throws Exception {
        final ByteArrayInputStream bis = new ByteArrayInputStream(serializedFoo);
        final ObjectInputStream ios = new ObjectInputStream(bis);
        final LazyObjectFoo foo = LazyObjectFoo.class.cast(ios.readObject());
        ios.close();
        return foo;
    }

}