-
Notifications
You must be signed in to change notification settings - Fork 14
/
SesamePersistenceUnitTest.java
106 lines (95 loc) · 4.48 KB
/
SesamePersistenceUnitTest.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
/**
* Copyright (C) 2016 Czech Technical University in Prague
*
* This program is free software: you can redistribute it and/or modify it under
* the terms of the GNU General Public License as published by the Free Software
* Foundation, either version 3 of the License, or (at your option) any
* later version.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
* details. You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package cz.cvut.kbss.jopa.test.integration;
import cz.cvut.kbss.jopa.Persistence;
import cz.cvut.kbss.jopa.model.EntityManager;
import cz.cvut.kbss.jopa.model.EntityManagerFactory;
import cz.cvut.kbss.jopa.model.JOPAPersistenceProperties;
import cz.cvut.kbss.jopa.model.JOPAPersistenceProvider;
import cz.cvut.kbss.jopa.test.OWLClassA;
import cz.cvut.kbss.jopa.test.environment.Generators;
import cz.cvut.kbss.jopa.test.environment.SesamePersistenceFactory;
import cz.cvut.kbss.jopa.test.runner.PersistenceUnitTestRunner;
import cz.cvut.kbss.ontodriver.config.OntoDriverProperties;
import cz.cvut.kbss.ontodriver.sesame.SesameDataSource;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.File;
import java.nio.file.Files;
import java.util.*;
import static org.junit.Assert.assertNotNull;
public class SesamePersistenceUnitTest extends PersistenceUnitTestRunner {
private static final Logger LOG = LoggerFactory.getLogger(SesamePersistenceUnitTest.class);
private SesamePersistenceFactory persistenceFactory;
public SesamePersistenceUnitTest() {
super(LOG);
this.persistenceFactory = new SesamePersistenceFactory();
}
@Override
protected EntityManager getEntityManager(String repositoryName, boolean cacheEnabled) {
return getEntityManager(repositoryName, cacheEnabled, Collections.emptyMap());
}
@Override
protected EntityManager getEntityManager(String repositoryName, boolean cacheEnabled,
Map<String, String> properties) {
return persistenceFactory.getEntityManager(repositoryName, cacheEnabled, properties);
}
@Test
public void multiplePersistenceUnitsOnLocalNativeStoreCanExistSimultaneously() throws Exception {
final File dir = Files.createTempDirectory("sesame-native-test").toFile();
dir.deleteOnExit();
final int count = Generators.randomPositiveInt(5);
final List<EntityManagerFactory> emfs = new ArrayList<>();
for (int i = 0; i < count; i++) {
emfs.add(buildLocalNativeEmf("testPu" + i, dir));
}
final List<OWLClassA> instances = new ArrayList<>();
emfs.forEach(emf -> {
final EntityManager em = emf.createEntityManager();
try {
em.getTransaction().begin();
final OWLClassA a = new OWLClassA();
a.setUri(Generators.generateUri());
em.persist(a);
em.getTransaction().commit();
instances.add(a);
} finally {
em.close();
}
});
emfs.forEach(emf -> {
final OWLClassA a = instances.get(0);
final EntityManager em = emf.createEntityManager();
try {
assertNotNull(em.find(OWLClassA.class, a.getUri()));
} finally {
em.close();
}
});
emfs.forEach(EntityManagerFactory::close);
}
private EntityManagerFactory buildLocalNativeEmf(String puName, File directory) {
final Map<String, String> config = new HashMap<>();
config.put(JOPAPersistenceProperties.ONTOLOGY_PHYSICAL_URI_KEY,
directory.getAbsolutePath() + File.separator + "repositories" + File.separator + "test");
config.put(JOPAPersistenceProperties.JPA_PERSISTENCE_PROVIDER, JOPAPersistenceProvider.class.getName());
config.put(OntoDriverProperties.USE_TRANSACTIONAL_ONTOLOGY, Boolean.TRUE.toString());
config.put(JOPAPersistenceProperties.LANG, "en");
config.put(JOPAPersistenceProperties.DATA_SOURCE_CLASS, SesameDataSource.class.getName());
config.put(JOPAPersistenceProperties.SCAN_PACKAGE, "cz.cvut.kbss.jopa.test");
return Persistence.createEntityManagerFactory(puName, config);
}
}