import static org.junit.Assert.assertEquals;

import java.util.Iterator;

import javax.persistence.Column;
import javax.persistence.Entity;
import javax.persistence.FetchType;
import javax.persistence.GeneratedValue;
import javax.persistence.Id;
import javax.persistence.JoinColumn;
import javax.persistence.OneToOne;
import javax.persistence.Table;

import org.hibernate.annotations.ForeignKey;
import org.hibernate.testing.junit4.BaseCoreFunctionalTestCase;
import org.junit.Test;

/**
 * Created by alexandr on 02.02.15.
 */
public class HibernateForeignKeyTest extends BaseCoreFunctionalTestCase
{
    private static final String USER_ID = "USER_ID";
    private static final String PERSONAL_DATA_ID = "PERSONAL_DATA_ID";

    private static final String USER_SETTINGS_TABLE_NAME = "user_settings";
    private static final String USERS_TABLE_NAME = "users";
    private static final String PERSONALDATA_TABLE_NAME = "personal_data";

    private static final String EXPLICIT_USER_SETTING_FK_NAME = "FK_USER_SETTING_TO_USER";
    private static final String EXPLICIT_USERS_FK_NAME = "FK_USER_SETTING_TO_USER_BACK";
    private static final String EXPLICIT_PD_FK_NAME = "FK_PERSONAL_DATA_TO_USER";


    @Entity
    @Table( name = USER_SETTINGS_TABLE_NAME )
    public static class UserSetting {
        @Id
        @GeneratedValue
        @Column(name = USER_ID, nullable = false)
        public long id;

        @OneToOne(optional = false, fetch = FetchType.LAZY, targetEntity = User.class)
        @JoinColumn(name = USER_ID, nullable = false, updatable = false)
        @ForeignKey(name = EXPLICIT_USER_SETTING_FK_NAME)
        private User user;
    }

    @Entity
    @Table( name = USERS_TABLE_NAME )
    public static class User {
        @Id
        @GeneratedValue
        @Column(name = USER_ID)
        public long id;

        @OneToOne(fetch = FetchType.LAZY, optional = false, targetEntity = UserSetting.class)
        @JoinColumn(name = USER_ID)
        @ForeignKey(name = EXPLICIT_USERS_FK_NAME)
        public UserSetting userSetting;
    }

    @Test
    public void testForeignKeyUserSetting(){

        Iterator<org.hibernate.mapping.Table> tableItr = configuration().getTableMappings();
        while (tableItr.hasNext())
        {
            org.hibernate.mapping.Table table = tableItr.next();
            Iterator fkItr = table.getForeignKeyIterator();
            while (fkItr.hasNext())
            {
                org.hibernate.mapping.ForeignKey fk = (org.hibernate.mapping.ForeignKey) fkItr.next();
                if (USER_SETTINGS_TABLE_NAME.equals(table.getName())){
                    assertEquals(EXPLICIT_USER_SETTING_FK_NAME, fk.getName());
                }
            }
        }
    }

    @Test
    public void testForeignKeyUsers(){

        Iterator<org.hibernate.mapping.Table> tableItr = configuration().getTableMappings();
        while (tableItr.hasNext())
        {
            org.hibernate.mapping.Table table = tableItr.next();
            Iterator fkItr = table.getForeignKeyIterator();
            while (fkItr.hasNext())
            {
                org.hibernate.mapping.ForeignKey fk = (org.hibernate.mapping.ForeignKey) fkItr.next();
                if (USERS_TABLE_NAME.equals(table.getName())){
                    assertEquals(EXPLICIT_USERS_FK_NAME, fk.getName());
                }
            }
        }
    }

    @Override
    protected Class<?>[] getAnnotatedClasses() {
        return new Class<?>[] {
                User.class, UserSetting.class
        };
    }
}