@Test
public void testVDBRestart() throws Exception {
String vdbName = "test";
String testVDB =
"<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?>\n"
+ "<vdb name=\"test\" version=\"1\">\n"
+ " <property name=\"UseConnectorMetadata\" value=\"cached\" />\n"
+ " <model name=\"loopy\">\n"
+ " <source name=\"loop\" translator-name=\"loopy\" />\n"
+ " </model>\n"
+ "</vdb>";
Collection<?> vdbs = admin.getVDBs();
assertTrue(vdbs.isEmpty());
JavaArchive jar = getLoopyArchive();
admin.deploy("loopy.jar", jar.as(ZipExporter.class).exportAsInputStream());
// normal load
admin.deploy("test-vdb.xml", new ByteArrayInputStream(testVDB.getBytes()));
AdminUtil.waitForVDBLoad(admin, vdbName, 1, 3);
int count = assertMetadataLoadCount(false, 1);
// 1st restart
admin.restartVDB(vdbName, 1);
AdminUtil.waitForVDBLoad(admin, vdbName, 1, 3);
count = assertMetadataLoadCount(true, count + 1);
// 2nd restart
admin.restartVDB(vdbName, 1);
AdminUtil.waitForVDBLoad(admin, vdbName, 1, 3);
count = assertMetadataLoadCount(true, count + 1);
admin.undeploy("loopy.jar");
}
@Test
public void testSystemPropertiesInVDBXML() throws Exception {
String vdbName = "test";
String testVDB =
"<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?>\n"
+ "<vdb name=\"test\" version=\"1\">\n"
+ " <property name=\"UseConnectorMetadata\" value=\"${teiid.vdb.UseConnectorMetadata:none}\" />\n"
+ " <model name=\"loopy\">\n"
+ " <source name=\"loop\" translator-name=\"loopy\" />\n"
+ " </model>\n"
+ "</vdb>";
Collection<?> vdbs = admin.getVDBs();
assertTrue(vdbs.isEmpty());
JavaArchive jar = getLoopyArchive();
admin.deploy("loopy.jar", jar.as(ZipExporter.class).exportAsInputStream());
// normal load
admin.deploy("test-vdb.xml", new ByteArrayInputStream(testVDB.getBytes()));
AdminUtil.waitForVDBLoad(admin, vdbName, 1, 3);
VDB vdb = admin.getVDB(vdbName, 1);
String value = vdb.getPropertyValue("UseConnectorMetadata");
// see the arquillian.zml file in resources in the JVM proeprties section for the expected value
assertEquals("custom", value);
admin.undeploy("loopy.jar");
admin.undeploy("test-vdb.xml");
}
@Test
public void testDDLExport() throws Exception {
String vdbName = "test";
String testVDB =
"<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?>\n"
+ "<vdb name=\"test\" version=\"1\">\n"
+ " <property name=\"UseConnectorMetadata\" value=\"cached\" />\n"
+ " <model name=\"loopy\">\n"
+ " <source name=\"loop\" translator-name=\"loopy\" />\n"
+ " </model>\n"
+ "</vdb>";
Collection<?> vdbs = admin.getVDBs();
assertTrue(vdbs.isEmpty());
JavaArchive jar = getLoopyArchive();
admin.deploy("loopy.jar", jar.as(ZipExporter.class).exportAsInputStream());
// normal load
admin.deploy("test-vdb.xml", new ByteArrayInputStream(testVDB.getBytes()));
AdminUtil.waitForVDBLoad(admin, vdbName, 1, 3);
String ddl = admin.getSchema(vdbName, 1, "loopy", null, null);
String expected = "CREATE FOREIGN TABLE Matadata (\n" + " execCount integer\n" + ");";
assertEquals(expected, ddl);
admin.undeploy("loopy.jar");
}
@Test
public void testVDBConnectionType() throws Exception {
admin.deploy("bqt.vdb", new FileInputStream(UnitTestUtil.getTestDataFile("bqt.vdb")));
VDB vdb = admin.getVDB("bqt", 1);
Model model = vdb.getModels().get(0);
admin.updateSource("bqt", 1, "Source", "h2", "java:jboss/datasources/ExampleDS");
try {
// should not be able to remove from non-multisource
admin.removeSource("bqt", 1, model.getName(), "Source");
fail();
} catch (AdminException e) {
}
assertEquals(ConnectionType.BY_VERSION, vdb.getConnectionType());
Connection conn =
TeiidDriver.getInstance()
.connect("jdbc:teiid:bqt@mm://localhost:31000;user=user;password=user", null);
conn.close();
admin.changeVDBConnectionType("bqt", 1, ConnectionType.NONE);
try {
TeiidDriver.getInstance()
.connect("jdbc:teiid:bqt@mm://localhost:31000;user=user;password=user", null);
fail("should have failed to connect as no new connections allowed");
} catch (Exception e) {
// pass
}
admin.deploy("bqt2.vdb", new FileInputStream(UnitTestUtil.getTestDataFile("bqt2.vdb")));
admin.updateSource("bqt", 2, "Source", "h2", "java:jboss/datasources/ExampleDS");
conn =
TeiidDriver.getInstance()
.connect("jdbc:teiid:bqt@mm://localhost:31000;user=user;password=user", null);
conn.close();
admin.changeVDBConnectionType("bqt", 2, ConnectionType.ANY);
conn =
TeiidDriver.getInstance()
.connect("jdbc:teiid:bqt@mm://localhost:31000;user=user;password=user", null);
conn.close();
vdb = admin.getVDB("bqt", 2);
model = vdb.getModels().get(0);
assertEquals(model.getSourceConnectionJndiName("Source"), "java:jboss/datasources/ExampleDS");
assertEquals(model.getSourceTranslatorName("Source"), "h2");
assertEquals(ConnectionType.ANY, vdb.getConnectionType());
}
@Test
public void testChainedDelegates() throws Exception {
Properties props = new Properties();
props.setProperty("connection-url", "jdbc:h2:mem:test;DB_CLOSE_DELAY=-1");
props.setProperty("user-name", "sa");
props.setProperty("password", "sa");
AdminUtil.createDataSource(admin, "Oracle11_PushDS", "h2", props);
admin.deploy("fake.jar", new FileInputStream(UnitTestUtil.getTestDataFile("fake.jar")));
try {
admin.deploy(
"chained-vdb.xml", new FileInputStream(UnitTestUtil.getTestDataFile("chained-vdb.xml")));
} finally {
admin.undeploy("fake.jar");
}
}
@Test
public void testODBCConnectionSuccess() throws Exception {
admin.deploy("bqt.vdb", new FileInputStream(UnitTestUtil.getTestDataFile("bqt.vdb")));
Driver d = new Driver();
Properties p = new Properties();
p.setProperty("user", "user");
p.setProperty("password", "user");
Connection c = d.connect("jdbc:postgresql://127.0.0.1:35432/bqt", p);
c.close();
}
@Test
public void testErrorDeployment() throws Exception {
Collection<?> vdbs = admin.getVDBs();
assertTrue(vdbs.isEmpty());
admin.deploy(
"error-vdb.xml", new FileInputStream(UnitTestUtil.getTestDataFile("error-vdb.xml")));
AdminUtil.waitForVDBLoad(admin, "error", 1, 3);
VDB vdb = admin.getVDB("error", 1);
assertEquals(Status.FAILED, vdb.getStatus());
}
private boolean deployVdb() throws AdminException, FileNotFoundException {
boolean vdbOneDeployed;
admin.deploy("bqt.vdb", new FileInputStream(UnitTestUtil.getTestDataFile("bqt.vdb")));
AdminUtil.waitForVDBLoad(admin, "bqt", 1, 3);
vdbOneDeployed = true;
VDB vdb = admin.getVDB("bqt", 1);
Model model = vdb.getModels().get(0);
admin.assignToModel(
"bqt", 1, model.getName(), "Source", "h2", "java:jboss/datasources/ExampleDS");
return vdbOneDeployed;
}
@Test
public void testGetRequests() throws Exception {
JavaArchive jar = getLoopyArchive();
try {
admin.deploy("loopy.jar", jar.as(ZipExporter.class).exportAsInputStream());
deployVdb();
Translator t = admin.getTranslator("loopy");
assertNotNull(t);
admin.updateSource("bqt", 1, "Source", "loopy", "java:jboss/datasources/ExampleDS");
Connection conn =
TeiidDriver.getInstance()
.connect("jdbc:teiid:bqt@mm://localhost:31000;user=user;password=user", null);
Collection<? extends Session> sessions = admin.getSessions();
assertEquals(1, sessions.size());
Session s = sessions.iterator().next();
Statement stmt = conn.createStatement();
Collection<? extends Request> requests = admin.getRequests();
assertEquals(0, requests.size());
stmt.execute("select * from source.smalla");
requests = admin.getRequests();
assertEquals(1, requests.size());
Request r = requests.iterator().next();
assertEquals("select * from source.smalla", r.getCommand());
assertNotNull(r.getExecutionId());
assertNotNull(r.getSessionId());
stmt.execute("select * from source.smalla");
Collection<? extends Request> requests2 = admin.getRequestsForSession(s.getSessionId());
assertEquals(1, requests2.size());
Request r2 = requests.iterator().next();
assertEquals("select * from source.smalla", r2.getCommand());
assertEquals(s.getSessionId(), r2.getSessionId());
stmt.close();
conn.close();
requests = admin.getRequests();
assertEquals(0, requests.size());
} finally {
admin.undeploy("loopy.jar");
}
}
@Test
public void testODBCConnectionFailure() throws Exception {
admin.deploy("bqt.vdb", new FileInputStream(UnitTestUtil.getTestDataFile("bqt.vdb")));
Driver d = new Driver();
Properties p = new Properties();
p.setProperty("user", "user");
p.setProperty("password", "notpassword");
try {
d.connect("jdbc:postgresql://127.0.0.1:35432/bqt", p);
fail("failed due to bad credentials");
} catch (SQLException e) {
}
}
@Test
public void testDataRoleMapping() throws Exception {
admin.deploy("bqt2.vdb", new FileInputStream(UnitTestUtil.getTestDataFile("bqt2.vdb")));
VDB vdb = admin.getVDB("bqt", 2);
Model model = vdb.getModels().get(0);
admin.assignToModel(
"bqt", 2, model.getName(), "Source", "h2", "java:jboss/datasources/ExampleDS");
vdb = admin.getVDB("bqt", 2);
assertTrue(vdb.isValid());
List<DataPolicy> policies = vdb.getDataPolicies();
assertEquals(1, policies.size());
DataPolicy dp = policies.get(0);
assertEquals("roleOne", dp.getName());
assertEquals(2, dp.getPermissions().size());
assertTrue(dp.isAllowCreateTemporaryTables());
assertTrue(dp.isAnyAuthenticated());
List<String> roleNames = dp.getMappedRoleNames();
assertArrayEquals(new String[] {"ROLE1", "ROLE2"}, roleNames.toArray());
admin.removeDataRoleMapping("bqt", 2, "roleOne", "ROLE1");
vdb = admin.getVDB("bqt", 2);
policies = vdb.getDataPolicies();
dp = policies.get(0);
roleNames = dp.getMappedRoleNames();
assertArrayEquals(new String[] {"ROLE2"}, roleNames.toArray());
admin.addDataRoleMapping("bqt", 2, "roleOne", "ROLE3");
vdb = admin.getVDB("bqt", 2);
policies = vdb.getDataPolicies();
dp = policies.get(0);
roleNames = dp.getMappedRoleNames();
assertArrayEquals(new String[] {"ROLE2", "ROLE3"}, roleNames.toArray());
admin.setAnyAuthenticatedForDataRole("bqt", 2, "roleOne", false);
vdb = admin.getVDB("bqt", 2);
policies = vdb.getDataPolicies();
dp = policies.get(0);
assertFalse(dp.isAnyAuthenticated());
}
@Test
public void testTraslators() throws Exception {
Collection<? extends Translator> translators = admin.getTranslators();
assertEquals(38, translators.size());
JavaArchive jar = getLoopyArchive();
try {
admin.deploy("loopy.jar", jar.as(ZipExporter.class).exportAsInputStream());
VDBTranslatorMetaData t = (VDBTranslatorMetaData) admin.getTranslator("loopy");
assertNotNull(t);
assertEquals("ANY", t.getPropertyValue("SupportedJoinCriteria"));
assertEquals("true", t.getPropertyValue("supportsSelectDistinct"));
} finally {
admin.undeploy("loopy.jar");
}
VDBTranslatorMetaData t = (VDBTranslatorMetaData) admin.getTranslator("orcl");
assertNull(t);
}
@Test
public void testVDBDeployment() throws Exception {
Collection<?> vdbs = admin.getVDBs();
assertTrue(vdbs.isEmpty());
Collection<String> dsNames = admin.getDataSourceNames();
if (dsNames.contains("Oracle11_PushDS")) {
admin.deleteDataSource("Oracle11_PushDS");
}
admin.deploy("bqt.vdb", new FileInputStream(UnitTestUtil.getTestDataFile("bqt.vdb")));
vdbs = admin.getVDBs();
assertFalse(vdbs.isEmpty());
VDB vdb = admin.getVDB("bqt", 1);
assertFalse(vdb.isValid());
assertTrue(AdminUtil.waitForVDBLoad(admin, "bqt", 1, 3));
assertFalse(vdb.isValid());
Properties props = new Properties();
props.setProperty("connection-url", "jdbc:h2:mem:test;DB_CLOSE_DELAY=-1");
props.setProperty("user-name", "sa");
props.setProperty("password", "sa");
props.setProperty("connection-properties", "foo=bar,blah=blah");
admin.createDataSource("Oracle11_PushDS", "h2", props);
vdb = admin.getVDB("bqt", 1);
assertTrue(vdb.isValid());
assertTrue(vdb.getStatus().equals(Status.ACTIVE));
dsNames = admin.getDataSourceNames();
assertTrue(dsNames.contains("Oracle11_PushDS"));
admin.deleteDataSource("Oracle11_PushDS");
vdb = admin.getVDB("bqt", 1);
assertFalse(vdb.isValid());
}