/*
 * 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.openjpa.persistence.jpql.joins.leftfetch;

import java.util.List;

import javax.persistence.EntityManager;
import javax.persistence.Query;

import org.apache.openjpa.persistence.OpenJPAPersistence;
import org.apache.openjpa.persistence.OpenJPAQuery;
import org.apache.openjpa.persistence.test.SingleEMFTestCase;

public class TestJoinLeftFetch extends SingleEMFTestCase {    
    
    public void setUp() {
        setUp(DROP_TABLES, DepartmentTest.class, PersonTest.class);
        createTestData();
    }    
    
    /*
     * This test fails (prior to OJ-2475) because the 
     * DepartmentTests are not populated with the correct
     * number of PersonTests
     */
    public void testReadDepartmentsWithLeftJoinFetch() {

        EntityManager em = emf.createEntityManager();

        String qStrDIST = "SELECT DISTINCT dept FROM DepartmentTest "
            + "dept LEFT JOIN FETCH dept.persons";
        
        Query query = em.createQuery(qStrDIST);
        List<DepartmentTest> depts = query.getResultList();
        verifySize(depts);
        em.close();
    }

    public void verifySize(List<DepartmentTest> depts){
        for (DepartmentTest department : depts) {
            if (department.getPrimaryKey().equals("001")) {           
//                System.out.println("Dept: " + department.getName());
  //              Iterator i = department.getPersons().iterator();
    //            while (i.hasNext()){
      //              System.out.println("i.next() = " + i.next());
        //        }
                assertEquals("Size should be 3", 3, department.getPersons().size());
            }
            if (department.getPrimaryKey().equals("002")) {           
//                System.out.println("Dept: " + department.getName());
  //              Iterator i = department.getPersons().iterator();
    //            while (i.hasNext()){
      //              System.out.println("i.next() = " + i.next());
        //        }
                assertEquals("Size should be 2", 2, department.getPersons().size());
            }
        }
    }
    
    /*
     * This test works as expected.
     */
    public void testReadDepartmentsWithFetchPlan() {

        EntityManager em = emf.createEntityManager();

        OpenJPAQuery<DepartmentTest> query = OpenJPAPersistence.cast(em.createQuery(" SELECT dept FROM "
            + " DepartmentTest dept "));
        query.getFetchPlan().addField(DepartmentTest.class, "persons");
        
        verifySize(query.getResultList());
        
        em.close();
    }

    /*
     * This test works as expected.
     */
    public void testReadDepartmentsWithLeftJoinFetchAndOrderBy() {

        EntityManager em = emf.createEntityManager();

        Query query = em.createQuery(" SELECT dept FROM " + " DepartmentTest dept "
            + " LEFT JOIN FETCH dept.persons ORDER BY dept.primaryKey");
        verifySize(query.getResultList());
        
        em.close();
    }

    public void createTestData() {
        // NOTE: This test depends upon the the PersonTest
        // to be un-ordered w.r.t the DepartmentTest FK.
        // I've executed a flush after each entity creation 
        // in an attempt that the FKs will not be ordered.  
        // @OrderBy is used in the DepartmentTest in order
        // to ensure things aren't orderd by the FK.
        
        EntityManager em = emf.createEntityManager();
        em.getTransaction().begin();
        
        DepartmentTest dt1 = new DepartmentTest();
        dt1.setPrimaryKey("001");
        dt1.setName("Dept001");
        em.persist(dt1);
                
        DepartmentTest dt2 = new DepartmentTest();
        dt2.setPrimaryKey("002");
        dt2.setName("Dept002");
        em.persist(dt2);  
        
        PersonTest pt = new PersonTest();
        pt.setPrimaryKey("1");
        pt.setName("John");
        pt.setDepartmentTest(dt1);
        em.persist(pt);
        em.flush();

        pt = new PersonTest();
        pt.setPrimaryKey("2");
        pt.setName("Mark");
        pt.setDepartmentTest(dt1);
        em.persist(pt);
        em.flush();
        
        pt = new PersonTest();
        pt.setPrimaryKey("3");
        pt.setName("Stuart");
        pt.setDepartmentTest(dt2);
        em.persist(pt);
        em.flush();
        
        pt = new PersonTest();
        pt.setPrimaryKey("4");
        pt.setName("Jim");
        pt.setDepartmentTest(dt1);
        em.persist(pt);
        em.flush();

        pt = new PersonTest();
        pt.setPrimaryKey("5");
        pt.setName("Fred");
        pt.setDepartmentTest(dt2);
        em.persist(pt);
        em.flush();

        em.getTransaction().commit();
        em.close();
    }    
}