KieBase kbase = ks.newKieContainer(ks.getRepository().getDefaultReleaseId()).getKieBase();
Properties properties = new Properties();
properties.put("drools.processInstanceManagerFactory", "com.example.CustomJPAProcessInstanceManagerFactory");
KieSessionConfiguration config = ks.newKieSessionConfiguration(properties);
KieSession ksession = ks.getStoreServices().newKieSession( kbase, config, env );
SessionConfiguration sessionConfig = (SessionConfiguration)ksession.getSessionConfiguration();
assertEquals("com.example.CustomJPAProcessInstanceManagerFactory", sessionConfig.getProcessInstanceManagerFactory());