/*
 * Copyright (c) 2009, 2012, Oracle and/or its affiliates. All rights reserved.
 * ORACLE PROPRIETARY/CONFIDENTIAL. Use is subject to license terms.
 */
package com.sun.ts.tests.jpa.core.derivedid.ex4b;


import com.sun.javatest.Status;
import com.sun.ts.lib.util.TestUtil;
import com.sun.ts.tests.jpa.common.PMClientBase;

import java.util.List;
import java.util.Properties;

/**
 * @author Raja Perumal
 */
public class Client extends PMClientBase {

    public Client() {
    }

    public static void main(String[] args) {
        Client theTests = new Client();
        Status s = theTests.run(args, System.out, System.err);
        s.exit();
    }

    public void setup(String[] args, Properties p) throws Fault {
        TestUtil.logTrace("setup");
        try {
            super.setup(args, p);
            removeTestData();
        } catch (Exception e) {
            TestUtil.logErr("Exception: " , e);
            throw new Fault("Setup failed:", e);
        }
    }

    /**
     * @testName: DIDTest
     * @assertion_ids: PERSISTENCE:SPEC:1339;
     * @test_Strategy: Derived Identifier
     * <p/>
     * The parent entity has a simple primary key
     * Case (b): The primary key consists of a single attribute corresponding
     * to the simple primary key of the parent entity. The dependent
     * entity has a primary key attribute in addition to the
     * relationship attribute corresponding to the primary key.
     * This attribute is mapped to the primary key by the
     * MappedByIdannotation applied to the relationship.
     */
    public void DIDTest() throws Fault {
        boolean pass = false;


        try {


            getEntityTransaction().begin();
            final DID4bPerson person = new DID4bPerson("123456789", "DUKE");
            final DID4bMedicalHistory mHistory = new DID4bMedicalHistory("123456789", person, "drFoo");

            getEntityManager().persist(person);
            getEntityManager().persist(mHistory);

            TestUtil.logTrace("persisted Patient and MedicalHistory");
            getEntityManager().flush();

            // Refresh MedicalHistory
            DID4bMedicalHistory newMHistory = getEntityManager().find(DID4bMedicalHistory.class, "123456789");
            if (newMHistory != null) {
                getEntityManager().refresh(newMHistory);
            }

            final List depList = getEntityManager().createQuery("Select m from DID4bMedicalHistory m where m.patient.ssn='123456789'").getResultList();
            newMHistory = null;
            if (depList.size() > 0) {
                newMHistory = (DID4bMedicalHistory) depList.get(0);
                if (newMHistory != null) {
                    if (newMHistory.getPatient() == person) {
                        pass = true;
                        TestUtil.logTrace("Received Expected Patient");
                    } else {
                        TestUtil.logErr("Searched Patient not found");
                    }
                } else {
                    TestUtil.logErr("getEntityManager().createQuery returned null entry");
                }
            } else {
                TestUtil.logErr("getEntityManager().createQuery returned null");
            }

            getEntityTransaction().commit();
        } catch (Exception e) {
            TestUtil.logErr("Unexpected exception occurred", e);
            getEntityTransaction().rollback();
        }

        if (!pass) {
            throw new Fault("DTDTest failed");
        }
    }

    public void cleanup() throws Fault {
        TestUtil.logTrace("cleanup");
        removeTestData();
        TestUtil.logTrace("cleanup complete, calling super.cleanup");
        super.cleanup();
    }

    private void removeTestData() {
        TestUtil.logTrace("removeTestData");
        if (getEntityTransaction().isActive()) {
            getEntityTransaction().rollback();
        }
        try {
            getEntityTransaction().begin();
            getEntityManager().createNativeQuery("DELETE FROM DID4BMEDICALHISTORY").executeUpdate();
            getEntityManager().createNativeQuery("DELETE FROM DID4BPERSON").executeUpdate();
            getEntityTransaction().commit();
        } catch (Exception e) {
            TestUtil.logErr("Exception encountered while removing entities:",e);
        } finally {
            try {
                if (getEntityTransaction().isActive()) {
                    getEntityTransaction().rollback();
                }
            } catch (Exception re) {
                TestUtil.logErr("Unexpected Exception in removeTestData:",re);
            }
        }
    }
}
