JPAIdentityStoreConfiguration configuration = new JPAIdentityStoreConfiguration();
configuration.addRealm(Realm.DEFAULT_REALM);
configuration.addRealm("Testing");
configuration.setIdentityClass(IdentityObject.class);
configuration.setAttributeClass(IdentityObjectAttribute.class);
configuration.setRelationshipClass(RelationshipObject.class);
configuration.setRelationshipIdentityClass(RelationshipIdentityObject.class);
configuration.setRelationshipAttributeClass(RelationshipObjectAttribute.class);
configuration.setCredentialClass(CredentialObject.class);
configuration.setCredentialAttributeClass(CredentialObjectAttribute.class);
configuration.setPartitionClass(PartitionObject.class);

FeatureSet.addFeatureSupport(configuration.getFeatureSet());
FeatureSet.addRelationshipSupport(configuration.getFeatureSet());
FeatureSet.addRelationshipSupport(configuration.getFeatureSet(), CustomRelationship.class);  //You can install custom relationships
FeatureSet.addRelationshipSupport(configuration.getFeatureSet(), Authorization.class);
configuration.getFeatureSet().setSupportsCustomRelationships(true);
configuration.getFeatureSet().setSupportsMultiRealm(true);

return configuration;

EntityManagerFactory emf = Persistence.createEntityManagerFactory("jpa-identity-store-tests-pu");

EntityManager entityManager = emf.createEntityManager();
entityManager.getTransaction().begin();

IdentityConfiguration config = new IdentityConfiguration();
config.addStoreConfiguration(getDefaultConfiguration());

IdentityManager identityManager = new DefaultIdentityManager();
DefaultIdentityStoreInvocationContextFactory icf = new DefaultIdentityStoreInvocationContextFactory(this.emf);
icf.setEntityManager(this.entityManager);
identityManager.bootstrap(config, icf);