@Test
public void testChangeNameOnExistingDataSources() throws Exception {
sm.register("file", super.getAnyNonSpatialResource());
DataSource ds = dsf.getDataSourceFromSQL("select * from file;");
sm.rename(ds.getName(), "sql");
DataSource ds2 = dsf.getDataSource("sql");
assertEquals(ds.getName(), ds2.getName());
}
@Test
public void testSQLSources() throws Exception {
sm.register("testH", super.getAnyNonSpatialResource());
sm.register("sql", "select * from testH;");
DataSource ds = dsf.getDataSource("sql");
assertEquals((ds.getSource().getType() & SourceManager.SQL), SourceManager.SQL);
assertFalse(ds.isEditable());
}
@Test
public void testNameAndID() {
Map<String, Object> raw = new HashMap<String, Object>();
raw.put("id", "abcdef");
raw.put("name", "MyDS");
DataSource dataSource = new DataSource(raw, mock(ApiClient.class));
assertEquals(dataSource.getId(), raw.get("id"));
assertEquals(dataSource.getName(), raw.get("name"));
}
@Test
public void toJson() throws Exception {
List<Student> ls = new ArrayList<>(data.StudentRepo().values());
InputMeta testObj = new InputMeta(ls, data.ProjectRepo());
String expected = testObj.getHottestProject();
String allRaw = testObj.toJson();
assertTrue(allRaw.contains("hottestProject"));
String head = "\"hottestProject\": \"";
int strBegin = allRaw.indexOf(head) + head.length();
int strEnd = strBegin + expected.length();
String got = allRaw.substring(strBegin, strEnd);
assertTrue(expected.equals(got));
}
@Test
public void testRemoveDataSources() throws Exception {
sm.register("temp", super.getAnyNonSpatialResource());
DataSource d = dsf.getDataSource("temp");
sm.remove(d.getName());
try {
dsf.getDataSource("temp");
fail();
} catch (NoSuchTableException e) {
}
}
@Test
public void testRemoveCallsAPIEndpoint() {
ServerBridge bridge = mock(ServerBridge.class);
when(bridge.delete("datasources/a")).thenReturn("");
ApiClient api = new ApiClient("abc");
api.setServerBridge(bridge);
Map<String, Object> raw = new HashMap<String, Object>();
raw.put("id", "a");
raw.put("name", "MyDS");
DataSource dataSource = new DataSource(raw, api);
dataSource.remove();
verify(bridge).delete("datasources/a");
}
@Test
public void testGetVariablesEmptyList() {
ServerBridge bridge = mock(ServerBridge.class);
when(bridge.get("datasources/a/variables")).thenReturn("[]");
ApiClient api = new ApiClient("abc");
api.setServerBridge(bridge);
Map<String, Object> raw = new HashMap<String, Object>();
raw.put("id", "a");
raw.put("name", "MyDS");
DataSource dataSource = new DataSource(raw, api);
Variable[] variables = dataSource.getVariables();
verify(bridge).get("datasources/a/variables");
assertEquals(0, variables.length);
}
@Test
public void testGetVariablesThreeVariables() {
ServerBridge bridge = mock(ServerBridge.class);
when(bridge.get("datasources/a/variables"))
.thenReturn(
"["
+ "{'id': 'a', 'name': 'A', 'unit': 'unitA', 'icon': 'iconA'},"
+ "{'id': 'b', 'name': 'B', 'unit': 'unitB', 'icon': 'iconB'},"
+ "{'id': 'c', 'name': 'C', 'unit': 'unitC', 'icon': 'iconC'}"
+ "]");
ApiClient api = new ApiClient("abc");
api.setServerBridge(bridge);
Map<String, Object> raw = new HashMap<String, Object>();
raw.put("id", "a");
raw.put("name", "MyDS");
DataSource dataSource = new DataSource(raw, api);
Variable[] variables = dataSource.getVariables();
verify(bridge).get("datasources/a/variables");
assertEquals(3, variables.length);
// Check IDs
assertEquals("a", variables[0].getId());
assertEquals("b", variables[1].getId());
assertEquals("c", variables[2].getId());
// Check names
assertEquals("A", variables[0].getName());
assertEquals("B", variables[1].getName());
assertEquals("C", variables[2].getName());
// Check units
assertEquals("unitA", variables[0].getUnit());
assertEquals("unitB", variables[1].getUnit());
assertEquals("unitC", variables[2].getUnit());
// Check icons
assertEquals("iconA", variables[0].getIcon());
assertEquals("iconB", variables[1].getIcon());
assertEquals("iconC", variables[2].getIcon());
}
@Test
public void testSecondNameWorksWithIndexes() throws Exception {
String dsName = "test8";
sm.register(dsName, super.getAnySpatialResource());
String secondName = "secondName";
DataSource ds = dsf.getDataSource(dsName);
ds.open();
String spatialFieldName =
ds.getMetadata().getFieldName(MetadataUtilities.getSpatialFieldIndex(ds.getMetadata()));
ds.close();
sm.addName(dsName, secondName);
dsf.getIndexManager()
.buildIndex(dsName, spatialFieldName, IndexManager.RTREE_SPATIAL_INDEX, null);
SpatialIndexQuery query =
new DefaultSpatialIndexQuery(spatialFieldName, new Envelope(0, 0, 0, 0));
assertNotNull(dsf.getIndexManager().getIndex(dsName, spatialFieldName));
assertNotNull(dsf.getIndexManager().getIndexedFieldNames(dsName));
assertNotNull(dsf.getIndexManager().queryIndex(dsName, query));
assertNotNull(dsf.getIndexManager().getIndex(secondName, spatialFieldName));
assertNotNull(dsf.getIndexManager().getIndexedFieldNames(secondName));
assertNotNull(dsf.getIndexManager().queryIndex(secondName, query));
}
private void checkNames(String dsName, String secondName)
throws DriverLoadException, NoSuchTableException, DataSourceCreationException,
DriverException, AlreadyClosedException {
assertEquals(sm.getSource(dsName), sm.getSource(secondName));
DataSource ds1 = dsf.getDataSource(dsName);
DataSource ds2 = dsf.getDataSource(secondName);
ds1.open();
Value[][] ds1c = getDataSourceContents(ds1);
ds1.close();
ds2.open();
Value[][] ds2c = getDataSourceContents(ds2);
ds2.close();
assertTrue(equals(ds1c, ds2c));
}
@Test
public void testOperationDataSourceName() throws Throwable {
sm.register("temp", super.getAnyNonSpatialResource());
DataSource d = dsf.getDataSourceFromSQL("select * from temp;");
assertNotNull(dsf.getDataSource(d.getName()));
}
@Test
public void testAllGeters() throws Exception {
AllTypesObjectDriver testD = new AllTypesObjectDriver();
DataSet test = testD.getTable("main");
DataSource d = dsf.getDataSource("alltypes");
d.open();
assertTrue(
ValueFactory.createValue(d.getBinary(0, 0))
.equals(test.getFieldValue(0, 0))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getBinary(0, "binary"))
.equals(test.getFieldValue(0, 0))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getBoolean(0, 1))
.equals(test.getFieldValue(0, 1))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getBoolean(0, "boolean"))
.equals(test.getFieldValue(0, 1))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getByte(0, 2)).equals(test.getFieldValue(0, 2)).getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getByte(0, "byte"))
.equals(test.getFieldValue(0, 2))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getDate(0, 3)).equals(test.getFieldValue(0, 3)).getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getDate(0, "date"))
.equals(test.getFieldValue(0, 3))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getDouble(0, 4))
.equals(test.getFieldValue(0, 4))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getDouble(0, "double"))
.equals(test.getFieldValue(0, 4))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getFloat(0, 5)).equals(test.getFieldValue(0, 5)).getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getFloat(0, "float"))
.equals(test.getFieldValue(0, 5))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getInt(0, 6)).equals(test.getFieldValue(0, 6)).getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getInt(0, "int"))
.equals(test.getFieldValue(0, 6))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getLong(0, 7)).equals(test.getFieldValue(0, 7)).getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getLong(0, "long"))
.equals(test.getFieldValue(0, 7))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getShort(0, 8)).equals(test.getFieldValue(0, 8)).getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getShort(0, "short"))
.equals(test.getFieldValue(0, 8))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getString(0, 9))
.equals(test.getFieldValue(0, 9))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getString(0, "string"))
.equals(test.getFieldValue(0, 9))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getTimestamp(0, 10))
.equals(test.getFieldValue(0, 10))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getTimestamp(0, "timestamp"))
.equals(test.getFieldValue(0, 10))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getTime(0, 11))
.equals(test.getFieldValue(0, 11))
.getAsBoolean());
assertTrue(
ValueFactory.createValue(d.getTime(0, "time"))
.equals(test.getFieldValue(0, 11))
.getAsBoolean());
d.close();
}
@Test
public void testSetters() throws Exception {
DataSource d = dsf.getDataSource("alltypes");
d.open();
d.setBinary(0, 0, d.getFieldValue(1, 0).getAsBinary());
d.setBinary(0, "binary", d.getFieldValue(1, 0).getAsBinary());
d.setBoolean(0, 1, d.getFieldValue(1, 1).getAsBoolean());
d.setBoolean(0, "boolean", d.getFieldValue(1, 1).getAsBoolean());
d.setByte(0, 2, d.getFieldValue(1, 2).getAsByte());
d.setByte(0, "byte", d.getFieldValue(1, 2).getAsByte());
d.setDate(0, 3, d.getFieldValue(1, 3).getAsDate());
d.setDate(0, "date", d.getFieldValue(1, 3).getAsDate());
d.setDouble(0, 4, d.getFieldValue(1, 4).getAsDouble());
d.setDouble(0, "double", d.getFieldValue(1, 4).getAsDouble());
d.setFloat(0, 5, d.getFieldValue(1, 5).getAsFloat());
d.setFloat(0, "float", d.getFieldValue(1, 5).getAsFloat());
d.setInt(0, 6, d.getFieldValue(1, 6).getAsInt());
d.setInt(0, "int", d.getFieldValue(1, 6).getAsInt());
d.setLong(0, 7, d.getFieldValue(1, 7).getAsLong());
d.setLong(0, "long", d.getFieldValue(1, 7).getAsLong());
d.setShort(0, 8, d.getFieldValue(1, 8).getAsShort());
d.setShort(0, "short", d.getFieldValue(1, 8).getAsShort());
d.setString(0, 9, d.getFieldValue(1, 9).getAsString());
d.setString(0, "string", d.getFieldValue(1, 9).getAsString());
d.setTimestamp(0, "timestamp", d.getFieldValue(1, 10).getAsTimestamp());
d.setTime(0, 11, d.getFieldValue(1, 11).getAsTime());
d.setTime(0, "time", d.getFieldValue(1, 11).getAsTime());
for (int i = 0; i < d.getMetadata().getFieldCount(); i++) {
assertEquals(d.getFieldValue(0, i), d.getFieldValue(1, i));
}
d.close();
}