/**********************************************************************

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.DatastoreService;

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.api.datastore.Query;

import com.google.appengine.api.datastore.Transaction;

import com.google.appengine.api.datastore.TransactionOptions;

import com.google.appengine.datanucleus.DatastoreServiceFactoryInternal;

import com.google.appengine.datanucleus.DatastoreServiceInterceptor;

import com.google.appengine.datanucleus.StorageVersion;

import com.google.appengine.datanucleus.Utils;

import com.google.appengine.datanucleus.test.jdo.BidirectionalChildJDO;

import com.google.appengine.datanucleus.test.jdo.BidirectionalChildLongPkJDO;

import com.google.appengine.datanucleus.test.jdo.BidirectionalChildUnencodedStringPkJDO;

import com.google.appengine.datanucleus.test.jdo.Flight;

import com.google.appengine.datanucleus.test.jdo.HasExplicitIndexColumnJDO;

import com.google.appengine.datanucleus.test.jdo.HasKeyPkJDO;

import com.google.appengine.datanucleus.test.jdo.HasOneToManyJDO;

import com.google.appengine.datanucleus.test.jdo.HasOneToManyKeyPkJDO;

import com.google.appengine.datanucleus.test.jdo.HasOneToManyListJDO;

import com.google.appengine.datanucleus.test.jdo.HasOneToManyListWithOrderByJDO;

import com.google.appengine.datanucleus.test.jdo.HasOneToManyLongPkJDO;

import com.google.appengine.datanucleus.test.jdo.HasOneToManyUnencodedStringPkJDO;

import com.google.appengine.datanucleus.test.jdo.HasOneToManyWithOrderByJDO;



import org.easymock.EasyMock;



import java.lang.reflect.Method;

import java.util.Collection;

import java.util.Collections;

import java.util.ConcurrentModificationException;

import java.util.Iterator;

import java.util.List;

import java.util.Set;



import javax.jdo.JDOFatalUserException;



import static com.google.appengine.datanucleus.TestUtils.assertKeyParentEquals;



/**

 * @author Max Ross <maxr@google.com>

 */

abstract class JDOOneToManyTestCase extends JDOTestCase {



  @Override

  protected void tearDown() throws Exception {

    try {

      if (!pm.isClosed() && pm.currentTransaction().isActive()) {

        pm.currentTransaction().rollback();

      }

      pmf.close();

    } finally {

      super.tearDown();

    }

  }



  void testInsert_NewParentAndChild(HasOneToManyJDO parent,

      BidirectionalChildJDO bidirChild, StartEnd startEnd) throws EntityNotFoundException {

    bidirChild.setChildVal("yam");



    Flight f = newFlight();



    HasKeyPkJDO hasKeyPk = new HasKeyPkJDO();

    hasKeyPk.setStr("yag");



    parent.addBidirChild(bidirChild);

    bidirChild.setParent(parent);

    parent.addFlight(f);

    parent.addHasKeyPk(hasKeyPk);

    parent.setVal("yar");



    startEnd.start();

    pm.makePersistent(parent);

    startEnd.end();



    assertNotNull(bidirChild.getId());

    assertNotNull(f.getId());

    assertNotNull(hasKeyPk.getKey());



    Entity bidirChildEntity = ds.get(KeyFactory.stringToKey(bidirChild.getId()));

    assertNotNull(bidirChildEntity);

    assertEquals("yam", bidirChildEntity.getProperty("childVal"));

    assertEquals(KeyFactory.stringToKey(bidirChild.getId()), bidirChildEntity.getKey());

    assertKeyParentEquals(parent.getId(), bidirChildEntity, bidirChild.getId());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, bidirChildEntity.getProperty("bidirChildren_INTEGER_IDX"));

    }



    Entity flightEntity = ds.get(KeyFactory.stringToKey(f.getId()));

    assertNotNull(flightEntity);

    assertEquals("bos", flightEntity.getProperty("origin"));

    assertEquals("mia", flightEntity.getProperty("dest"));

    assertEquals("jimmy", flightEntity.getProperty("name"));

    assertEquals(KeyFactory.stringToKey(f.getId()), flightEntity.getKey());

    assertKeyParentEquals(parent.getId(), flightEntity, f.getId());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, flightEntity.getProperty("flights_INTEGER_IDX"));

    }



    Entity hasKeyPkEntity = ds.get(hasKeyPk.getKey());

    assertNotNull(hasKeyPkEntity);

    assertEquals("yag", hasKeyPkEntity.getProperty("str"));

    assertEquals(hasKeyPk.getKey(), hasKeyPkEntity.getKey());

    assertKeyParentEquals(parent.getId(), hasKeyPkEntity, hasKeyPk.getKey());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, hasKeyPkEntity.getProperty("hasKeyPks_INTEGER_IDX"));

    }



    Entity parentEntity = ds.get(KeyFactory.stringToKey(parent.getId()));

    assertNotNull(parentEntity);

    assertEquals(4, parentEntity.getProperties().size());

    assertEquals("yar", parentEntity.getProperty("val"));

    assertEquals(Utils.newArrayList(bidirChildEntity.getKey()), parentEntity.getProperty("bidirChildren"));

    assertEquals(Utils.newArrayList(flightEntity.getKey()), parentEntity.getProperty("flights"));

    assertEquals(Utils.newArrayList(hasKeyPkEntity.getKey()), parentEntity.getProperty("hasKeyPks"));



    assertCountsInDatastore(parent.getClass(), bidirChild.getClass(), 1, 1);

  }



  void testInsert_ExistingParentNewChild(HasOneToManyJDO pojo,

      BidirectionalChildJDO bidirChild, StartEnd startEnd) throws EntityNotFoundException {

    pojo.setVal("yar");



    startEnd.start();

    pm.makePersistent(pojo);

    startEnd.end();



    startEnd.start();

    assertNotNull(pojo.getId());

    assertTrue(pojo.getFlights().isEmpty());

    assertTrue(pojo.getHasKeyPks().isEmpty());

    assertTrue(pojo.getBidirChildren().isEmpty());



    Entity pojoEntity = ds.get(KeyFactory.stringToKey(pojo.getId()));

    assertNotNull(pojoEntity);

    assertEquals(4, pojoEntity.getProperties().size());

    assertEquals("yar", pojoEntity.getProperty("val"));

    assertTrue(pojoEntity.hasProperty("bidirChildren"));

    assertNull(pojoEntity.getProperty("bidirChildren"));

    assertTrue(pojoEntity.hasProperty("flights"));

    assertNull(pojoEntity.getProperty("flights"));

    assertTrue(pojoEntity.hasProperty("hasKeyPks"));

    assertNull(pojoEntity.getProperty("hasKeyPks"));

    startEnd.end();



    startEnd.start();

    pojo = pm.makePersistent(pojo);

    assertEquals("yar", pojo.getVal());

    Flight f = newFlight();

    pojo.addFlight(f);



    HasKeyPkJDO hasKeyPk = new HasKeyPkJDO();

    hasKeyPk.setStr("yag");

    pojo.addHasKeyPk(hasKeyPk);

    bidirChild.setChildVal("yam");

    pojo.addBidirChild(bidirChild);

    startEnd.end();



    startEnd.start();

    assertNotNull(bidirChild.getId());

    assertNotNull(bidirChild.getParent());

    assertNotNull(f.getId());

    assertNotNull(hasKeyPk.getKey());

    startEnd.end();

    

    Entity bidirChildEntity = ds.get(KeyFactory.stringToKey(bidirChild.getId()));

    assertNotNull(bidirChildEntity);

    assertEquals("yam", bidirChildEntity.getProperty("childVal"));

    assertEquals(KeyFactory.stringToKey(bidirChild.getId()), bidirChildEntity.getKey());

    assertKeyParentEquals(pojo.getId(), bidirChildEntity, bidirChild.getId());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, bidirChildEntity.getProperty("bidirChildren_INTEGER_IDX"));

    }



    Entity flightEntity = ds.get(KeyFactory.stringToKey(f.getId()));

    assertNotNull(flightEntity);

    assertEquals("bos", flightEntity.getProperty("origin"));

    assertEquals("mia", flightEntity.getProperty("dest"));

    assertEquals("jimmy", flightEntity.getProperty("name"));

    assertEquals(KeyFactory.stringToKey(f.getId()), flightEntity.getKey());

    assertKeyParentEquals(pojo.getId(), flightEntity, f.getId());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, flightEntity.getProperty("flights_INTEGER_IDX"));

    }



    Entity hasKeyPkEntity = ds.get(hasKeyPk.getKey());

    assertNotNull(hasKeyPkEntity);

    assertEquals("yag", hasKeyPkEntity.getProperty("str"));

    assertEquals(hasKeyPk.getKey(), hasKeyPkEntity.getKey());

    assertKeyParentEquals(pojo.getId(), hasKeyPkEntity, hasKeyPk.getKey());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, hasKeyPkEntity.getProperty("hasKeyPks_INTEGER_IDX"));

    }



    Entity parentEntity = ds.get(KeyFactory.stringToKey(pojo.getId()));

    assertNotNull(parentEntity);

    assertEquals(4, parentEntity.getProperties().size());

    assertEquals("yar", parentEntity.getProperty("val"));

    assertEquals(Utils.newArrayList(bidirChildEntity.getKey()), parentEntity.getProperty("bidirChildren"));

    assertEquals(Utils.newArrayList(flightEntity.getKey()), parentEntity.getProperty("flights"));

    assertEquals(Utils.newArrayList(hasKeyPkEntity.getKey()), parentEntity.getProperty("hasKeyPks"));



    assertCountsInDatastore(pojo.getClass(), bidirChild.getClass(), 1, 1);

  }



  void testSwapAtPosition(HasOneToManyJDO pojo,

      BidirectionalChildJDO bidir1,

      BidirectionalChildJDO bidir2, StartEnd startEnd) throws EntityNotFoundException {

    pojo.setVal("yar");

    bidir2.setChildVal("yam");

    Flight f = newFlight();

    HasKeyPkJDO hasKeyPk = new HasKeyPkJDO();



    pojo.addFlight(f);

    pojo.addHasKeyPk(hasKeyPk);

    pojo.addBidirChild(bidir1);

    bidir1.setParent(pojo);



    startEnd.start();

    pm.makePersistent(pojo);

    startEnd.end();



    assertCountsInDatastore(pojo.getClass(), bidir1.getClass(), 1, 1);



    startEnd.start();

    pojo = pm.getObjectById(pojo.getClass(), pojo.getId());

    String bidir1Id = pojo.getBidirChildren().iterator().next().getId();

    String flight1Id = pojo.getFlights().iterator().next().getId();

    Key hasKeyPk1Key = pojo.getHasKeyPks().iterator().next().getKey();

    pojo.addBidirChildAtPosition(bidir2, 0);



    Flight f2 = newFlight();

    f2.setName("another name");

    pojo.addFlightAtPosition(f2, 0);



    HasKeyPkJDO hasKeyPk2 = new HasKeyPkJDO();

    hasKeyPk2.setStr("another str");

    pojo.addHasKeyPkAtPosition(hasKeyPk2, 0);

    startEnd.end();



    startEnd.start();

    pojo = pm.getObjectById(pojo.getClass(), pojo.getId());

    assertNotNull(pojo.getId());

    assertEquals(1, pojo.getFlights().size());

    assertEquals(1, pojo.getHasKeyPks().size());

    assertEquals(1, pojo.getBidirChildren().size());

    assertNotNull(bidir2.getId());

    assertNotNull(bidir2.getParent());

    assertNotNull(f2.getId());

    assertNotNull(hasKeyPk2.getKey());



    Entity pojoEntity = ds.get(KeyFactory.stringToKey(pojo.getId()));

    assertNotNull(pojoEntity);

    assertEquals(4, pojoEntity.getProperties().size());

    assertEquals(Utils.newArrayList(KeyFactory.stringToKey(bidir2.getId())), pojoEntity.getProperty("bidirChildren"));

    assertEquals(Utils.newArrayList(KeyFactory.stringToKey(f2.getId())), pojoEntity.getProperty("flights"));

    assertEquals(Utils.newArrayList(hasKeyPk2.getKey()), pojoEntity.getProperty("hasKeyPks"));



    startEnd.end();



    try {

      ds.get(KeyFactory.stringToKey(bidir1Id));

      fail("expected EntityNotFoundException");

    } catch (EntityNotFoundException enfe) {

      // good

    }

    try {

      ds.get(KeyFactory.stringToKey(flight1Id));

      fail("expected EntityNotFoundException");

    } catch (EntityNotFoundException enfe) {

      // good

    }

    try {

      ds.get(hasKeyPk1Key);

      fail("expected EntityNotFoundException");

    } catch (EntityNotFoundException enfe) {

      // good

    }

    Entity bidirChildEntity = ds.get(KeyFactory.stringToKey(bidir2.getId()));

    assertNotNull(bidirChildEntity);

    assertEquals("yam", bidirChildEntity.getProperty("childVal"));

    assertEquals(KeyFactory.stringToKey(bidir2.getId()), bidirChildEntity.getKey());

    assertKeyParentEquals(pojo.getId(), bidirChildEntity, bidir2.getId());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, bidirChildEntity.getProperty("bidirChildren_INTEGER_IDX"));

    }



    Entity flightEntity = ds.get(KeyFactory.stringToKey(f2.getId()));

    assertNotNull(flightEntity);

    assertEquals("bos", flightEntity.getProperty("origin"));

    assertEquals("mia", flightEntity.getProperty("dest"));

    assertEquals("another name", flightEntity.getProperty("name"));

    assertEquals(KeyFactory.stringToKey(f2.getId()), flightEntity.getKey());

    assertKeyParentEquals(pojo.getId(), flightEntity, f2.getId());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, flightEntity.getProperty("flights_INTEGER_IDX"));

    }



    Entity hasKeyPkEntity = ds.get(hasKeyPk2.getKey());

    assertNotNull(hasKeyPkEntity);

    assertEquals("another str", hasKeyPkEntity.getProperty("str"));

    assertEquals(hasKeyPk2.getKey(), hasKeyPkEntity.getKey());

    assertKeyParentEquals(pojo.getId(), hasKeyPkEntity, hasKeyPk2.getKey());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, hasKeyPkEntity.getProperty("hasKeyPks_INTEGER_IDX"));

    }



    Entity parentEntity = ds.get(KeyFactory.stringToKey(pojo.getId()));

    assertNotNull(parentEntity);

    assertEquals(4, parentEntity.getProperties().size());

    assertEquals("yar", parentEntity.getProperty("val"));

    assertEquals(Utils.newArrayList(bidirChildEntity.getKey()), parentEntity.getProperty("bidirChildren"));

    assertEquals(Utils.newArrayList(flightEntity.getKey()), parentEntity.getProperty("flights"));

    assertEquals(Utils.newArrayList(hasKeyPkEntity.getKey()), parentEntity.getProperty("hasKeyPks"));



    assertCountsInDatastore(pojo.getClass(), bidir2.getClass(), 1, 1);

  }



  void testRemoveAtPosition(HasOneToManyJDO pojo,

      BidirectionalChildJDO bidir1,

      BidirectionalChildJDO bidir2,

      BidirectionalChildJDO bidir3, StartEnd startEnd) throws EntityNotFoundException {

    pojo.setVal("yar");

    bidir2.setChildVal("another yam");

    bidir3.setChildVal("yet another yam");

    Flight f = newFlight();

    Flight f2 = newFlight();

    Flight f3 = newFlight();

    f2.setName("another name");

    f3.setName("yet another name");

    HasKeyPkJDO hasKeyPk = new HasKeyPkJDO();

    HasKeyPkJDO hasKeyPk2 = new HasKeyPkJDO();

    HasKeyPkJDO hasKeyPk3 = new HasKeyPkJDO();

    hasKeyPk2.setStr("another str");

    hasKeyPk3.setStr("yet another str");



    pojo.addFlight(f);

    pojo.addFlight(f2);

    pojo.addFlight(f3);

    pojo.addHasKeyPk(hasKeyPk);

    pojo.addHasKeyPk(hasKeyPk2);

    pojo.addHasKeyPk(hasKeyPk3);

    pojo.addBidirChild(bidir1);

    pojo.addBidirChild(bidir2);

    pojo.addBidirChild(bidir3);



    startEnd.start();

    pm.makePersistent(pojo);

    startEnd.end();



    assertCountsInDatastore(pojo.getClass(), bidir1.getClass(), 1, 3);



    startEnd.start();

    pojo = pm.getObjectById(pojo.getClass(), pojo.getId());

    String bidir1Id = pojo.getBidirChildren().iterator().next().getId();

    String flight1Id = pojo.getFlights().iterator().next().getId();

    Key hasKeyPk1Key = pojo.getHasKeyPks().iterator().next().getKey();

    pojo.removeBidirChildAtPosition(0);

    pojo.removeFlightAtPosition(0);

    pojo.removeHasKeyPkAtPosition(0);

    startEnd.end();



    startEnd.start();

    assertNotNull(pojo.getId());

    assertEquals(2, pojo.getFlights().size());

    assertEquals(2, pojo.getHasKeyPks().size());

    assertEquals(2, pojo.getBidirChildren().size());



    Entity pojoEntity = ds.get(KeyFactory.stringToKey(pojo.getId()));

    assertNotNull(pojoEntity);

    assertEquals(4, pojoEntity.getProperties().size());

    assertEquals(Utils.newArrayList(

        KeyFactory.stringToKey(bidir2.getId()),

        KeyFactory.stringToKey(bidir3.getId())), pojoEntity.getProperty("bidirChildren"));

    assertEquals(Utils.newArrayList(

        KeyFactory.stringToKey(f2.getId()),

        KeyFactory.stringToKey(f3.getId())), pojoEntity.getProperty("flights"));

    assertEquals(Utils.newArrayList(

        hasKeyPk2.getKey(),

        hasKeyPk3.getKey()), pojoEntity.getProperty("hasKeyPks"));



    startEnd.end();



    try {

      ds.get(KeyFactory.stringToKey(bidir1Id));

      fail("expected EntityNotFoundException");

    } catch (EntityNotFoundException enfe) {

      // good

    }

    try {

      ds.get(KeyFactory.stringToKey(flight1Id));

      fail("expected EntityNotFoundException");

    } catch (EntityNotFoundException enfe) {

      // good

    }

    try {

      ds.get(hasKeyPk1Key);

      fail("expected EntityNotFoundException");

    } catch (EntityNotFoundException enfe) {

      // good

    }

    Entity bidirChildEntity = ds.get(KeyFactory.stringToKey(bidir2.getId()));

    assertNotNull(bidirChildEntity);

    assertEquals("another yam", bidirChildEntity.getProperty("childVal"));

    assertEquals(KeyFactory.stringToKey(bidir2.getId()), bidirChildEntity.getKey());

    assertKeyParentEquals(pojo.getId(), bidirChildEntity, bidir2.getId());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, bidirChildEntity.getProperty("bidirChildren_INTEGER_IDX"));

    }



    bidirChildEntity = ds.get(KeyFactory.stringToKey(bidir3.getId()));

    assertNotNull(bidirChildEntity);

    assertEquals("yet another yam", bidirChildEntity.getProperty("childVal"));

    assertEquals(KeyFactory.stringToKey(bidir3.getId()), bidirChildEntity.getKey());

    assertKeyParentEquals(pojo.getId(), bidirChildEntity, bidir2.getId());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(1L, bidirChildEntity.getProperty("bidirChildren_INTEGER_IDX"));

    }



    Entity flightEntity = ds.get(KeyFactory.stringToKey(f2.getId()));

    assertNotNull(flightEntity);

    assertEquals("bos", flightEntity.getProperty("origin"));

    assertEquals("mia", flightEntity.getProperty("dest"));

    assertEquals("another name", flightEntity.getProperty("name"));

    assertEquals(KeyFactory.stringToKey(f2.getId()), flightEntity.getKey());

    assertKeyParentEquals(pojo.getId(), flightEntity, f2.getId());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, flightEntity.getProperty("flights_INTEGER_IDX"));

    }



    flightEntity = ds.get(KeyFactory.stringToKey(f3.getId()));

    assertNotNull(flightEntity);

    assertEquals("bos", flightEntity.getProperty("origin"));

    assertEquals("mia", flightEntity.getProperty("dest"));

    assertEquals("yet another name", flightEntity.getProperty("name"));

    assertEquals(KeyFactory.stringToKey(f3.getId()), flightEntity.getKey());

    assertKeyParentEquals(pojo.getId(), flightEntity, f2.getId());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(1L, flightEntity.getProperty("flights_INTEGER_IDX"));

    }



    Entity hasKeyPkEntity = ds.get(hasKeyPk2.getKey());

    assertNotNull(hasKeyPkEntity);

    assertEquals("another str", hasKeyPkEntity.getProperty("str"));

    assertEquals(hasKeyPk2.getKey(), hasKeyPkEntity.getKey());

    assertKeyParentEquals(pojo.getId(), hasKeyPkEntity, hasKeyPk2.getKey());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, hasKeyPkEntity.getProperty("hasKeyPks_INTEGER_IDX"));

    }



    hasKeyPkEntity = ds.get(hasKeyPk3.getKey());

    assertNotNull(hasKeyPkEntity);

    assertEquals("yet another str", hasKeyPkEntity.getProperty("str"));

    assertEquals(hasKeyPk3.getKey(), hasKeyPkEntity.getKey());

    assertKeyParentEquals(pojo.getId(), hasKeyPkEntity, hasKeyPk2.getKey());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(1L, hasKeyPkEntity.getProperty("hasKeyPks_INTEGER_IDX"));

    }



    Entity parentEntity = ds.get(KeyFactory.stringToKey(pojo.getId()));

    assertNotNull(parentEntity);

    assertEquals(4, parentEntity.getProperties().size());

    assertEquals("yar", parentEntity.getProperty("val"));

    assertEquals(Utils.newArrayList(

        KeyFactory.stringToKey(bidir2.getId()),

        KeyFactory.stringToKey(bidir3.getId())), parentEntity.getProperty("bidirChildren"));

    assertEquals(Utils.newArrayList(

        KeyFactory.stringToKey(f2.getId()),

        KeyFactory.stringToKey(f3.getId())), parentEntity.getProperty("flights"));

    assertEquals(Utils.newArrayList(

        hasKeyPk2.getKey(),

        hasKeyPk3.getKey()), parentEntity.getProperty("hasKeyPks"));



    assertCountsInDatastore(pojo.getClass(), bidir2.getClass(), 1, 2);

  }



  void testAddAtPosition(HasOneToManyJDO pojo,

      BidirectionalChildJDO bidir1,

      BidirectionalChildJDO bidir2, StartEnd startEnd) throws EntityNotFoundException {

    pojo.setVal("yar");

    bidir2.setChildVal("yam");

    Flight f = newFlight();

    Flight f2 = newFlight();

    f2.setName("another name");

    HasKeyPkJDO hasKeyPk = new HasKeyPkJDO();

    HasKeyPkJDO hasKeyPk2 = new HasKeyPkJDO();

    hasKeyPk2.setStr("another str");



    pojo.addFlight(f);

    pojo.addHasKeyPk(hasKeyPk);

    pojo.addBidirChild(bidir1);



    startEnd.start();

    pm.makePersistent(pojo);

    startEnd.end();



    assertCountsInDatastore(pojo.getClass(), bidir1.getClass(), 1, 1);



    startEnd.start();

    pojo = pm.getObjectById(pojo.getClass(), pojo.getId());

    String bidir1Id = pojo.getBidirChildren().iterator().next().getId();

    String flight1Id = pojo.getFlights().iterator().next().getId();

    Key hasKeyPk1Key = pojo.getHasKeyPks().iterator().next().getKey();

    pojo.addAtPosition(0, bidir2);

    pojo.addAtPosition(0, f2);

    pojo.addAtPosition(0, hasKeyPk2);

    startEnd.end();



    startEnd.start();

    pojo = pm.getObjectById(pojo.getClass(), pojo.getId());

    assertNotNull(pojo.getId());

    assertEquals(2, pojo.getFlights().size());

    assertEquals(2, pojo.getHasKeyPks().size());

    assertEquals(2, pojo.getBidirChildren().size());

    assertNotNull(bidir2.getId());

    assertNotNull(bidir2.getParent());

    assertNotNull(f2.getId());

    assertNotNull(hasKeyPk2.getKey());



    Entity pojoEntity = ds.get(KeyFactory.stringToKey(pojo.getId()));

    assertNotNull(pojoEntity);

    assertEquals(4, pojoEntity.getProperties().size());

    assertEquals(Utils.newArrayList(

        KeyFactory.stringToKey(bidir2.getId()),

        KeyFactory.stringToKey(bidir1.getId())), pojoEntity.getProperty("bidirChildren"));

    assertEquals(Utils.newArrayList(

        KeyFactory.stringToKey(f2.getId()),

        KeyFactory.stringToKey(f.getId())), pojoEntity.getProperty("flights"));

    assertEquals(Utils.newArrayList(

        hasKeyPk2.getKey(),

        hasKeyPk.getKey()), pojoEntity.getProperty("hasKeyPks"));



    startEnd.end();



    ds.get(KeyFactory.stringToKey(bidir1Id));

    ds.get(KeyFactory.stringToKey(flight1Id));

    ds.get(hasKeyPk1Key);

    Entity bidirChildEntity = ds.get(KeyFactory.stringToKey(bidir2.getId()));

    assertNotNull(bidirChildEntity);

    assertEquals("yam", bidirChildEntity.getProperty("childVal"));

    assertEquals(KeyFactory.stringToKey(bidir2.getId()), bidirChildEntity.getKey());

    assertKeyParentEquals(pojo.getId(), bidirChildEntity, bidir2.getId());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, bidirChildEntity.getProperty("bidirChildren_INTEGER_IDX"));

    }



    Entity flightEntity = ds.get(KeyFactory.stringToKey(f2.getId()));

    assertNotNull(flightEntity);

    assertEquals("bos", flightEntity.getProperty("origin"));

    assertEquals("mia", flightEntity.getProperty("dest"));

    assertEquals("another name", flightEntity.getProperty("name"));

    assertEquals(KeyFactory.stringToKey(f2.getId()), flightEntity.getKey());

    assertKeyParentEquals(pojo.getId(), flightEntity, f2.getId());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, flightEntity.getProperty("flights_INTEGER_IDX"));

    }



    Entity hasKeyPkEntity = ds.get(hasKeyPk2.getKey());

    assertNotNull(hasKeyPkEntity);

    assertEquals("another str", hasKeyPkEntity.getProperty("str"));

    assertEquals(hasKeyPk2.getKey(), hasKeyPkEntity.getKey());

    assertKeyParentEquals(pojo.getId(), hasKeyPkEntity, hasKeyPk2.getKey());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, hasKeyPkEntity.getProperty("hasKeyPks_INTEGER_IDX"));

    }



    Entity parentEntity = ds.get(KeyFactory.stringToKey(pojo.getId()));

    assertNotNull(parentEntity);

    assertEquals(4, parentEntity.getProperties().size());

    assertEquals("yar", parentEntity.getProperty("val"));

    assertEquals(Utils.newArrayList(

        KeyFactory.stringToKey(bidir2.getId()),

        KeyFactory.stringToKey(bidir1.getId())), parentEntity.getProperty("bidirChildren"));

    assertEquals(Utils.newArrayList(

        KeyFactory.stringToKey(f2.getId()),

        KeyFactory.stringToKey(f.getId())), parentEntity.getProperty("flights"));

    assertEquals(Utils.newArrayList(

        hasKeyPk2.getKey(),

        hasKeyPk.getKey()), parentEntity.getProperty("hasKeyPks"));



    assertCountsInDatastore(pojo.getClass(), bidir2.getClass(), 1, 2);

  }



  void testUpdate_UpdateChildWithMerge(HasOneToManyJDO pojo,

      BidirectionalChildJDO bidir, StartEnd startEnd) throws EntityNotFoundException {

    Flight f = newFlight();

    HasKeyPkJDO hasKeyPk = new HasKeyPkJDO();



    pojo.addFlight(f);

    pojo.addHasKeyPk(hasKeyPk);

    pojo.addBidirChild(bidir);

    bidir.setParent(pojo);



    startEnd.start();

    pm.makePersistent(pojo);

    startEnd.end();



    assertNotNull(f.getId());

    assertNotNull(hasKeyPk.getKey());

    assertNotNull(bidir.getId());

    assertNotNull(pojo.getId());



    startEnd.start();

    f.setOrigin("yam");

    hasKeyPk.setStr("yar");

    bidir.setChildVal("yap");

    pm.makePersistent(pojo);

    startEnd.end();



    Entity flightEntity = ds.get(KeyFactory.stringToKey(f.getId()));

    assertNotNull(flightEntity);

    assertEquals("yam", flightEntity.getProperty("origin"));

    assertKeyParentEquals(pojo.getId(), flightEntity, f.getId());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, flightEntity.getProperty("flights_INTEGER_IDX"));

    }



    Entity hasKeyPkEntity = ds.get(hasKeyPk.getKey());

    assertNotNull(hasKeyPkEntity);

    assertEquals("yar", hasKeyPkEntity.getProperty("str"));

    assertKeyParentEquals(pojo.getId(), hasKeyPkEntity, hasKeyPk.getKey());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, hasKeyPkEntity.getProperty("hasKeyPks_INTEGER_IDX"));

    }



    Entity bidirEntity = ds.get(KeyFactory.stringToKey(bidir.getId()));

    assertNotNull(bidirEntity);

    assertEquals("yap", bidirEntity.getProperty("childVal"));

    assertKeyParentEquals(pojo.getId(), bidirEntity, bidir.getId());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, bidirEntity.getProperty("bidirChildren_INTEGER_IDX"));

    }



    assertCountsInDatastore(pojo.getClass(), bidir.getClass(), 1, 1);

  }



  void testUpdate_UpdateChild(HasOneToManyJDO pojo,

      BidirectionalChildJDO bidir, StartEnd startEnd) throws EntityNotFoundException {

    Flight f = newFlight();

    HasKeyPkJDO hasKeyPk = new HasKeyPkJDO();



    pojo.addFlight(f);

    pojo.addHasKeyPk(hasKeyPk);

    pojo.addBidirChild(bidir);

    bidir.setParent(pojo);



    startEnd.start();

    pm.makePersistent(pojo);

    startEnd.end();



    assertNotNull(f.getId());

    assertNotNull(hasKeyPk.getKey());

    assertNotNull(bidir.getId());

    assertNotNull(pojo.getId());



    startEnd.start();

    pojo = pm.getObjectById(pojo.getClass(), pojo.getId());

    pojo.getFlights().iterator().next().setOrigin("yam");

    pojo.getHasKeyPks().iterator().next().setStr("yar");

    pojo.getBidirChildren().iterator().next().setChildVal("yap");

    startEnd.end();



    Entity flightEntity = ds.get(KeyFactory.stringToKey(f.getId()));

    assertNotNull(flightEntity);

    assertEquals("yam", flightEntity.getProperty("origin"));

    assertKeyParentEquals(pojo.getId(), flightEntity, f.getId());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, flightEntity.getProperty("flights_INTEGER_IDX"));

    }



    Entity hasKeyPkEntity = ds.get(hasKeyPk.getKey());

    assertNotNull(hasKeyPkEntity);

    assertEquals("yar", hasKeyPkEntity.getProperty("str"));

    assertKeyParentEquals(pojo.getId(), hasKeyPkEntity, hasKeyPk.getKey());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, hasKeyPkEntity.getProperty("hasKeyPks_INTEGER_IDX"));

    }



    Entity bidirEntity = ds.get(KeyFactory.stringToKey(bidir.getId()));

    assertNotNull(bidirEntity);

    assertEquals("yap", bidirEntity.getProperty("childVal"));

    assertKeyParentEquals(pojo.getId(), bidirEntity, bidir.getId());

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, bidirEntity.getProperty("bidirChildren_INTEGER_IDX"));

    }



    assertCountsInDatastore(pojo.getClass(), bidir.getClass(), 1, 1);

  }



  void testUpdate_NullOutChildren(HasOneToManyJDO pojo,

    BidirectionalChildJDO bidir, StartEnd startEnd) throws EntityNotFoundException {

    Flight f = newFlight();

    HasKeyPkJDO hasKeyPk = new HasKeyPkJDO();



    pojo.addFlight(f);

    pojo.addHasKeyPk(hasKeyPk);

    pojo.addBidirChild(bidir);

    bidir.setParent(pojo);



    startEnd.start();

    pm.makePersistent(pojo);

    startEnd.end();



    assertCountsInDatastore(pojo.getClass(), bidir.getClass(), 1, 1);



    startEnd.start();

    String flightId = f.getId();

    Key hasKeyPkKey = hasKeyPk.getKey();

    String bidirChildId = bidir.getId();



    pojo.nullFlights();

    pojo.nullHasKeyPks();

    pojo.nullBidirChildren();

    pm.makePersistent(pojo);

    startEnd.end();



    try {

      ds.get(KeyFactory.stringToKey(flightId));

      fail("expected enfe");

    } catch (EntityNotFoundException enfe) {

      // good

    }



    try {

      ds.get(hasKeyPkKey);

      fail("expected enfe");

    } catch (EntityNotFoundException enfe) {

      // good

    }



    try {

      ds.get(KeyFactory.stringToKey(bidirChildId));

      fail("expected enfe");

    } catch (EntityNotFoundException enfe) {

      // good

    }



    Entity pojoEntity = ds.get(KeyFactory.stringToKey(pojo.getId()));

    assertEquals(4, pojoEntity.getProperties().size());

    assertTrue(pojoEntity.hasProperty("bidirChildren"));

    assertNull(pojoEntity.getProperty("bidirChildren"));

    assertTrue(pojoEntity.hasProperty("flights"));

    assertNull(pojoEntity.getProperty("flights"));

    assertTrue(pojoEntity.hasProperty("hasKeyPks"));

    assertNull(pojoEntity.getProperty("hasKeyPks"));



    assertCountsInDatastore(pojo.getClass(), bidir.getClass(), 1, 0);

  }



  void testUpdate_ClearOutChildren(HasOneToManyJDO pojo,

      BidirectionalChildJDO bidir, StartEnd startEnd) throws EntityNotFoundException {

    Flight f = newFlight();

    HasKeyPkJDO hasKeyPk = new HasKeyPkJDO();



    pojo.addFlight(f);

    pojo.addHasKeyPk(hasKeyPk);

    pojo.addBidirChild(bidir);

    bidir.setParent(pojo);



    startEnd.start();

    pm.makePersistent(pojo);

    startEnd.end();

    String flightId = f.getId();

    Key hasKeyPkId = hasKeyPk.getKey();

    String bidirChildId = bidir.getId();

    assertCountsInDatastore(pojo.getClass(), bidir.getClass(), 1, 1);



    startEnd.start();

    pojo = pm.makePersistent(pojo);

    pojo.clearFlights();

    pojo.clearHasKeyPks();

    pojo.clearBidirChildren();

    startEnd.end();



    try {

      ds.get(KeyFactory.stringToKey(flightId));

      fail("expected enfe");

    } catch (EntityNotFoundException enfe) {

      // good

    }



    try {

      ds.get(hasKeyPkId);

      fail("expected enfe");

    } catch (EntityNotFoundException enfe) {

      // good

    }



    try {

      ds.get(KeyFactory.stringToKey(bidirChildId));

      fail("expected enfe");

    } catch (EntityNotFoundException enfe) {

      // good

    }



    Entity pojoEntity = ds.get(KeyFactory.stringToKey(pojo.getId()));

    assertEquals(4, pojoEntity.getProperties().size());

    assertTrue(pojoEntity.hasProperty("bidirChildren"));

    assertNull(pojoEntity.getProperty("bidirChildren"));

    assertTrue(pojoEntity.hasProperty("flights"));

    assertNull(pojoEntity.getProperty("flights"));

    assertTrue(pojoEntity.hasProperty("hasKeyPks"));

    assertNull(pojoEntity.getProperty("hasKeyPks"));



    assertCountsInDatastore(pojo.getClass(), bidir.getClass(), 1, 0);

  }



  void testFindWithOrderBy(Class<? extends HasOneToManyWithOrderByJDO> pojoClass, StartEnd startEnd)

      throws EntityNotFoundException {

    getExecutionContext().getNucleusContext().getPersistenceConfiguration().setProperty(

        "datanucleus.appengine.allowMultipleRelationsOfSameType", true);

    Entity pojoEntity = new Entity(pojoClass.getSimpleName());

    ds.put(pojoEntity);



    Entity flightEntity1 = newFlightEntity(pojoEntity.getKey(), "bos1", "mia2", "name 1");

    flightEntity1.setProperty("flightsByOrigAndDest_INTEGER_IDX", 0);

    flightEntity1.setProperty("flightsByIdAndOrig_INTEGER_IDX", 0);

    flightEntity1.setProperty("flightsByOrigAndId_INTEGER_IDX", 0);

    ds.put(flightEntity1);



    Entity flightEntity2 = newFlightEntity(pojoEntity.getKey(), "bos2", "mia2", "name 2");

    flightEntity2.setProperty("flightsByOrigAndDest_INTEGER_IDX", 1);

    flightEntity2.setProperty("flightsByIdAndOrig_INTEGER_IDX", 1);

    flightEntity2.setProperty("flightsByOrigAndId_INTEGER_IDX", 1);

    ds.put(flightEntity2);



    Entity flightEntity3 = newFlightEntity(pojoEntity.getKey(), "bos1", "mia1", "name 0");

    flightEntity3.setProperty("flightsByOrigAndDest_INTEGER_IDX", 2);

    flightEntity3.setProperty("flightsByIdAndOrig_INTEGER_IDX", 2);

    flightEntity3.setProperty("flightsByOrigAndId_INTEGER_IDX", 2);

    ds.put(flightEntity3);



    Entity explicitIndexEntity1 =

        new Entity(HasExplicitIndexColumnJDO.class.getSimpleName(), pojoEntity.getKey());

    explicitIndexEntity1.setProperty("index", 3);

    ds.put(explicitIndexEntity1);



    Entity explicitIndexEntity2 =

        new Entity(HasExplicitIndexColumnJDO.class.getSimpleName(), pojoEntity.getKey());

    explicitIndexEntity2.setProperty("index", 2);

    ds.put(explicitIndexEntity2);



    Entity explicitIndexEntity3 =

        new Entity(HasExplicitIndexColumnJDO.class.getSimpleName(), pojoEntity.getKey());

    explicitIndexEntity3.setProperty("index", 1);

    ds.put(explicitIndexEntity3);



    pojoEntity.setProperty("flightsByOrigAndDest", Utils.newArrayList(flightEntity2.getKey(), flightEntity3.getKey(), flightEntity1.getKey()));

    pojoEntity.setProperty("flightsByIdAndOrig", Utils.newArrayList(flightEntity3.getKey(), flightEntity2.getKey(), flightEntity1.getKey()));

    pojoEntity.setProperty("flightsByOrigAndId", Utils.newArrayList(flightEntity2.getKey(), flightEntity1.getKey(), flightEntity3.getKey()));

    pojoEntity.setProperty("hasIndexColumn", Utils.newArrayList(explicitIndexEntity3.getKey(), explicitIndexEntity2.getKey(), explicitIndexEntity1.getKey()));

    ds.put(pojoEntity);



    startEnd.start();

    HasOneToManyWithOrderByJDO pojo = pm.getObjectById(

        pojoClass, KeyFactory.keyToString(pojoEntity.getKey()));

    assertNotNull(pojo);

    assertNotNull(pojo.getFlightsByOrigAndDest());

    assertEquals(3, pojo.getFlightsByOrigAndDest().size());

    assertEquals("name 2", pojo.getFlightsByOrigAndDest().get(0).getName());

    assertEquals("name 0", pojo.getFlightsByOrigAndDest().get(1).getName());

    assertEquals("name 1", pojo.getFlightsByOrigAndDest().get(2).getName());



    assertNotNull(pojo.getFlightsByIdAndOrig());

    assertEquals(3, pojo.getFlightsByIdAndOrig().size());

    assertEquals("name 0", pojo.getFlightsByIdAndOrig().get(0).getName());

    assertEquals("name 2", pojo.getFlightsByIdAndOrig().get(1).getName());

    assertEquals("name 1", pojo.getFlightsByIdAndOrig().get(2).getName());



    assertNotNull(pojo.getFlightsByOrigAndId());

    assertEquals(3, pojo.getFlightsByOrigAndId().size());

    assertEquals("name 2", pojo.getFlightsByOrigAndId().get(0).getName());

    assertEquals("name 1", pojo.getFlightsByOrigAndId().get(1).getName());

    assertEquals("name 0", pojo.getFlightsByOrigAndId().get(2).getName());



    assertNotNull(pojo.getHasIndexColumn());

    assertEquals(3, pojo.getHasIndexColumn().size());

    assertEquals(explicitIndexEntity3.getKey(), pojo.getHasIndexColumn().get(0).getId());

    assertEquals(explicitIndexEntity2.getKey(), pojo.getHasIndexColumn().get(1).getId());

    assertEquals(explicitIndexEntity1.getKey(), pojo.getHasIndexColumn().get(2).getId());



    startEnd.end();

  }



  void testFind(Class<? extends HasOneToManyJDO> pojoClass,

      Class<? extends BidirectionalChildJDO> bidirClass, StartEnd startEnd) throws EntityNotFoundException {

    Entity pojoEntity = new Entity(pojoClass.getSimpleName());

    ds.put(pojoEntity);



    Entity flightEntity = newFlightEntity(pojoEntity.getKey(), "bos1", "mia2", "name 1");

    ds.put(flightEntity);



    Entity hasKeyPkEntity = new Entity(HasKeyPkJDO.class.getSimpleName(), pojoEntity.getKey());

    hasKeyPkEntity.setProperty("str", "yar");

    hasKeyPkEntity.setProperty("hasKeyPks_INTEGER_IDX", 1);

    ds.put(hasKeyPkEntity);



    Entity bidirEntity = new Entity(bidirClass.getSimpleName(), pojoEntity.getKey());

    bidirEntity.setProperty("childVal", "yap");

    bidirEntity.setProperty("bidirChildren_INTEGER_IDX", 1);

    ds.put(bidirEntity);



    pojoEntity.setProperty("bidirChildren", Utils.newArrayList(bidirEntity.getKey()));

    pojoEntity.setProperty("hasKeyPks", Utils.newArrayList(hasKeyPkEntity.getKey()));

    pojoEntity.setProperty("flights", Utils.newArrayList(flightEntity.getKey()));

    ds.put(pojoEntity);



    startEnd.start();

    HasOneToManyJDO pojo =

        pm.getObjectById(pojoClass, KeyFactory.keyToString(pojoEntity.getKey()));

    assertNotNull(pojo);

    assertNotNull(pojo.getFlights());

    assertEquals(1, pojo.getFlights().size());

    assertEquals("bos1", pojo.getFlights().iterator().next().getOrigin());

    assertNotNull(pojo.getHasKeyPks());

    assertEquals(1, pojo.getHasKeyPks().size());

    assertEquals("yar", pojo.getHasKeyPks().iterator().next().getStr());

    assertNotNull(pojo.getBidirChildren());

    assertEquals(1, pojo.getBidirChildren().size());

    assertEquals("yap", pojo.getBidirChildren().iterator().next().getChildVal());

    assertEquals(pojo, pojo.getBidirChildren().iterator().next().getParent());

    startEnd.end();

  }



  void testQuery(Class<? extends HasOneToManyJDO> pojoClass,

      Class<? extends BidirectionalChildJDO> bidirClass, StartEnd startEnd) throws EntityNotFoundException {

    Entity pojoEntity = new Entity(pojoClass.getSimpleName());

    ds.put(pojoEntity);



    Entity flightEntity = newFlightEntity(pojoEntity.getKey(), "bos", "mia2", "name");

    ds.put(flightEntity);



    Entity hasKeyPkEntity = new Entity(HasKeyPkJDO.class.getSimpleName(), pojoEntity.getKey());

    hasKeyPkEntity.setProperty("str", "yar");

    hasKeyPkEntity.setProperty("hasKeyPks_INTEGER_IDX", 1);

    ds.put(hasKeyPkEntity);



    Entity bidirEntity = new Entity(bidirClass.getSimpleName(), pojoEntity.getKey());

    bidirEntity.setProperty("childVal", "yap");

    bidirEntity.setProperty("bidirChildren_INTEGER_IDX", 1);

    ds.put(bidirEntity);



    pojoEntity.setProperty("bidirChildren", Utils.newArrayList(bidirEntity.getKey()));

    pojoEntity.setProperty("hasKeyPks", Utils.newArrayList(hasKeyPkEntity.getKey()));

    pojoEntity.setProperty("flights", Utils.newArrayList(flightEntity.getKey()));

    ds.put(pojoEntity);



    javax.jdo.Query q = pm.newQuery(

        "select from " + pojoClass.getName() + " where id == key parameters String key");

    startEnd.start();

    @SuppressWarnings("unchecked")

    List<HasOneToManyListJDO> result =

        (List<HasOneToManyListJDO>) q.execute(KeyFactory.keyToString(pojoEntity.getKey()));

    assertEquals(1, result.size());

    HasOneToManyJDO pojo = result.get(0);

    assertNotNull(pojo.getFlights());

    assertEquals(1, pojo.getFlights().size());

    assertEquals("bos", pojo.getFlights().iterator().next().getOrigin());

    assertEquals(1, pojo.getFlights().size());

    assertNotNull(pojo.getHasKeyPks());

    assertEquals(1, pojo.getHasKeyPks().size());

    assertEquals("yar", pojo.getHasKeyPks().iterator().next().getStr());

    assertNotNull(pojo.getBidirChildren());

    assertEquals(1, pojo.getBidirChildren().size());

    assertEquals("yap", pojo.getBidirChildren().iterator().next().getChildVal());

    assertEquals(pojo, pojo.getBidirChildren().iterator().next().getParent());

    startEnd.end();

  }



  void testChildFetchedLazily(Class<? extends HasOneToManyJDO> pojoClass,

      Class<? extends BidirectionalChildJDO> bidirClass) throws Exception {

    DatastoreServiceConfig config = getStoreManager().getDefaultDatastoreServiceConfigForReads();

    tearDown();

    DatastoreService mockDatastore = EasyMock.createMock(DatastoreService.class);

    DatastoreService original = DatastoreServiceFactoryInternal.getDatastoreService(config);

    DatastoreServiceFactoryInternal.setDatastoreService(mockDatastore);

    try {

      setUp();



      Entity pojoEntity = new Entity(pojoClass.getSimpleName());

      ds.put(pojoEntity);



      Entity FlightEntity = newFlightEntity(pojoEntity.getKey(), "bos", "mia", "name");

      ds.put(FlightEntity);



      Entity hasKeyPkEntity = new Entity(HasKeyPkJDO.class.getSimpleName(), pojoEntity.getKey());

      hasKeyPkEntity.setProperty("str", "yar");

      ds.put(hasKeyPkEntity);



      Entity bidirEntity = new Entity(bidirClass.getSimpleName(), pojoEntity.getKey());

      bidirEntity.setProperty("childVal", "yap");

      ds.put(bidirEntity);



      Transaction txn = EasyMock.createMock(Transaction.class);

      EasyMock.expect(txn.getId()).andReturn("1").times(2);

      txn.commit();

      EasyMock.expectLastCall();

      EasyMock.replay(txn);

      EasyMock.expect(mockDatastore.beginTransaction(EasyMock.isA(TransactionOptions.class))).andReturn(txn);

      // the only get we're going to perform is for the pojo

      EasyMock.expect(mockDatastore.get(txn, pojoEntity.getKey())).andReturn(pojoEntity);

      EasyMock.replay(mockDatastore);



      beginTxn();

      HasOneToManyJDO pojo =

          pm.getObjectById(pojoClass, KeyFactory.keyToString(pojoEntity.getKey()));

      assertNotNull(pojo);

      pojo.getId();

      commitTxn();

    } finally {

      DatastoreServiceFactoryInternal.setDatastoreService(original);

    }

    EasyMock.verify(mockDatastore);

  }



  void testDeleteParentDeletesChild(Class<? extends HasOneToManyJDO> pojoClass,

      Class<? extends BidirectionalChildJDO> bidirClass, StartEnd startEnd) throws Exception {

    Entity pojoEntity = new Entity(pojoClass.getSimpleName());

    ds.put(pojoEntity);



    Entity flightEntity = newFlightEntity(pojoEntity.getKey(), "bos", "mia", "name");

    flightEntity.setProperty("flights_INTEGER_IDX", 1);

    ds.put(flightEntity);



    Entity hasKeyPkEntity = new Entity(HasKeyPkJDO.class.getSimpleName(), pojoEntity.getKey());

    hasKeyPkEntity.setProperty("str", "yar");

    hasKeyPkEntity.setProperty("hasKeyPks_INTEGER_IDX", 1);

    ds.put(hasKeyPkEntity);



    Entity bidirEntity = new Entity(bidirClass.getSimpleName(), pojoEntity.getKey());

    bidirEntity.setProperty("childVal", "yap");

    bidirEntity.setProperty("bidirChildren_INTEGER_IDX", 1);

    ds.put(bidirEntity);



    pojoEntity.setProperty("bidirChildren", Utils.newArrayList(bidirEntity.getKey()));

    pojoEntity.setProperty("hasKeyPks", Utils.newArrayList(hasKeyPkEntity.getKey()));

    pojoEntity.setProperty("flights", Utils.newArrayList(flightEntity.getKey()));

    ds.put(pojoEntity);



    startEnd.start();

    HasOneToManyJDO pojo = pm.getObjectById(pojoClass, KeyFactory.keyToString(pojoEntity.getKey()));

    pm.deletePersistent(pojo);

    startEnd.end();

    assertCountsInDatastore(pojoClass, bidirClass, 0, 0);

  }



  void testRemoveAll(HasOneToManyJDO pojo, BidirectionalChildJDO bidir1,

                     BidirectionalChildJDO bidir2, BidirectionalChildJDO bidir3, StartEnd startEnd)

      throws EntityNotFoundException {

    Flight f1 = new Flight();

    Flight f2 = new Flight();

    Flight f3 = new Flight();

    pojo.addFlight(f1);

    pojo.addFlight(f2);

    pojo.addFlight(f3);



    pojo.addBidirChild(bidir1);

    pojo.addBidirChild(bidir2);

    pojo.addBidirChild(bidir3);



    startEnd.start();

    pm.makePersistent(pojo);

    startEnd.end();

    startEnd.start();

    pojo = pm.makePersistent(pojo);

    String f2Id = f2.getId();

    String bidir2Id = bidir2.getId();

    pojo.removeFlights(Collections.singleton(f2));

    pojo.removeBidirChildren(Collections.singleton(bidir2));

    startEnd.end();

    startEnd.start();



    pojo = pm.getObjectById(pojo.getClass(), pojo.getId());



    assertEquals(2, pojo.getFlights().size());

    Set<String> flightIds = Utils.newHashSet(f1.getId(), f2.getId(), f3.getId());

    for (Flight f : pojo.getFlights()) {

      flightIds.remove(f.getId());

    }

    assertEquals(1, flightIds.size());

    assertEquals(f2.getId(), flightIds.iterator().next());



    assertEquals(2, pojo.getBidirChildren().size());

    Set<String> bidirIds = Utils.newHashSet(bidir1.getId(), bidir2.getId(), bidir3.getId());

    for (BidirectionalChildJDO b : pojo.getBidirChildren()) {

      bidirIds.remove(b.getId());

    }

    assertEquals(1, bidirIds.size());

    assertEquals(bidir2.getId(), bidirIds.iterator().next());

    startEnd.end();



    Entity flightEntity1 = ds.get(KeyFactory.stringToKey(f1.getId()));

    Entity flightEntity3 = ds.get(KeyFactory.stringToKey(f3.getId()));

    Entity bidirEntity1 = ds.get(KeyFactory.stringToKey(bidir1.getId()));

    Entity bidirEntity3 = ds.get(KeyFactory.stringToKey(bidir3.getId()));

    if (isIndexed() && getStorageVersion(pm) == StorageVersion.PARENTS_DO_NOT_REFER_TO_CHILDREN) {

      assertEquals(0L, flightEntity1.getProperty("flights_INTEGER_IDX"));

      assertEquals(1L, flightEntity3.getProperty("flights_INTEGER_IDX"));

      assertEquals(0L, bidirEntity1.getProperty("bidirChildren_INTEGER_IDX"));

      assertEquals(1L, bidirEntity3.getProperty("bidirChildren_INTEGER_IDX"));

    }

    try {

      ds.get(KeyFactory.stringToKey(f2Id));

      fail("expected enfe");

    } catch (EntityNotFoundException enfe) {

      // good

    }

    try {

      ds.get(KeyFactory.stringToKey(bidir2Id));

      fail("expected enfe");

    } catch (EntityNotFoundException enfe) {

      // good

    }

  }



  void testRemoveAll_LongPkOnParent(HasOneToManyLongPkJDO pojo, BidirectionalChildLongPkJDO bidir1,

                     BidirectionalChildLongPkJDO bidir2, BidirectionalChildLongPkJDO bidir3, StartEnd startEnd)

      throws EntityNotFoundException {

    Flight f1 = new Flight();

    Flight f2 = new Flight();

    Flight f3 = new Flight();

    pojo.addFlight(f1);

    pojo.addFlight(f2);

    pojo.addFlight(f3);



    pojo.addBidirChild(bidir1);

    pojo.addBidirChild(bidir2);

    pojo.addBidirChild(bidir3);



    startEnd.start();

    pm.makePersistent(pojo);

    startEnd.end();

    startEnd.start();

    pojo = pm.makePersistent(pojo);

    String f2Id = f2.getId();

    String bidir2Id = bidir2.getId();

    pojo.removeFlights(Collections.singleton(f2));

    pojo.removeBidirChildren(Collections.singleton(bidir2));

    startEnd.end();

    startEnd.start();



    pojo = pm.getObjectById(pojo.getClass(), pojo.getId());

    assertEquals(2, pojo.getFlights().size());

    Set<String…