-
Notifications
You must be signed in to change notification settings - Fork 14
/
SesameDataSourceTest.java
124 lines (109 loc) · 4.31 KB
/
SesameDataSourceTest.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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
/**
* Copyright (C) 2016 Czech Technical University in Prague
* <p>
* 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.
* <p>
* 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.ontodriver.sesame;
import cz.cvut.kbss.ontodriver.Connection;
import cz.cvut.kbss.ontodriver.sesame.environment.TestUtils;
import cz.cvut.kbss.ontodriver.sesame.exceptions.SesameDriverException;
import org.eclipse.rdf4j.repository.Repository;
import org.eclipse.rdf4j.repository.sail.SailRepository;
import org.eclipse.rdf4j.sail.memory.MemoryStore;
import org.hamcrest.CoreMatchers;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.ExpectedException;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import java.lang.reflect.Field;
import static org.junit.Assert.*;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.*;
public class SesameDataSourceTest {
@Rule
public ExpectedException thrown = ExpectedException.none();
@Mock
private SesameDriver driverMock;
private SesameDataSource dataSource;
@Before
public void setUp() throws Exception {
MockitoAnnotations.initMocks(this);
this.dataSource = new SesameDataSource();
TestUtils.setMock("driver", dataSource, driverMock);
}
@Test(expected = NullPointerException.class)
public void testSesameDataSourceSetNullStorageProperties() {
final SesameDataSource ds = new SesameDataSource();
ds.setStorageProperties(null);
fail("This line should not have been reached.");
assert ds == null;
}
@Test
public void testClose() throws Exception {
assertTrue(dataSource.isOpen());
dataSource.close();
assertFalse(dataSource.isOpen());
dataSource.close();
assertFalse(dataSource.isOpen());
}
@Test
public void testGetConnection() throws Exception {
Field connected = dataSource.getClass().getDeclaredField("connected");
connected.setAccessible(true);
connected.set(dataSource, true);
final Connection res = dataSource.getConnection();
verify(driverMock).acquireConnection();
assertNull(res);
}
@Test(expected = IllegalStateException.class)
public void testGetConnectionWithoutInitialization() throws Exception {
final Connection res = dataSource.getConnection();
// This shouldn't be reached
assert res == null;
}
@Test(expected = IllegalStateException.class)
public void testGetConnectionOnClosed() throws Exception {
dataSource.close();
assertFalse(dataSource.isOpen());
try {
dataSource.getConnection();
} finally {
verify(driverMock, never()).acquireConnection();
}
}
@Test
public void setRepositorySetsUnderlyingRepository() throws Exception {
Field connected = dataSource.getClass().getDeclaredField("connected");
connected.setAccessible(true);
connected.set(dataSource, true);
final Repository repo = new SailRepository(new MemoryStore());
repo.initialize();
try {
dataSource.setRepository(repo);
verify(driverMock).setRepository(repo);
} finally {
repo.shutDown();
}
}
@Test
public void setRepositoryWrapsIllegalStateExceptionIntoSesameDriverException() throws Exception {
Field connected = dataSource.getClass().getDeclaredField("connected");
connected.setAccessible(true);
connected.set(dataSource, true);
doThrow(new IllegalStateException()).when(driverMock).setRepository(any());
thrown.expect(SesameDriverException.class);
thrown.expectCause(CoreMatchers.isA(IllegalStateException.class));
dataSource.setRepository(null);
}
}