JDOBatchInsertTest.java - This Java code is a test suite fo…

/tests/com/google/appengine/datanucleus/jdo/JDOBatchInsertTest.java

http://datanucleus-appengine.googlecode.com/ · Java · 415 lines · 364 code · 29 blank · 22 comment · 1 complexity · 66f830f42dc613368cf3bb7724847523 MD5 · raw file

/**********************************************************************
Copyright (c) 2009 Google Inc.

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 com.google.appengine.datanucleus.jdo;

import com.google.appengine.api.datastore.DatastoreServiceConfig;
import com.google.appengine.api.datastore.Entity;
import com.google.appengine.api.datastore.EntityNotFoundException;
import com.google.appengine.api.datastore.Key;
import com.google.appengine.api.datastore.KeyFactory;
import com.google.appengine.datanucleus.Utils;
import com.google.appengine.datanucleus.test.jdo.BidirectionalChildListJDO;
import com.google.appengine.datanucleus.test.jdo.Flight;
import com.google.appengine.datanucleus.test.jdo.HasKeyAncestorKeyPkJDO;
import com.google.appengine.datanucleus.test.jdo.HasOneToManyListJDO;
import com.google.appengine.datanucleus.test.jdo.HasOneToOneJDO;

import java.lang.reflect.Method;
import java.util.List;

import javax.jdo.JDOHelper;
import javax.jdo.JDOUserException;
import javax.jdo.ObjectState;

/**
 * @author Max Ross <maxr@google.com>
 */
public class JDOBatchInsertTest extends JDOBatchTestCase {

  private static Flight newFlight() {
    Flight f1 = new Flight();
    f1.setOrigin("BOS");
    f1.setDest("MIA");
    f1.setMe(2);
    f1.setYou(4);
    f1.setName("Harold");

    return f1;
  }

  BatchRecorder newBatchRecorder() {
    DatastoreServiceConfig config = getStoreManager().getDefaultDatastoreServiceConfigForReads();
    return new BatchRecorder(config) {
      boolean isBatchMethod(Method method) {
        return method.getName().equals("put") && List.class.isAssignableFrom(method.getReturnType());
      }
    };
  }

  public void testMakePersistentAll_NoTxn() {
    switchDatasource(JDOTestCase.PersistenceManagerFactoryName.nontransactional);
    Flight f1 = newFlight();
    Flight f2 = newFlight();
    pm.makePersistentAll(f1, f2);
    assertEquals(2, countForClass(Flight.class));
    assertEquals(1, batchRecorder.batchOps);

    Flight f3 = newFlight();
    Flight f4 = newFlight();
    pm.makePersistentAll(Utils.newArrayList(f3, f4));
    assertEquals(4, countForClass(Flight.class));
    assertEquals(2, batchRecorder.batchOps);
  }

  public void testMakePersistentAll_OneEntity_NoTxn() {
    switchDatasource(JDOTestCase.PersistenceManagerFactoryName.nontransactional);
    Flight f1 = newFlight();
    pm.makePersistentAll(f1);
    assertEquals(1, countForClass(Flight.class));
    assertEquals(0, batchRecorder.batchOps);

    Flight f2 = newFlight();
    pm.makePersistentAll(Utils.newArrayList(f2));
    assertEquals(2, countForClass(Flight.class));
    assertEquals(0, batchRecorder.batchOps);
  }

  public void testMakePersistentAll_Txn_MultipleEntityGroups() {
    switchDatasource(JDOTestCase.PersistenceManagerFactoryName.transactional);
    Flight f1 = newFlight();
    Flight f2 = newFlight();
    beginTxn();
    try {
      pm.makePersistentAll(f1, f2);
      fail("expected iae");
    } catch (JDOUserException nue) {
      // good
    }
    rollbackTxn();
    assertEquals(0, countForClass(Flight.class));
    assertEquals(1, batchRecorder.batchOps);

    Flight f3 = newFlight();
    Flight f4 = newFlight();
    beginTxn();
    try {
      pm.makePersistentAll(Utils.newArrayList(f3, f4));
      fail("expected iae");
    } catch (JDOUserException nue) {
      // good
    }
    assertEquals(0, countForClass(Flight.class));
    assertEquals(2, batchRecorder.batchOps);
  }

  public void testMakePersistentAll_Txn_OneEntityGroup() {
    switchDatasource(JDOTestCase.PersistenceManagerFactoryName.transactional);
    Key parentKey = KeyFactory.createKey("yar", 24);
    HasKeyAncestorKeyPkJDO pojo1 = new HasKeyAncestorKeyPkJDO();
    pojo1.setAncestorKey(parentKey);
    HasKeyAncestorKeyPkJDO pojo2 = new HasKeyAncestorKeyPkJDO();
    pojo2.setAncestorKey(parentKey);
    beginTxn();
    pm.makePersistentAll(pojo1, pojo2);
    commitTxn();
    assertEquals(2, countForClass(HasKeyAncestorKeyPkJDO.class));
    assertEquals(1, batchRecorder.batchOps);
  }

  public void testMakePersistentAll_OneEntity_Txn() {
    switchDatasource(JDOTestCase.PersistenceManagerFactoryName.transactional);
    Flight f1 = newFlight();
    beginTxn();
    pm.makePersistentAll(f1);
    commitTxn();
    assertEquals(1, countForClass(Flight.class));
    assertEquals(0, batchRecorder.batchOps);

    Flight f2 = newFlight();
    beginTxn();
    pm.makePersistentAll(Utils.newArrayList(f2));
    commitTxn();
    assertEquals(2, countForClass(Flight.class));
    assertEquals(0, batchRecorder.batchOps);
  }

  public void testMakePersistentAll_CascadeInsert_OneToOne_NoTxn() throws EntityNotFoundException {
    switchDatasource(JDOTestCase.PersistenceManagerFactoryName.nontransactional);
    Flight f1 = newFlight();
    Flight f2 = newFlight();
    HasOneToOneJDO parent1 = new HasOneToOneJDO();
    parent1.setFlight(f1);
    HasOneToOneJDO parent2 = new HasOneToOneJDO();
    parent2.setFlight(f2);
    pm.makePersistentAll(parent1, parent2);
    assertEquals(2, countForClass(HasOneToOneJDO.class));
    assertEquals(2, countForClass(Flight.class));
    assertEquals(1, batchRecorder.batchOps);

    Entity parent1Entity = ds.get(KeyFactory.stringToKey(parent1.getId()));
    Entity flight1Entity = ds.get(KeyFactory.stringToKey(f1.getId()));
    assertEquals(6, parent1Entity.getProperties().size());
    assertTrue(parent1Entity.hasProperty("hasKeyPK_key_OID"));
    assertNull(parent1Entity.getProperty("hasKeyPK_key_OID"));
    assertTrue(parent1Entity.hasProperty("hasParent_key_OID"));
    assertNull(parent1Entity.getProperty("hasParent_key_OID"));
    assertTrue(parent1Entity.hasProperty("hasParentKeyPK_key_OID"));
    assertNull(parent1Entity.getProperty("hasParentKeyPK_key_OID"));
    assertTrue(parent1Entity.hasProperty("notDependent_id_OID"));
    assertNull(parent1Entity.getProperty("notDependent_id_OID"));
    assertEquals(flight1Entity.getKey(), parent1Entity.getProperty("flight_id_OID"));
    Entity parent2Entity = ds.get(KeyFactory.stringToKey(parent2.getId()));
    Entity flight2Entity = ds.get(KeyFactory.stringToKey(f2.getId()));
    assertEquals(6, parent2Entity.getProperties().size());
    assertTrue(parent2Entity.hasProperty("hasKeyPK_key_OID"));
    assertNull(parent2Entity.getProperty("hasKeyPK_key_OID"));
    assertTrue(parent2Entity.hasProperty("hasParent_key_OID"));
    assertNull(parent2Entity.getProperty("hasParent_key_OID"));
    assertTrue(parent2Entity.hasProperty("hasParentKeyPK_key_OID"));
    assertNull(parent2Entity.getProperty("hasParentKeyPK_key_OID"));
    assertTrue(parent2Entity.hasProperty("notDependent_id_OID"));
    assertNull(parent2Entity.getProperty("notDependent_id_OID"));
    assertEquals(flight2Entity.getKey(), parent2Entity.getProperty("flight_id_OID"));
  }

  public void testMakePersistentAll_CascadeInsert_OneToOne_MultipleEntityGroups_Txn() {
    switchDatasource(JDOTestCase.PersistenceManagerFactoryName.transactional);
    Flight f1 = newFlight();
    Flight f2 = newFlight();
    HasOneToOneJDO parent1 = new HasOneToOneJDO();
    parent1.setFlight(f1);
    HasOneToOneJDO parent2 = new HasOneToOneJDO();
    parent2.setFlight(f2);
    beginTxn();
    try {
      pm.makePersistentAll(parent1, parent2);
      fail("expected exception");
    } catch (JDOUserException nue) {
      // good
    }
  }

  public void testMakePersistentAll_CascadeInsert_OneToOne_OneEntityGroup_Txn()
      throws EntityNotFoundException {
    switchDatasource(JDOTestCase.PersistenceManagerFactoryName.transactional);
    Flight f1 = newFlight();
    Flight f2 = newFlight();
    HasOneToOneJDO parent1 = new HasOneToOneJDO();
    parent1.setId(new KeyFactory.Builder("Yar", 43).addChild(HasOneToOneJDO.class.getSimpleName(), "k1").getString());
    parent1.setFlight(f1);
    HasOneToOneJDO parent2 = new HasOneToOneJDO();
    parent2.setId(new KeyFactory.Builder("Yar", 43).addChild(HasOneToOneJDO.class.getSimpleName(), "k2").getString());
    parent2.setFlight(f2);
    beginTxn();
    pm.makePersistentAll(parent1, parent2);
    commitTxn();
    assertEquals(2, countForClass(HasOneToOneJDO.class));
    assertEquals(2, countForClass(Flight.class));
    assertEquals(1, batchRecorder.batchOps);

    Entity parent1Entity = ds.get(KeyFactory.stringToKey(parent1.getId()));
    Entity flight1Entity = ds.get(KeyFactory.stringToKey(f1.getId()));
    assertEquals(6, parent1Entity.getProperties().size());
    assertTrue(parent1Entity.hasProperty("hasKeyPK_key_OID"));
    assertNull(parent1Entity.getProperty("hasKeyPK_key_OID"));
    assertTrue(parent1Entity.hasProperty("hasParent_key_OID"));
    assertNull(parent1Entity.getProperty("hasParent_key_OID"));
    assertTrue(parent1Entity.hasProperty("hasParentKeyPK_key_OID"));
    assertNull(parent1Entity.getProperty("hasParentKeyPK_key_OID"));
    assertTrue(parent1Entity.hasProperty("notDependent_id_OID"));
    assertNull(parent1Entity.getProperty("notDependent_id_OID"));
    assertEquals(flight1Entity.getKey(), parent1Entity.getProperty("flight_id_OID"));
    Entity parent2Entity = ds.get(KeyFactory.stringToKey(parent2.getId()));
    Entity flight2Entity = ds.get(KeyFactory.stringToKey(f2.getId()));
    assertEquals(6, parent2Entity.getProperties().size());
    assertTrue(parent2Entity.hasProperty("hasKeyPK_key_OID"));
    assertNull(parent2Entity.getProperty("hasKeyPK_key_OID"));
    assertTrue(parent2Entity.hasProperty("hasParent_key_OID"));
    assertNull(parent2Entity.getProperty("hasParent_key_OID"));
    assertTrue(parent2Entity.hasProperty("hasParentKeyPK_key_OID"));
    assertNull(parent2Entity.getProperty("hasParentKeyPK_key_OID"));
    assertTrue(parent2Entity.hasProperty("notDependent_id_OID"));
    assertNull(parent2Entity.getProperty("notDependent_id_OID"));
    assertEquals(flight2Entity.getKey(), parent2Entity.getProperty("flight_id_OID"));
  }

  public void testMakePersistentAll_CascadeInsert_OneToMany_NoTxn() throws EntityNotFoundException {
    switchDatasource(JDOTestCase.PersistenceManagerFactoryName.nontransactional);
    Flight f1 = newFlight();
    Flight f2 = newFlight();
    Flight f3 = newFlight();
    Flight f4 = newFlight();
    BidirectionalChildListJDO bidirChild1 = new BidirectionalChildListJDO();
    BidirectionalChildListJDO bidirChild2 = new BidirectionalChildListJDO();
    BidirectionalChildListJDO bidirChild3 = new BidirectionalChildListJDO();
    BidirectionalChildListJDO bidirChild4 = new BidirectionalChildListJDO();
    HasOneToManyListJDO parent1 = new HasOneToManyListJDO();
    parent1.getFlights().add(f1);
    parent1.getFlights().add(f2);
    parent1.getBidirChildren().add(bidirChild1);
    parent1.getBidirChildren().add(bidirChild2);
    HasOneToManyListJDO parent2 = new HasOneToManyListJDO();
    parent2.getFlights().add(f3);
    parent2.getFlights().add(f4);
    parent2.getBidirChildren().add(bidirChild3);
    parent2.getBidirChildren().add(bidirChild4);
    pm.makePersistentAll(parent1, parent2);
    assertEquals(2, countForClass(HasOneToManyListJDO.class));
    assertEquals(4, countForClass(Flight.class));
    assertEquals(4, countForClass(BidirectionalChildListJDO.class));
    assertEquals(1, batchRecorder.batchOps);

    Entity parent1Entity = ds.get(KeyFactory.stringToKey(parent1.getId()));
    Entity parent2Entity = ds.get(KeyFactory.stringToKey(parent2.getId()));
    Entity flight1Entity = ds.get(KeyFactory.stringToKey(f1.getId()));
    Entity flight2Entity = ds.get(KeyFactory.stringToKey(f2.getId()));
    Entity flight3Entity = ds.get(KeyFactory.stringToKey(f3.getId()));
    Entity flight4Entity = ds.get(KeyFactory.stringToKey(f4.getId()));
    Entity bidir1Entity = ds.get(KeyFactory.stringToKey(bidirChild1.getId()));
    Entity bidir2Entity = ds.get(KeyFactory.stringToKey(bidirChild2.getId()));
    Entity bidir3Entity = ds.get(KeyFactory.stringToKey(bidirChild3.getId()));
    Entity bidir4Entity = ds.get(KeyFactory.stringToKey(bidirChild4.getId()));
    assertEquals(4, parent1Entity.getProperties().size());
    assertTrue(parent1Entity.hasProperty("hasKeyPks"));
    assertNull(parent1Entity.getProperty("hasKeyPks"));
    assertEquals(
        Utils.newArrayList(flight1Entity.getKey(), flight2Entity.getKey()),
        parent1Entity.getProperty("flights"));
    assertEquals(
        Utils.newArrayList(bidir1Entity.getKey(), bidir2Entity.getKey()),
        parent1Entity.getProperty("bidirChildren"));

    assertEquals(4, parent2Entity.getProperties().size());
    assertTrue(parent2Entity.hasProperty("hasKeyPks"));
    assertNull(parent2Entity.getProperty("hasKeyPks"));
    assertEquals(
        Utils.newArrayList(flight3Entity.getKey(), flight4Entity.getKey()),
        parent2Entity.getProperty("flights"));
    assertEquals(
        Utils.newArrayList(bidir3Entity.getKey(), bidir4Entity.getKey()),
        parent2Entity.getProperty("bidirChildren"));
  }

  public void testMakePersistentAll_CascadeInsert_OneToMany_MultipleEntityGroups_Txn() {
    switchDatasource(JDOTestCase.PersistenceManagerFactoryName.transactional);
    Flight f1 = newFlight();
    Flight f2 = newFlight();
    Flight f3 = newFlight();
    Flight f4 = newFlight();
    BidirectionalChildListJDO bidirChild1 = new BidirectionalChildListJDO();
    BidirectionalChildListJDO bidirChild2 = new BidirectionalChildListJDO();
    BidirectionalChildListJDO bidirChild3 = new BidirectionalChildListJDO();
    BidirectionalChildListJDO bidirChild4 = new BidirectionalChildListJDO();
    HasOneToManyListJDO parent1 = new HasOneToManyListJDO();
    parent1.getFlights().add(f1);
    parent1.getFlights().add(f2);
    parent1.getBidirChildren().add(bidirChild1);
    parent1.getBidirChildren().add(bidirChild2);
    HasOneToManyListJDO parent2 = new HasOneToManyListJDO();
    parent2.getFlights().add(f3);
    parent2.getFlights().add(f4);
    parent2.getBidirChildren().add(bidirChild3);
    parent2.getBidirChildren().add(bidirChild4);
    beginTxn();
    try {
      pm.makePersistentAll(parent1, parent2);
      fail("expected exception");
    } catch (JDOUserException nue) {
      // good
    }
  }

  public void testMakePersistentAll_CascadeInsert_OneToMany_OneEntityGroup_Txn()
      throws EntityNotFoundException {
    switchDatasource(JDOTestCase.PersistenceManagerFactoryName.transactional);
    Flight f1 = newFlight();
    Flight f2 = newFlight();
    Flight f3 = newFlight();
    Flight f4 = newFlight();
    BidirectionalChildListJDO bidirChild1 = new BidirectionalChildListJDO();
    BidirectionalChildListJDO bidirChild2 = new BidirectionalChildListJDO();
    BidirectionalChildListJDO bidirChild3 = new BidirectionalChildListJDO();
    BidirectionalChildListJDO bidirChild4 = new BidirectionalChildListJDO();
    HasOneToManyListJDO parent1 = new HasOneToManyListJDO();
    parent1.setId(new KeyFactory.Builder("Yar", 43).addChild(HasOneToManyListJDO.class.getSimpleName(), "k1").getString());
    parent1.getFlights().add(f1);
    parent1.getFlights().add(f2);
    parent1.getBidirChildren().add(bidirChild1);
    parent1.getBidirChildren().add(bidirChild2);
    HasOneToManyListJDO parent2 = new HasOneToManyListJDO();
    parent2.setId(new KeyFactory.Builder("Yar", 43).addChild(HasOneToManyListJDO.class.getSimpleName(), "k2").getString());
    parent2.getFlights().add(f3);
    parent2.getFlights().add(f4);
    parent2.getBidirChildren().add(bidirChild3);
    parent2.getBidirChildren().add(bidirChild4);
    beginTxn();
    pm.makePersistentAll(parent1, parent2);
    commitTxn();
    assertEquals(2, countForClass(HasOneToManyListJDO.class));
    assertEquals(4, countForClass(Flight.class));
    assertEquals(4, countForClass(BidirectionalChildListJDO.class));
    assertEquals(1, batchRecorder.batchOps);

    Entity parent1Entity = ds.get(KeyFactory.stringToKey(parent1.getId()));
    Entity parent2Entity = ds.get(KeyFactory.stringToKey(parent2.getId()));
    Entity flight1Entity = ds.get(KeyFactory.stringToKey(f1.getId()));
    Entity flight2Entity = ds.get(KeyFactory.stringToKey(f2.getId()));
    Entity flight3Entity = ds.get(KeyFactory.stringToKey(f3.getId()));
    Entity flight4Entity = ds.get(KeyFactory.stringToKey(f4.getId()));
    Entity bidir1Entity = ds.get(KeyFactory.stringToKey(bidirChild1.getId()));
    Entity bidir2Entity = ds.get(KeyFactory.stringToKey(bidirChild2.getId()));
    Entity bidir3Entity = ds.get(KeyFactory.stringToKey(bidirChild3.getId()));
    Entity bidir4Entity = ds.get(KeyFactory.stringToKey(bidirChild4.getId()));
    assertEquals(4, parent1Entity.getProperties().size());
    assertTrue(parent1Entity.hasProperty("hasKeyPks"));
    assertNull(parent1Entity.getProperty("hasKeyPks"));
    assertEquals(
        Utils.newArrayList(flight1Entity.getKey(), flight2Entity.getKey()),
        parent1Entity.getProperty("flights"));
    assertEquals(
        Utils.newArrayList(bidir1Entity.getKey(), bidir2Entity.getKey()),
        parent1Entity.getProperty("bidirChildren"));

    assertEquals(4, parent2Entity.getProperties().size());
    assertTrue(parent2Entity.hasProperty("hasKeyPks"));
    assertNull(parent2Entity.getProperty("hasKeyPks"));
    assertEquals(
        Utils.newArrayList(flight3Entity.getKey(), flight4Entity.getKey()),
        parent2Entity.getProperty("flights"));
    assertEquals(
        Utils.newArrayList(bidir3Entity.getKey(), bidir4Entity.getKey()),
        parent2Entity.getProperty("bidirChildren"));
  }

  public void testCombineInsertAndUpdate_NoTxn() {
    switchDatasource(JDOTestCase.PersistenceManagerFactoryName.nontransactional);
    Flight f1 = newFlight();
    pm.makePersistent(f1);
    f1 = pm.detachCopy(f1);
    assertEquals(1, countForClass(Flight.class));
    assertEquals(0, batchRecorder.batchOps);
    pm.close();
    assertEquals(ObjectState.DETACHED_CLEAN, JDOHelper.getObjectState(f1));
    f1.setName("jimmy");
    pm = pmf.getPersistenceManager();
    Flight f2 = new Flight();
    Flight f3 = new Flight();
    pm.makePersistentAll(f1, f2, f3);
    pm.close();
    assertEquals(3, countForClass(Flight.class));
    assertEquals(1, batchRecorder.batchOps);
  }
}
Summary ✨

This Java code is a test suite for JDO (Java Data Objects) persistence manager functionality, specifically testing various scenarios such as persistent object creation, detachment, updating, and combining insertions with updates in different transactional states (with/without transactions). It verifies the expected behavior of the persistence manager under these conditions.
Tech Fingerprint

Alerts (3)

'=' Maintainability Info: Avoid using unnamed 'magic' numbers directly in comparisons or assignments. Use named constants (static final variables) instead to improve readability and maintainability.
120
'.close()' Manual .close() call detected. Prefer using try-with-resources (Java 7+) for automatic and safer resource management, especially handling exceptions during close.
404 411