Test_031_SchemaReading.java

/*
 * Copyright 2009-2016 Tilmann Zaeschke. All rights reserved.
 * 
 * This file is part of ZooDB.
 * 
 * ZooDB 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, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * ZooDB 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 ZooDB.  If not, see <http://www.gnu.org/licenses/>.
 * 
 * See the README and COPYING files for further information. 
 */
package org.zoodb.test.jdo;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.fail;

import java.util.ArrayList;

import javax.jdo.PersistenceManager;

import org.junit.After;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import org.zoodb.jdo.ZooJdoHelper;
import org.zoodb.schema.ZooClass;
import org.zoodb.schema.ZooHandle;
import org.zoodb.test.testutil.TestTools;

public class Test_031_SchemaReading {

	@BeforeClass
	public static void setUp() {
		TestTools.createDb();
		PersistenceManager pm = TestTools.openPM();
		pm.currentTransaction().begin();
		ZooJdoHelper.schema(pm).addClass(TestClass.class);
		pm.currentTransaction().commit();
		TestTools.closePM();
	}
	
	@AfterClass
	public static void tearDown() {
	    TestTools.closePM();
		TestTools.removeDb();
	}

	@After
	public void after() {
		TestTools.closePM();
	}
	
	@Test
	public void testSchemaZooHandle() {
		PersistenceManager pm0 = TestTools.openPM();
		pm0.currentTransaction().begin();
		
		TestClass t1 = new TestClass();
		t1.setData(1234567, true, 'x', (byte)126, (short)32000, 12345678901L, "haha", 
				new byte[]{1, 2, 3}, -1.1f, 33.3);
		TestClass t2 = new TestClass();
		t2.setData(-1234567, true, 'y', (byte)-126, (short)-32000, -12345678901L, "hihi", 
				new byte[]{-1, -2, -3}, 11.11f, -3.3);
		
		pm0.makePersistent(t1);
		pm0.makePersistent(t2);
		
		long oid1 = (Long) pm0.getObjectId(t1);
		long oid2 = (Long) pm0.getObjectId(t2);
		
		pm0.currentTransaction().commit();
		TestTools.closePM();
		
		PersistenceManager pm = TestTools.openPM();
		pm.currentTransaction().begin();
		
		ZooClass s01 = ZooJdoHelper.schema(pm).getClass(TestClass.class.getName());
		assertNotNull(s01);

		//closed pm
		try {
			ZooJdoHelper.schema(pm0).getHandle(oid1);
			fail();
		} catch (IllegalStateException e) {
			//good!
		}
		
		//wrong oid
		assertNull(ZooJdoHelper.schema(pm).getHandle(12345678));
		
		ZooHandle hdl1 = ZooJdoHelper.schema(pm).getHandle(oid1);
		ZooHandle hdl2 = ZooJdoHelper.schema(pm).getHandle(oid2);
		assertNotNull(hdl1);
		assertNotNull(hdl2);
		
		assertEquals((Long)oid1, (Long)hdl1.getOid());
		assertEquals(s01, hdl1.getType());
		
		TestTools.closePM();
	}
	

	@Test
	public void testSchemaZooHandleReader() {
		PersistenceManager pm0 = TestTools.openPM();
		pm0.currentTransaction().begin();
		
		TestClass t1 = new TestClass();
		t1.setData(1234567, true, 'x', (byte)126, (short)32000, 12345678901L, "haha", 
				new byte[]{1, 2, 3}, -1.1f, 33.3);
		TestClass t2 = new TestClass();
		t2.setData(-1234567, true, 'y', (byte)-126, (short)-32000, -12345678901L, "hihi", 
				new byte[]{-1, -2, -3}, 11.11f, -3.3);
		
		t1.setRef2(t2);
		
		pm0.makePersistent(t1);
		pm0.makePersistent(t2);
		
		long oid1 = (Long) pm0.getObjectId(t1);
		long oid2 = (Long) pm0.getObjectId(t2);
		
		pm0.currentTransaction().commit();
		TestTools.closePM();
		
		PersistenceManager pm = TestTools.openPM();
		pm.currentTransaction().begin();
		
		ZooClass s01 = ZooJdoHelper.schema(pm).getClass(TestClass.class.getName());
		assertNotNull(s01);

		ZooHandle hdl1 = ZooJdoHelper.schema(pm).getHandle(oid1);
		ZooHandle hdl2 = ZooJdoHelper.schema(pm).getHandle(oid2);

		assertEquals(126, hdl1.getAttrByte("_byte"));
		assertEquals(1234567, hdl1.getAttrInt("_int"));
		assertEquals(true, hdl1.getAttrBool("_bool"));
		assertEquals('x', hdl1.getAttrChar("_char"));
		assertEquals(32000, hdl1.getAttrShort("_short"));
		assertEquals(12345678901L, hdl1.getAttrLong("_long"));
		assertEquals(-1.1f, hdl1.getAttrFloat("_float"), 0.0);
		assertEquals(33.3, hdl1.getAttrDouble("_double"), 0.0);
		assertEquals("haha", hdl1.getAttrString("_string"));
		
		long oid2b = hdl1.getAttrRefOid("_ref2");
		assertEquals(oid2, oid2b);
		
		assertEquals(-126, hdl2.getAttrByte("_byte"));
		assertEquals(-1234567, hdl2.getAttrInt("_int"));
		assertEquals(true, hdl2.getAttrBool("_bool"));
		assertEquals('y', hdl2.getAttrChar("_char"));
		assertEquals(-32000, hdl2.getAttrShort("_short"));
		assertEquals(-12345678901L, hdl2.getAttrLong("_long"));
		assertEquals(11.11f, hdl2.getAttrFloat("_float"), 0.0);
		assertEquals(-3.3, hdl2.getAttrDouble("_double"), 0.0);
		assertEquals("hihi", hdl2.getAttrString("_string"));
		
		TestTools.closePM();
	}
	

	@Test
	public void testSchemaZooHandleReaderOnRenamed() {
		PersistenceManager pm0 = TestTools.openPM();
		pm0.currentTransaction().begin();
		
		TestClass t1 = new TestClass();
		t1.setData(1234567, true, 'x', (byte)126, (short)32000, 12345678901L, "haha", 
				new byte[]{1, 2, 3}, -1.1f, 33.3);
		TestClass t2 = new TestClass();
		t2.setData(-1234567, true, 'y', (byte)-126, (short)-32000, -12345678901L, "hihi", 
				new byte[]{-1, -2, -3}, 11.11f, -3.3);
		
		t1.setRef2(t2);
		
		pm0.makePersistent(t1);
		pm0.makePersistent(t2);
		
		long oid1 = (Long) pm0.getObjectId(t1);
		long oid2 = (Long) pm0.getObjectId(t2);
		
		pm0.currentTransaction().commit();
		TestTools.closePM();
		
		//rename schema
		pm0 = TestTools.openPM();
		pm0.currentTransaction().begin();
		ZooClass s = ZooJdoHelper.schema(pm0).getClass(TestClass.class);
		s.rename("x");
		pm0.currentTransaction().commit();
		TestTools.closePM();

		
		//test renamed
		PersistenceManager pm = TestTools.openPM();
		pm.currentTransaction().begin();
		
		ZooClass s01 = ZooJdoHelper.schema(pm).getClass("x");
		assertNotNull(s01);

		ZooHandle hdl1 = ZooJdoHelper.schema(pm).getHandle(oid1);
		ZooHandle hdl2 = ZooJdoHelper.schema(pm).getHandle(oid2);

		assertEquals(126, hdl1.getAttrByte("_byte"));
		assertEquals(1234567, hdl1.getAttrInt("_int"));
		assertEquals(true, hdl1.getAttrBool("_bool"));
		assertEquals('x', hdl1.getAttrChar("_char"));
		assertEquals(32000, hdl1.getAttrShort("_short"));
		assertEquals(12345678901L, hdl1.getAttrLong("_long"));
		assertEquals(-1.1f, hdl1.getAttrFloat("_float"), 0.0);
		assertEquals(33.3, hdl1.getAttrDouble("_double"), 0.0);
		assertEquals("haha", hdl1.getAttrString("_string"));
		
		long oid2b = hdl1.getAttrRefOid("_ref2");
		assertEquals(oid2, oid2b);
		
		assertEquals(-126, hdl2.getAttrByte("_byte"));
		assertEquals(-1234567, hdl2.getAttrInt("_int"));
		assertEquals(true, hdl2.getAttrBool("_bool"));
		assertEquals('y', hdl2.getAttrChar("_char"));
		assertEquals(-32000, hdl2.getAttrShort("_short"));
		assertEquals(-12345678901L, hdl2.getAttrLong("_long"));
		assertEquals(11.11f, hdl2.getAttrFloat("_float"), 0.0);
		assertEquals(-3.3, hdl2.getAttrDouble("_double"), 0.0);
		assertEquals("hihi", hdl2.getAttrString("_string"));

		TestTools.closePM();
		
		//rename back
		pm0 = TestTools.openPM();
		pm0.currentTransaction().begin();
		s = ZooJdoHelper.schema(pm0).getClass("x");
		s.rename(TestClass.class.getName());
		pm0.currentTransaction().commit();
		TestTools.closePM();
	}
	

	//Check that the handle attribute access works across page boundaries.
	@Test
	public void testSchemaZooHandleReaderWithPageWrap() {
		//populate
		PersistenceManager pm0 = TestTools.openPM();
		pm0.currentTransaction().begin();
		ArrayList<Long> oids = new ArrayList<Long>();
		for (int i = 0; i < 10000; i++) {
			TestClass t1 = new TestClass();
			t1.setData(1234567, true, 'x', (byte)126, (short)32000, 12345678901L, "haha", 
					new byte[]{1, 2, 3}, -1.1f, 33.3);
			t1.setRef2(t1);
			pm0.makePersistent(t1);
			oids.add( (Long)pm0.getObjectId(t1) );
		}
		pm0.currentTransaction().commit();
		TestTools.closePM();
		
		//check
		PersistenceManager pm = TestTools.openPM();
		pm.currentTransaction().begin();
		
		for (Long oid: oids) {
			ZooHandle hdl1 = ZooJdoHelper.schema(pm).getHandle(oid);
			assertEquals(126, hdl1.getAttrByte("_byte"));
			assertEquals(1234567, hdl1.getAttrInt("_int"));
			assertEquals(true, hdl1.getAttrBool("_bool"));
			assertEquals('x', hdl1.getAttrChar("_char"));
			assertEquals(32000, hdl1.getAttrShort("_short"));
			assertEquals(12345678901L, hdl1.getAttrLong("_long"));
			assertEquals(-1.1f, hdl1.getAttrFloat("_float"), 0.0);
			assertEquals(33.3, hdl1.getAttrDouble("_double"), 0.0);
			assertEquals("haha", hdl1.getAttrString("_string"));
			long oid2 = hdl1.getAttrRefOid("_ref2");
			assertEquals((long)oid, oid2);
		}
		
		TestTools.closePM();
	}
	

	@Test
	public void testHandleDeletion() {
		PersistenceManager pm = TestTools.openPM();
		pm.currentTransaction().begin();
		
		ZooClass s01 = ZooJdoHelper.schema(pm).getClass(TestClass.class.getName());

		//to delete now
		ZooHandle hdl1 = s01.newInstance();
		//to delete in next tx
		ZooHandle hdl2 = s01.newInstance();
		ZooHandle hdl3 = s01.newInstance();
		long oid1 = hdl1.getOid();
		long oid2 = hdl2.getOid();
		long oid3 = hdl3.getOid();
		
		hdl1.remove();
		
		pm.currentTransaction().commit();
		pm.currentTransaction().begin();
		
		try {
			hdl1.getAttrInt("_int");
		} catch (Exception e) {
			//good
		}
		assertNull(ZooJdoHelper.schema(pm).getHandle(oid1));
		
		hdl2.remove();
		
		pm.currentTransaction().commit();
		TestTools.closePM();

		//test in new session
		pm = TestTools.openPM();
		pm.currentTransaction().begin();
		
		assertNull(ZooJdoHelper.schema(pm).getHandle(oid1));
		assertNull(ZooJdoHelper.schema(pm).getHandle(oid2));
		ZooJdoHelper.schema(pm).getHandle(oid3).remove();
		
		pm.currentTransaction().commit();
		TestTools.closePM();
	}


	@Test
	public void testNewHandleRollback() {
		PersistenceManager pm = TestTools.openPM();
		pm.currentTransaction().begin();
		
		ZooClass s01 = ZooJdoHelper.schema(pm).getClass(TestClass.class.getName());

		//to delete
		ZooHandle hdl1 = s01.newInstance();
		//to r
		ZooHandle hdl2 = s01.newInstance(12345);
		long oid1 = hdl1.getOid();
		long oid2 = hdl2.getOid();
		
		pm.currentTransaction().rollback();
		pm.currentTransaction().begin();
		
		assertNull(ZooJdoHelper.schema(pm).getHandle(oid1));
		assertNull(ZooJdoHelper.schema(pm).getHandle(oid2));

		pm.currentTransaction().commit();
		TestTools.closePM();

		//test in new session
		pm = TestTools.openPM();
		pm.currentTransaction().begin();
		
		assertNull(ZooJdoHelper.schema(pm).getHandle(oid1));
		assertNull(ZooJdoHelper.schema(pm).getHandle(oid2));
		
		pm.currentTransaction().commit();
		TestTools.closePM();
	}

	@Test
	public void testChangeRollback() {
		PersistenceManager pm = TestTools.openPM();
		pm.currentTransaction().begin();
		
		ZooClass s01 = ZooJdoHelper.schema(pm).getClass(TestClass.class.getName());

		//to delete
		ZooHandle hdl1 = s01.newInstance();
		hdl1.setValue("_int", 21);
		
		pm.currentTransaction().commit();
		pm.currentTransaction().begin();

		assertEquals(21, hdl1.getAttrInt("_int"));
		hdl1.setValue("_int", 2);
		assertEquals(2, hdl1.getAttrInt("_int"));

		pm.currentTransaction().rollback();
		pm.currentTransaction().begin();

		assertEquals(21, hdl1.getAttrInt("_int"));
		
		
		pm.currentTransaction().commit();
		TestTools.closePM();
	}
}