/*
 * 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
 *
 *     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.jdo.tck.api.persistencemanager.lifecycle;

import java.util.Collection;
import java.util.HashSet;
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> MakeTransactional Prior To a Transaction Rolledback <br>
 * <B>Keywords:</B> <br>
 * <B>Assertion IDs:</B> A12.5.7-23 <br>
 * <B>Assertion Description: </B> If the transaction in which an instance is made transactional (by
 * calling PersistenceManager.makeTransactional or makeTransactionalAll) is rolled back, then the
 * transient instance takes its values as of the beginning of the transaction, if the call to
 * makeTransactional was made prior to the beginning of the current transaction.
 */
public class MakeTransactionalPriorToTransactionRolledback extends PersistenceManagerTest {

  /** */
  private static final String ASSERTION_FAILED =
      "Assertion A12.5.7-23 (MakeTransactionalPriorToTransactionRolledback) 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(MakeTransactionalPriorToTransactionRolledback.class);
  }

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

  private final Collection<PCPoint> col1 = new HashSet<>();
  private final Collection<PCPoint> col2 = new HashSet<>();

  /** */
  public void testTransactionalInst() {
    pm = getPM();
    pm.currentTransaction().setRestoreValues(true);

    createObjects();
    runTestMakeTransactionalPriorToTransactionRolledback(pm);
    runTestMakeTransactionalPriorToTransactionRolledbackAll1(pm);
    runTestMakeTransactionalPriorToTransactionRolledbackAll2(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);

    col1.add(p2);
    col1.add(p3);

    col2.add(p4);
    col2.add(p5);
  }

  public void runTestMakeTransactionalPriorToTransactionRolledback(PersistenceManager pm) {
    Transaction tx = pm.currentTransaction();
    try {
      if (debug) logger.debug(" ** in runTestMakeTransactionalPriorToTransactionRolledback()");
      p1.setY(Integer.valueOf(50));

      tx.begin();
      p1.setY(Integer.valueOf(100));
      pm.makeTransactional(p1);
      p1.setY(Integer.valueOf(200));
      tx.rollback();

      p1.setY(Integer.valueOf(300));
      tx.begin();
      p1.setY(Integer.valueOf(400));
      tx.rollback();

      if (p1.getY().intValue() != 300) {
        fail(
            ASSERTION_FAILED,
            "wrong value of p1.y, expected: 300, actual: " + p1.getY().intValue());
      }
    } finally {
      if ((tx != null) && tx.isActive()) tx.rollback();
    }
  }

  /** */
  private void runTestMakeTransactionalPriorToTransactionRolledbackAll1(PersistenceManager pm) {
    Transaction tx = pm.currentTransaction();
    try {
      if (debug) logger.debug(" ** in runTestMakeTransactionalPriorToTransactionRolledbackAll1()");
      p2.setY(Integer.valueOf(50));
      p3.setY(Integer.valueOf(50));
      tx.begin();
      p2.setY(Integer.valueOf(100));
      p3.setY(Integer.valueOf(100));
      pm.makeTransactionalAll(col1);
      p2.setY(Integer.valueOf(200));
      p3.setY(Integer.valueOf(200));
      tx.rollback();

      p2.setY(Integer.valueOf(300));
      p3.setY(Integer.valueOf(300));
      tx.begin();
      p2.setY(Integer.valueOf(400));
      p3.setY(Integer.valueOf(400));
      tx.rollback();

      if (p2.getY().intValue() != 300) {
        fail(
            ASSERTION_FAILED,
            "wrong value of p2.y, expected: 300, actual: " + p2.getY().intValue());
      }
      if (p3.getY().intValue() != 300) {
        fail(
            ASSERTION_FAILED,
            "wrong value of p3.y, expected: 300, actual: " + p3.getY().intValue());
      }
    } finally {
      if ((tx != null) && tx.isActive()) tx.rollback();
    }
  }

  /**
   * @param pm the PersistenceManager
   */
  public void runTestMakeTransactionalPriorToTransactionRolledbackAll2(PersistenceManager pm) {
    Transaction tx = pm.currentTransaction();
    try {
      if (debug) logger.debug(" ** in runTestMakeTransactionalPriorToTransactionRolledbackAll2()");
      p2.setY(Integer.valueOf(50));
      p3.setY(Integer.valueOf(50));
      tx.begin();
      p2.setY(Integer.valueOf(100));
      p3.setY(Integer.valueOf(100));

      Object[] objArray = col2.toArray();
      pm.makeTransactionalAll(objArray);
      p2.setY(Integer.valueOf(200));
      p3.setY(Integer.valueOf(200));
      tx.rollback();

      p2.setY(Integer.valueOf(300));
      p3.setY(Integer.valueOf(300));
      tx.begin();
      p2.setY(Integer.valueOf(400));
      p3.setY(Integer.valueOf(400));
      tx.rollback();

      if (p2.getY().intValue() != 300) {
        fail(
            ASSERTION_FAILED,
            "wrong value of p2.y, expected: 300, actual: " + p2.getY().intValue());
      }
      if (p3.getY().intValue() != 300) {
        fail(
            ASSERTION_FAILED,
            "wrong value of p3.y, expected: 300, actual: " + p3.getY().intValue());
      }
    } finally {
      if ((tx != null) && tx.isActive()) tx.rollback();
    }
  }
}
