EntityUtilsTest.java

package com.ctp.cdi.query.util;

import java.io.Serializable;

import org.junit.Assert;

import org.junit.Test;

import com.ctp.cdi.query.test.domain.Tee;
import com.ctp.cdi.query.test.domain.Tee2;
import com.ctp.cdi.query.test.domain.TeeId;

public class EntityUtilsTest {

    @Test
    public void should_find_id_property_class() {
        // given

        // when
        Class<? extends Serializable> pkClass = EntityUtils.primaryKeyClass(Tee.class);

        // then
        Assert.assertEquals(TeeId.class, pkClass);
    }

    @Test
    public void should_find_id_class() {
        // given

        // when
        Class<? extends Serializable> pkClass = EntityUtils.primaryKeyClass(Tee2.class);

        // then
        Assert.assertEquals(TeeId.class, pkClass);
    }
}