content: /*
   * 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.ambari.server.orm;
  
  import static org.apache.ambari.server.orm.AmbariJpaLocalTxnInterceptor.rollbackIfNecessary;
  import static org.easymock.EasyMock.expect;
  
  import java.io.FileNotFoundException;
  import java.io.IOException;
  
  import jakarta.persistence.EntityTransaction;
  
  import org.easymock.EasyMockSupport;
  import org.junit.Assert;
  import org.junit.Test;
  
  import com.google.inject.persist.Transactional;
  
  public class AmbariJpaLocalTxnInterceptorTest extends EasyMockSupport {
  
    @Test
    public void canBeCommittedIfExceptionsToBeRolledBackOnIsEmpty() {
      Transactional transactional = createNiceMock(Transactional.class);
      EntityTransaction transaction = createStrictMock(EntityTransaction.class);
  
      expect(transaction.getRollbackOnly()).andReturn(false);
      expect(transactional.rollbackOn()).andReturn(asArray());
  
      replayAll();
  
      boolean canCommit = rollbackIfNecessary(transactional, new RuntimeException(), transaction);
      Assert.assertTrue("Should be allowed to commit, since rollbackOn clause is empty", canCommit);
  
      verifyAll();
    }
  
    @Test
    public void canBeCommittedIfUnknownExceptionThrown() {
      Transactional transactional = createNiceMock(Transactional.class);
      EntityTransaction transaction = createStrictMock(EntityTransaction.class);
  
      expect(transaction.getRollbackOnly()).andReturn(false);
      expect(transactional.rollbackOn()).andReturn(asArray(IllegalArgumentException.class));
  
      replayAll();
  
      boolean canCommit = rollbackIfNecessary(transactional, new RuntimeException(), transaction);
      Assert.assertTrue("Should be allowed to commit, exception thrown does not match rollbackOn clause", canCommit);
  
      verifyAll();
    }
  
    @Test
    public void rolledBackForKnownException() {
      Transactional transactional = createNiceMock(Transactional.class);
      EntityTransaction transaction = createStrictMock(EntityTransaction.class);
  
      expect(transaction.getRollbackOnly()).andReturn(false);
      expect(transactional.rollbackOn()).andReturn(asArray(NullPointerException.class, IllegalArgumentException.class));
      expect(transactional.ignore()).andReturn(asArray());
      transaction.rollback();
  
      replayAll();
  
      boolean canCommit = rollbackIfNecessary(transactional, new IllegalArgumentException("rolling back"), transaction);
      Assert.assertFalse("Should be rolled back, since exception matches rollbackOn clause", canCommit);
  
      verifyAll();
    }
  
    @Test
    public void rolledBackForSubclassOfKnownException() {
      Transactional transactional = createNiceMock(Transactional.class);
      EntityTransaction transaction = createStrictMock(EntityTransaction.class);
  
      expect(transaction.getRollbackOnly()).andReturn(false);
      expect(transactional.rollbackOn()).andReturn(asArray(RuntimeException.class));
      expect(transactional.ignore()).andReturn(asArray());
      transaction.rollback();
  
      replayAll();
  
      boolean canCommit = rollbackIfNecessary(transactional, new IllegalArgumentException("rolling back"), transaction);
      Assert.assertFalse("Should be rolled back, since exception is subclass of the one in rollbackOn clause", canCommit);
  
      verifyAll();
    }
  
    @Test
    public void canBeCommittedIfIgnoredExceptionThrown() {
      Transactional transactional = createNiceMock(Transactional.class);
      EntityTransaction transaction = createStrictMock(EntityTransaction.class);
  
      expect(transaction.getRollbackOnly()).andReturn(false);
      expect(transactional.rollbackOn()).andReturn(asArray(IllegalArgumentException.class));
      expect(transactional.ignore()).andReturn(asArray(NumberFormatException.class));
  
      replayAll();
  
      boolean canCommit = rollbackIfNecessary(transactional, new NumberFormatException("rolling back"), transaction);
      Assert.assertTrue("Should be allowed to commit, since ignored exception was thrown", canCommit);
  
      verifyAll();
    }
  
    @Test
    public void canBeCommittedIfSubclassOfIgnoredExceptionThrown() {
      Transactional transactional = createNiceMock(Transactional.class);
      EntityTransaction transaction = createStrictMock(EntityTransaction.class);
  
      expect(transaction.getRollbackOnly()).andReturn(false);
      expect(transactional.rollbackOn()).andReturn(asArray(Exception.class));
      expect(transactional.ignore()).andReturn(asArray(IOException.class));
  
      replayAll();
  
      boolean canCommit = rollbackIfNecessary(transactional, new FileNotFoundException("rolling back"), transaction);
      Assert.assertTrue("Should be allowed to commit, since subclass of ignored exception was thrown", canCommit);
  
      verifyAll();
    }
  
    @Test
    public void rolledBackIfTransactionMarkedRollbackOnly() {
      Transactional transactional = createNiceMock(Transactional.class);
      EntityTransaction transaction = createStrictMock(EntityTransaction.class);
  
      expect(transaction.getRollbackOnly()).andReturn(true);
      transaction.rollback();
  
      replayAll();
  
      boolean canCommit = rollbackIfNecessary(transactional, null, transaction);
      Assert.assertFalse("Should be rolled back, since transaction was marked rollback-only", canCommit);
  
      verifyAll();
    }
  
    @SafeVarargs
    private static Class<? extends Exception>[] asArray(Class<? extends Exception>... exceptions) {
      return exceptions;
    }
  
  }
  