/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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
<<<<<<< Updated upstream
 *
 *     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
=======
 * 
 *     https://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 
>>>>>>> Stashed changes
 * limitations under the License.
 */

package org.apache.jdo.tck.api.persistencemanager.lifecycle;

import java.util.Collection;
import java.util.HashSet;
import javax.jdo.JDOUserException;
import javax.jdo.PersistenceManager;
import javax.jdo.Transaction;
import org.apache.jdo.tck.api.persistencemanager.PersistenceManagerTest;
import org.apache.jdo.tck.pc.mylib.PCPoint;
import org.apache.jdo.tck.util.BatchTestRunner;

/**
 * <B>Title:</B> make instance persistent <br>
 * <B>Keywords:</B> persistencemanager <br>
 * <B>Assertion IDs:</B> A12.5.7-6A <br>
 * <B>Assertion Description: </B> The method <code>PersistenceManager.makePersistent</code> and
 * <code>makePersistentAll</code> makes a transient instance(s) persistent directly. It must be
 * called in the context of an active transaction, or a <code>JDOUserException</code> is thrown.
 */
public class MakePersistent extends PersistenceManagerTest {

  /** */
  private static final String ASSERTION_FAILED = "Assertion A12.5.7-6A (MakePersistent) failed: ";

  /**
   * The <code>main</code> is called when the class is directly executed from the command line.
   *
   * @param args The arguments passed to the program.
   */
  public static void main(String[] args) {
    BatchTestRunner.run(MakePersistent.class);
  }

  private PCPoint p1 = null;
  private PCPoint p2 = null;
  private PCPoint p3 = null;
  private PCPoint p4 = null;
  private PCPoint p5 = null;

  /** Overrides parent method to avoid querying for classes that may not exist in the schema */
  @Override
  protected void localSetUp() {
    addTearDownClass(PCPoint.class);
  }

  /** */
  public void testMakePersistent() {
    pm = getPM();

    createObjects();

    /* positive tests */
    runTestMakePersistent1(pm);
    runTestMakePersistent2(pm);
    runTestMakePersistent3(pm);

    /* negative tests */
    runTestMakePersistent4(pm);
    runTestMakePersistent5(pm);
    runTestMakePersistent6(pm);

    pm.close();
    pm = null;
  }

  /** */
  private void createObjects() {
    p1 = new PCPoint(1, 3);
    p2 = new PCPoint(2, 4);
    p3 = new PCPoint(3, 5);
    p4 = new PCPoint(4, 6);
    p5 = new PCPoint(5, 7);
  }

  /* test makePersistent (Object pc) */
  private void runTestMakePersistent1(PersistenceManager pm) {
    Transaction tx = pm.currentTransaction();
    try {
      tx.begin();
      pm.makePersistent(p1);
      tx.commit();
      tx = null;
      if (debug) logger.debug(" \nPASSED in runTestMakePersistent1()");
    } finally {
      if ((tx != null) && tx.isActive()) tx.rollback();
    }
  }

  /* test makePersistentAll (Collection pcs) */
  private void runTestMakePersistent2(PersistenceManager pm) {
    Transaction tx = pm.currentTransaction();
    try {
      tx.begin();

      Collection<PCPoint> col1 = new HashSet<>();
      col1.add(p2);
      col1.add(p3);

      pm.makePersistentAll(col1);
      tx.commit();
      tx = null;
      if (debug) logger.debug(" \nPASSED in runTestMakePersistent2()");
    } finally {
      if ((tx != null) && tx.isActive()) tx.rollback();
    }
  }

  /* test makePersistentAll (Object[] o) */
  private void runTestMakePersistent3(PersistenceManager pm) {
    Transaction tx = pm.currentTransaction();
    try {
      tx = pm.currentTransaction();
      tx.begin();

      Collection<PCPoint> col1 = new HashSet<>();
      col1.add(p1);
      col1.add(p2);

      Object[] obj1 = col1.toArray();

      pm.makePersistentAll(obj1);
      tx.commit();
      tx = null;
      if (debug) logger.debug(" \nPASSED in runTestMakePersistent3()");
    } finally {
      if ((tx != null) && tx.isActive()) tx.rollback();
    }
  }

  /* test makePersistent (Object pc) */
  private void runTestMakePersistent4(PersistenceManager pm) {
    Transaction tx = pm.currentTransaction();
    if (tx.isActive()) tx.rollback();

    PCPoint np1 = new PCPoint(1, 3);
    try {
      pm.makePersistent(np1);
      fail(ASSERTION_FAILED, "pm.makePersistent outside of a tx should throw a JDOUserException");
    } catch (JDOUserException ex) {
      // expected expected
    }
  }

  /* test makePersistentAll (Collection pcs) */
  private void runTestMakePersistent5(PersistenceManager pm) {
    Transaction tx = pm.currentTransaction();
    if (tx.isActive()) tx.rollback();

    PCPoint np1 = new PCPoint(3, 3);
    PCPoint np2 = new PCPoint(4, 4);

    Collection<PCPoint> col1 = new HashSet<>();
    col1.add(np1);
    col1.add(np2);

    try {
      pm.makePersistentAll(col1);
      fail(
          ASSERTION_FAILED,
          "pm.makePersistentAll(Collection) outside of a tx should throw a JDOUserException");
    } catch (JDOUserException ex) {
      // expected expected
    }
  }

  /* test makePersistentAll (Object[] o) */
  private void runTestMakePersistent6(PersistenceManager pm) {
    Transaction tx = pm.currentTransaction();
    if (tx.isActive()) tx.rollback();

    PCPoint np1 = new PCPoint(3, 3);
    PCPoint np2 = new PCPoint(4, 4);

    Collection<PCPoint> col1 = new HashSet<>();
    col1.add(np1);
    col1.add(np2);

    Object[] obj1 = col1.toArray();

    try {
      pm.makePersistentAll(obj1);
      fail(
          ASSERTION_FAILED,
          "pm.makePersistentAll(Object[]) outside of a tx should throw a JDOUserException");
    } catch (JDOUserException ex) {
      // expected expected
    }
  }
}
