Add JPA test support