/*
 * 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.query.jdoql.methods;

import java.time.LocalTime;
import java.util.ArrayList;
import java.util.List;
import javax.jdo.PersistenceManager;
import javax.jdo.Query;
import javax.jdo.Transaction;
import org.apache.jdo.tck.pc.query.LocalTimeSample;
import org.apache.jdo.tck.query.QueryTest;
import org.apache.jdo.tck.util.BatchTestRunner;

/**
 * <B>Title:</B> Supported LocalTime methods. <br>
 * <B>Keywords:</B> query <br>
 * <B>Assertion ID:</B> A14.6.2-60. <br>
 * <B>Assertion Description: </B> New supported LocalTime methods:
 *
 * <ul>
 *   <li>getHour()
 *   <li>getMinute()
 *   <li>getSecond()
 * </ul>
 */
public class SupportedLocalTimeMethods extends QueryTest {

  /** */
  private static final String ASSERTION_FAILED =
      "Assertion A14.6.2-60 (SupportedLocalTimeMethods) failed: ";

  /** */
  private Object oidOfLocalTime1;

  /** */
  private Object oidOfLocalTime2;

  /**
   * 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(SupportedLocalTimeMethods.class);
  }

  /** */
  public void testHour() {
    final String filter = "localTime.getHour() == 14";
    PersistenceManager pm = getPM();
    Transaction tx = pm.currentTransaction();
    try {
      tx.begin();
      List<LocalTimeSample> expectedResult = new ArrayList<>();
      expectedResult.add((LocalTimeSample) pm.getObjectById(oidOfLocalTime1));

      Query<LocalTimeSample> q = pm.newQuery(LocalTimeSample.class, filter);
      List<LocalTimeSample> results = q.executeList();
      checkQueryResultWithoutOrder(ASSERTION_FAILED, filter, results, expectedResult);
      tx.commit();
      tx = null;
    } finally {
      if ((tx != null) && tx.isActive()) tx.rollback();
    }
  }

  /** */
  public void testMinute() {
    final String filter = "localTime.getMinute() == 22";
    PersistenceManager pm = getPM();
    Transaction tx = pm.currentTransaction();
    try {
      tx.begin();
      List<LocalTimeSample> expectedResult = new ArrayList<>();
      expectedResult.add((LocalTimeSample) pm.getObjectById(oidOfLocalTime2));

      Query<LocalTimeSample> q = pm.newQuery(LocalTimeSample.class, filter);
      List<LocalTimeSample> results = q.executeList();
      checkQueryResultWithoutOrder(ASSERTION_FAILED, filter, results, expectedResult);
      tx.commit();
      tx = null;
    } finally {
      if ((tx != null) && tx.isActive()) tx.rollback();
    }
  }

  /** */
  public void testSecond() {
    final String filter = "localTime.getSecond() == 25";
    PersistenceManager pm = getPM();
    Transaction tx = pm.currentTransaction();
    try {
      tx.begin();
      List<LocalTimeSample> expectedResult = new ArrayList<>();
      expectedResult.add((LocalTimeSample) pm.getObjectById(oidOfLocalTime1));

      Query<LocalTimeSample> q = pm.newQuery(LocalTimeSample.class, filter);
      List<LocalTimeSample> results = q.executeList();
      checkQueryResultWithoutOrder(ASSERTION_FAILED, filter, results, expectedResult);
      tx.commit();
      tx = null;
    } finally {
      if ((tx != null) && tx.isActive()) tx.rollback();
    }
  }

  /**
   * @see org.apache.jdo.tck.JDO_Test#localSetUp()
   */
  @Override
  protected void localSetUp() {
    addTearDownClass(LocalTimeSample.class);
    insertLocalTimeSampleData(getPM());
  }

  /** */
  private void insertLocalTimeSampleData(PersistenceManager pm) {
    Transaction tx = pm.currentTransaction();
    try {
      tx.begin();

      LocalTimeSample lds1 = new LocalTimeSample();
      lds1.setId(1);
      LocalTime localTime1 = LocalTime.of(14, 10, 25);
      lds1.setLocalTime(localTime1);
      pm.makePersistent(lds1);

      LocalTimeSample lds2 = new LocalTimeSample();
      lds2.setId(2);
      LocalTime localTime2 = LocalTime.of(9, 22, 12);
      lds2.setLocalTime(localTime2);
      pm.makePersistent(lds2);

      tx.commit();
      oidOfLocalTime1 = pm.getObjectId(lds1);
      oidOfLocalTime2 = pm.getObjectId(lds2);
      tx = null;
    } finally {
      if ((tx != null) && tx.isActive()) tx.rollback();
    }
  }
}
